lutris/tests/check_prefixes.py

#!/usr/bin/env python3
import os
import subprocess
import sys

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from lutris.util.wine.registry import WineRegistry

PREFIXES_PATH = os.path.expanduser("~/Games/wine/prefixes")


def get_registries():
    registry_list = []
    directories = os.listdir(PREFIXES_PATH)
    directories.append(os.path.expanduser("~/.wine"))
    for prefix in directories:
        for path in os.listdir(os.path.join(PREFIXES_PATH, prefix)):
            if path.endswith(".reg"):
                registry_list.append(os.path.join(PREFIXES_PATH, prefix, path))
    return registry_list


def check_registry(registry_path):
    with open(registry_path, "r") as registry_file:
        original_content = registry_file.read()

    try:
        wine_registry = WineRegistry(registry_path)
    except:
        sys.stderr.write("Error parsing {}\n".format(registry_path))
        raise
    content = wine_registry.render()
    if content != original_content:
        wrong_path = os.path.join(os.path.dirname(__file__), "error.reg")
        with open(wrong_path, "w") as wrong_reg:
            wrong_reg.write(content)

        print("Content of parsed registry doesn't match: {}".format(registry_path))
        subprocess.call(["meld", registry_path, wrong_path])
        sys.exit(2)


registries = get_registries()
for registry in registries:
    check_registry(registry)
print("All {} registry files validated!".format(len(registries)))