1 # Any copyright is dedicated to the Public Domain.
2 # http://creativecommons.org/publicdomain/zero/1.0/
4 # Creates the precomplete file containing the remove and rmdir application
5 # update instructions which is used to remove files and directories that are no
6 # longer present in a complete update. The current working directory is used for
7 # the location to enumerate and to create the precomplete file.
13 def get_build_entries(root_path
):
14 """Iterates through the root_path, creating a list for each file and
15 directory. Excludes any file paths ending with channel-prefs.js.
17 rel_file_path_set
= set()
18 rel_dir_path_set
= set()
19 for root
, dirs
, files
in os
.walk(root_path
):
20 for file_name
in files
:
21 parent_dir_rel_path
= root
[len(root_path
) + 1 :]
22 rel_path_file
= os
.path
.join(parent_dir_rel_path
, file_name
)
23 rel_path_file
= rel_path_file
.replace("\\", "/")
25 rel_path_file
.endswith("channel-prefs.js")
26 or rel_path_file
.endswith("update-settings.ini")
27 or "/ChannelPrefs.framework/" in rel_path_file
28 or rel_path_file
.startswith("ChannelPrefs.framework/")
29 or "/UpdateSettings.framework/" in rel_path_file
30 or rel_path_file
.startswith("UpdateSettings.framework/")
31 or "distribution/" in rel_path_file
33 rel_file_path_set
.add(rel_path_file
)
36 parent_dir_rel_path
= root
[len(root_path
) + 1 :]
37 rel_path_dir
= os
.path
.join(parent_dir_rel_path
, dir_name
)
38 rel_path_dir
= rel_path_dir
.replace("\\", "/") + "/"
39 if rel_path_dir
.find("distribution/") == -1:
40 rel_dir_path_set
.add(rel_path_dir
)
42 rel_file_path_list
= list(rel_file_path_set
)
43 rel_file_path_list
.sort(reverse
=True)
44 rel_dir_path_list
= list(rel_dir_path_set
)
45 rel_dir_path_list
.sort(reverse
=True)
47 return rel_file_path_list
, rel_dir_path_list
50 def generate_precomplete(root_path
):
51 """Creates the precomplete file containing the remove and rmdir
52 application update instructions. The given directory is used
53 for the location to enumerate and to create the precomplete file.
55 rel_path_precomplete
= "precomplete"
56 # If inside a Mac bundle use the root of the bundle for the path.
57 if os
.path
.basename(root_path
) == "Resources":
58 root_path
= os
.path
.abspath(os
.path
.join(root_path
, "../../"))
59 rel_path_precomplete
= "Contents/Resources/precomplete"
61 precomplete_file_path
= os
.path
.join(root_path
, rel_path_precomplete
)
62 # Open the file so it exists before building the list of files and open it
63 # in binary mode to prevent OS specific line endings.
64 precomplete_file
= io
.open(precomplete_file_path
, mode
="wt", newline
="\n")
65 rel_file_path_list
, rel_dir_path_list
= get_build_entries(root_path
)
66 for rel_file_path
in rel_file_path_list
:
67 precomplete_file
.write('remove "' + rel_file_path
+ '"\n')
69 for rel_dir_path
in rel_dir_path_list
:
70 precomplete_file
.write('rmdir "' + rel_dir_path
+ '"\n')
72 precomplete_file
.close()
75 if __name__
== "__main__":
76 generate_precomplete(os
.getcwd())