judge.py

import os, sys, subprocess

file = sys.argv[1] if len(sys.argv) > 1 else input('Filename?')
extension = file.split('.')[-1].lower() if '.' in file else ''
language = extension if extension in ['py', 'java', 'class', 'cpp', 'exe'] else input('Language?').lower()
problem = input('Problem?')

def get_args():
    global file
    if language in ['python', 'py', 'python3', 'py3']:
        return ['py', file]
    if language in ['java']:
        subprocess.run(['javac', file], capture_output=True, text=True, check=True)
    if language in ['java', 'class']:
        return ['java', os.path.splitext(os.path.basename(file))[0]]
    if language in ['cpp', 'c++']:
        subprocess.run(['g++', file], capture_output=True, text=True, check=True)
        file = 'a.exe'
    if language in ['cpp', 'c++', 'app', 'exe', 'executable']:
        return [file]

verdict = 'Something went wrong'
for i in os.listdir('.'):
    if i.startswith(str(problem).zfill(2)):
        if not os.path.isdir('!LOGS'):
            os.mkdir('!LOGS')
        for j in os.listdir('!LOGS'):
            os.remove(f'!LOGS/{j}')
        verdict = 'Verdict: '
        
        try:
            args = get_args()
        except subprocess.CalledProcessError as e:
            verdict = 'C'
            with open(f'!LOGS/err.err', 'w') as f:
                f.write(e.stderr)
            break
        if not args:
            verdict = 'Something went wrong'
            break
        for j in range(10):
            with open(f'{i}/{j}.txt') as f:
                process = subprocess.run(['py', f'{i}/main.py'], capture_output=True, text=True, input=f.read())
                a = [k.strip() for k in process.stdout.splitlines() if k.strip()]
                with open(f'!LOGS/{j}.txt', 'w') as g:
                    g.write(process.stdout)
                f.seek(0)
                v = 'W'
                process = None
                try:
                    process = subprocess.run(args, capture_output=True, text=True, input=f.read(), timeout=10, check=True)
                except subprocess.TimeoutExpired as e:
                    v = 'T'
                    process = e
                except (subprocess.CalledProcessError, IndexError) as e:
                    v = 'E'
                    process = e
                b = None
                try:
                    b = [k.strip() for k in (process.stdout or '').splitlines() if k.strip()]
                    with open(f'!LOGS/{j}.out', 'w') as g:
                        g.write(process.stdout or '')
                    with open(f'!LOGS/{j}.err', 'w') as g:
                        g.write(process.stderr or '')
                except AttributeError as e:
                    pass
                if a == b:
                    v = 'A'
                verdict += v
        break
print(verdict)
input()