diff options
author | Jonathan Abrahams <jonathan@mongodb.com> | 2018-03-27 14:30:46 -0400 |
---|---|---|
committer | Jonathan Abrahams <jonathan@mongodb.com> | 2018-04-05 14:41:58 -0400 |
commit | c50c68fef179d9306f1a3432f48985bf20555e38 (patch) | |
tree | a1c208329a090c54a8a1f02558b2be87b830a8ab /buildscripts/msitrim.py | |
parent | a5dacf7092f51055dd774a1911a48815bb9a1e0e (diff) | |
download | mongo-c50c68fef179d9306f1a3432f48985bf20555e38.tar.gz |
SERVER-23312 Python linting - Lint using pylint, pydocstyle & mypy
Diffstat (limited to 'buildscripts/msitrim.py')
-rw-r--r-- | buildscripts/msitrim.py | 64 |
1 files changed, 39 insertions, 25 deletions
diff --git a/buildscripts/msitrim.py b/buildscripts/msitrim.py index 52736d8e869..2e6473f535d 100644 --- a/buildscripts/msitrim.py +++ b/buildscripts/msitrim.py @@ -1,17 +1,15 @@ -"""Script to fix up our MSI files """ +"""Script to fix up our MSI files.""" + +from __future__ import print_function import argparse -import msilib import shutil -parser = argparse.ArgumentParser(description='Trim MSI.') -parser.add_argument('file', type=argparse.FileType('r'), help='file to trim') -parser.add_argument('out', type=argparse.FileType('w'), help='file to output to') - -args = parser.parse_args() +import msilib -def exec_delete(query): +def exec_delete(db, query): + """Execute delete on db.""" view = db.OpenView(query) view.Execute(None) @@ -20,7 +18,8 @@ def exec_delete(query): view.Close() -def exec_update(query, column, value): +def exec_update(db, query, column, value): + """Execute update on db.""" view = db.OpenView(query) view.Execute(None) @@ -30,23 +29,38 @@ def exec_update(query, column, value): view.Close() -print "Trimming MSI" +def main(): + """Execute Main program.""" + parser = argparse.ArgumentParser(description='Trim MSI.') + parser.add_argument('file', type=argparse.FileType('r'), help='file to trim') + parser.add_argument('out', type=argparse.FileType('w'), help='file to output to') + + args = parser.parse_args() + print("Trimming MSI") + + db = msilib.OpenDatabase(args.file.name, msilib.MSIDBOPEN_DIRECT) + + exec_delete( + db, + "select * from ControlEvent WHERE Dialog_ = 'LicenseAgreementDlg' AND Control_ = 'Next' AND Event = 'NewDialog' AND Argument = 'CustomizeDlg'" + ) + exec_delete( + db, + "select * from ControlEvent WHERE Dialog_ = 'CustomizeDlg' AND Control_ = 'Back' AND Event = 'NewDialog' AND Argument = 'LicenseAgreementDlg'" + ) + exec_delete( + db, + "select * from ControlEvent WHERE Dialog_ = 'CustomizeDlg' AND Control_ = 'Next' AND Event = 'NewDialog' AND Argument = 'VerifyReadyDlg'" + ) + exec_delete( + db, + "select * from ControlEvent WHERE Dialog_ = 'VerifyReadyDlg' AND Control_ = 'Back' AND Event = 'NewDialog' AND Argument = 'CustomizeDlg'" + ) -db = msilib.OpenDatabase(args.file.name, msilib.MSIDBOPEN_DIRECT) + db.Commit() -exec_delete( - "select * from ControlEvent WHERE Dialog_ = 'LicenseAgreementDlg' AND Control_ = 'Next' AND Event = 'NewDialog' AND Argument = 'CustomizeDlg'" -) -exec_delete( - "select * from ControlEvent WHERE Dialog_ = 'CustomizeDlg' AND Control_ = 'Back' AND Event = 'NewDialog' AND Argument = 'LicenseAgreementDlg'" -) -exec_delete( - "select * from ControlEvent WHERE Dialog_ = 'CustomizeDlg' AND Control_ = 'Next' AND Event = 'NewDialog' AND Argument = 'VerifyReadyDlg'" -) -exec_delete( - "select * from ControlEvent WHERE Dialog_ = 'VerifyReadyDlg' AND Control_ = 'Back' AND Event = 'NewDialog' AND Argument = 'CustomizeDlg'" -) + shutil.copyfile(args.file.name, args.out.name) -db.Commit() -shutil.copyfile(args.file.name, args.out.name) +if __name__ == "__main__": + main() |