import os

replacements = {
    "app.modules.takeoff.application": "app.modules.takeoff.application",
    "app.modules.takeoff.application.drawing_takeoff_service": "app.modules.takeoff.application.drawing_takeoff_service",
    "app.modules.takeoff.application": "app.modules.takeoff.application",
}

def replace_in_file(filepath):
    try:
        with open(filepath, "r", encoding="utf-8") as f:
            content = f.read()
        
        new_content = content
        for old_import, new_import in replacements.items():
            new_content = new_content.replace(old_import, new_import)
            
        if content != new_content:
            with open(filepath, "w", encoding="utf-8") as f:
                f.write(new_content)
            print(f"Updated {filepath}")
    except Exception as e:
        pass

for root, _, files in os.walk("."):
    if ".venv" in root or "__pycache__" in root or ".git" in root:
        continue
    for file in files:
        if file.endswith(".py"):
            replace_in_file(os.path.join(root, file))
