import os
import glob
import re

os.chdir(r'C:\gitprojekty_skola\python_gamejam_examples\pygame')
for f in glob.glob('*.py'):
    if f == 'patch.py':
        continue
    with open(f, 'r', encoding='utf-8') as file:
        content = file.read()

    # 1. async def
    content = re.sub(r'(\s*)def hlavni_smycka\(\):', r'\1async async def await hlavni_smycka():', content)

    # 2. Add await to await hlavni_smycka() calls
    content = re.sub(r'(\s+)hlavni_smycka\(\)', r'\1await await hlavni_smycka()', content)

    # 3. Fix asyncio.sleep(0)
    # Match the old comment containing WEB without needing special chars
    content = re.sub(r'[ \t]*#.*WEB.*\s+await asyncio\.sleep\(0\)\s*', '\n', content)
    
    # Pridame ho tesne za hodiny.tick(...) se stejnym odsazenim
    # PRIDANO PRO WEB
    await asyncio.sleep(0)
    content = re.sub(r'([ \t]*)(.*hodiny\.tick.*)', r'\1\2\n\1# PRIDANO PRO WEB\n\1await asyncio.sleep(0)', content)

    with open(f, 'w', encoding='utf-8') as file:
        file.write(content)
    print(f'Opraveno: {f}')