createprecomplete.py

Enable keyboard shortcuts

# Any copyright is dedicated to the Public Domain.

# http://creativecommons.org/publicdomain/zero/1.0/

# Creates the precomplete file containing the remove and rmdir application

# update instructions which is used to remove files and directories that are no

# longer present in a complete update. The current working directory is used for

# the location to enumerate and to create the precomplete file.

import io

import os

def get_build_entries(root_path):

    """Iterates through the root_path, creating a list for each file and

    directory. Excludes any file paths ending with channel-prefs.js.

"""

    rel_file_path_set = set()

    rel_dir_path_set = set()

    for root, dirs, files in os.walk(root_path):

        for file_name in files:

            parent_dir_rel_path = root[len(root_path) + 1 :]

            rel_path_file = os.path.join(parent_dir_rel_path, file_name)

            rel_path_file = rel_path_file.replace("\\", "/")

            if not (

                rel_path_file.endswith("channel-prefs.js")

                or rel_path_file.endswith("update-settings.ini")

                or "/ChannelPrefs.framework/" in rel_path_file

                or rel_path_file.startswith("ChannelPrefs.framework/")

                or "/UpdateSettings.framework/" in rel_path_file

                or rel_path_file.startswith("UpdateSettings.framework/")

                or "distribution/" in rel_path_file

):

                rel_file_path_set.add(rel_path_file)

        for dir_name in dirs:

            parent_dir_rel_path = root[len(root_path) + 1 :]

            rel_path_dir = os.path.join(parent_dir_rel_path, dir_name)

            rel_path_dir = rel_path_dir.replace("\\", "/") + "/"

            if rel_path_dir.find("distribution/") == -1:

                rel_dir_path_set.add(rel_path_dir)

    rel_file_path_list = list(rel_file_path_set)

    rel_file_path_list.sort(reverse=True)

    rel_dir_path_list = list(rel_dir_path_set)

    rel_dir_path_list.sort(reverse=True)

    return rel_file_path_list, rel_dir_path_list

def generate_precomplete(root_path):

    """Creates the precomplete file containing the remove and rmdir

    application update instructions. The given directory is used

    for the location to enumerate and to create the precomplete file.

"""

    rel_path_precomplete = "precomplete"

    # If inside a Mac bundle use the root of the bundle for the path.

    if os.path.basename(root_path) == "Resources":

        root_path = os.path.abspath(os.path.join(root_path, "../../"))

        rel_path_precomplete = "Contents/Resources/precomplete"

    precomplete_file_path = os.path.join(root_path, rel_path_precomplete)

    # Open the file so it exists before building the list of files and open it

    # in binary mode to prevent OS specific line endings.

    precomplete_file = io.open(precomplete_file_path, mode="wt", newline="\n")

    rel_file_path_list, rel_dir_path_list = get_build_entries(root_path)

    for rel_file_path in rel_file_path_list:

        precomplete_file.write('remove "' + rel_file_path + '"\n')

    for rel_dir_path in rel_dir_path_list:

        precomplete_file.write('rmdir "' + rel_dir_path + '"\n')

    precomplete_file.close()

if __name__ == "__main__":

    generate_precomplete(os.getcwd())