def change_ownership_of_files()

in scripts/in_container/run_fix_ownership.py [0:0]

• 49 lines of code
• 25 McCabe index (conditional complexity)

def change_ownership_of_files(path: Path) -> None:
    host_user_id = os.environ.get("HOST_USER_ID", "")
    host_group_id = os.environ.get("HOST_GROUP_ID", "")
    if host_user_id == "" or host_group_id == "":
        print(
            f"ERROR: HOST_USER_ID or HOST_GROUP_ID environment variables "
            f"are not set: {host_user_id}:{host_group_id}"
        )
        sys.exit(1)
    if not host_user_id.isnumeric() or not host_group_id.isnumeric():
        print(
            f"ERROR: HOST_USER_ID or HOST_GROUP_ID environment variables "
            f"should be numeric: {host_user_id}:{host_group_id}"
        )
        sys.exit(1)
    count_files = 0
    root_uid = pwd.getpwnam("root").pw_uid
    skip_folders = {".venv", "node_modules"}
    for root, dirs, files in os.walk(path):
        original_length = len(dirs)
        original_dirs = dirs.copy()
        # skip known big folders if they are not owned by root
        dirs[:] = [d for d in dirs if d not in skip_folders or (Path(root) / d).stat().st_uid == root_uid]
        new_length = len(dirs)
        if new_length != original_length:
            if os.environ.get("VERBOSE", "false") == "true":
                print(
                    f"{root}: Skipped {original_length - new_length} "
                    f"folders: {set(original_dirs) - set(dirs)}"
                )
        for dir_name in dirs:
            os.chown(Path(root) / dir_name, int(host_user_id), int(host_group_id), follow_symlinks=False)
        for name in files:
            file = Path(root) / name
            try:
                if file.is_symlink() and file.lstat().st_uid == root_uid:
                    # Change ownership of symlink itself (by default stat/chown follow the symlinks)
                    os.chown(file, int(host_user_id), int(host_group_id), follow_symlinks=False)
                    count_files += 1
                    if os.environ.get("VERBOSE_COMMANDS", "false") == "true":
                        print(f"Changed ownership of symlink {file}")
                if file.stat().st_uid == root_uid:
                    # And here change ownership of the file (or if it is a symlink - the file it points to)
                    os.chown(file, int(host_user_id), int(host_group_id))
                    count_files += 1
                    if os.environ.get("VERBOSE_COMMANDS", "false") == "true":
                        print(f"Changed ownership of {file.resolve()}")
            except FileNotFoundError:
                # This is OK - file might have been deleted in the meantime or linked in Host from
                # another place
                if os.environ.get("VERBOSE_COMMANDS", "false") == "true":
                    print(f"Could not change ownership of {file}")
    if count_files and os.environ.get("VERBOSE_COMMANDS", "false") == "true":
        print(f"Changed ownership of {count_files} files back to {host_user_id}:{host_group_id}.")