"""Checks if all *.pyc files have later mtime than their *.py files."""

import os

import sys

from importlib.util import cache_from_source

from pathlib import Path

RPM_BUILD_ROOT = os.environ.get('RPM_BUILD_ROOT', '')

# ...cpython-3X.pyc

# ...cpython-3X.opt-1.pyc

# ...cpython-3X.opt-2.pyc

LEVELS = (None, 1, 2)

# list of globs of test and other files that we expect not to have bytecode

not_compiled = [

    '/usr/bin/*',

    '*/test/bad_coding.py',

    '*/test/bad_coding2.py',

    '*/test/badsyntax_*.py',

    '*/lib2to3/tests/data/bom.py',

    '*/lib2to3/tests/data/crlf.py',

    '*/lib2to3/tests/data/different_encoding.py',

    '*/lib2to3/tests/data/false_encoding.py',

    '*/lib2to3/tests/data/py2_test_grammar.py',

    '*.debug-gdb.py',

]

def bytecode_expected(path):

    path = Path(path[len(RPM_BUILD_ROOT):])

    for glob in not_compiled:

        if path.match(glob):

            return False

    return True

failed = 0

compiled = (path for path in sys.argv[1:] if bytecode_expected(path))

for path in compiled:

    to_check = (cache_from_source(path, optimization=opt) for opt in LEVELS)

    f_mtime = os.path.getmtime(path)

    for pyc in to_check:

        c_mtime = os.path.getmtime(pyc)

        if c_mtime < f_mtime:

            print('Failed bytecompilation timestamps check: '

                  f'Bytecode file {pyc} is older than source file {path}',

                  file=sys.stderr)

            failed += 1

if failed:

    print(f'\n{failed} files failed bytecompilation timestamps check.',

          file=sys.stderr)

    sys.exit(1)