|
| 1 | +import base64 |
| 2 | +import hashlib |
| 3 | +import sys |
| 4 | + |
| 5 | +os = sys.argv[1] |
| 6 | + |
| 7 | +files = [ |
| 8 | + 'CADD/Raccoon2/BoincClient.py', |
| 9 | + 'CADD/Raccoon2/gui/AA_setup.py', |
| 10 | + 'CADD/Raccoon2/gui/BB_ligand.py', |
| 11 | + 'CADD/Raccoon2/gui/CC_receptor.py', |
| 12 | + 'CADD/Raccoon2/gui/EE_jobmanager.py', |
| 13 | + 'CADD/Raccoon2/gui/FF_anaylsis.py', |
| 14 | + 'CADD/Raccoon2/gui/Raccoon2GUI.py', |
| 15 | + 'CADD/Raccoon2/gui/RaccoonEngine.py', |
| 16 | + 'CADD/Raccoon2/gui/icons/boinc.png' |
| 17 | +] |
| 18 | + |
| 19 | +def get_file_hash(file): |
| 20 | + with open(file, 'rb') as f: |
| 21 | + return hashlib.sha512(f.read()).hexdigest() |
| 22 | + |
| 23 | +def get_file_content(file): |
| 24 | + with open(file, 'rb') as f: |
| 25 | + return base64.b64encode(f.read()) |
| 26 | + |
| 27 | +def build_installer_py(): |
| 28 | + with open('raccoon2_boinc_installer.py', 'w') as f: |
| 29 | + f.write('# IMPORTANT: Do not run this file directly.\n') |
| 30 | + f.write('# Please read README.md first for usage instructions.\n') |
| 31 | + f.write('import base64\n') |
| 32 | + f.write('import hashlib\n') |
| 33 | + f.write('import os\n') |
| 34 | + f.write('import shutil\n') |
| 35 | + f.write('import sys\n') |
| 36 | + f.write('curdir = os.getcwd()\n') |
| 37 | + f.write('if len(sys.argv) > 1 and sys.argv[1] == \'install\':\n') |
| 38 | + for file in files: |
| 39 | + hash = get_file_hash(file) |
| 40 | + content = get_file_content(file) |
| 41 | + if os == 'linux' or os == 'macos': |
| 42 | + f.write(' file = curdir+\'/MGLToolsPckgs/%s\'\n' % file) |
| 43 | + elif os == 'windows': |
| 44 | + f.write(' file = curdir+\'/Lib/site-packages/%s\'\n' % file) |
| 45 | + f.write(' content = %s\n' % content) |
| 46 | + f.write(' if os.path.exists(file):\n') |
| 47 | + f.write(' with open(file, \'rb\') as f:\n') |
| 48 | + f.write(' hash = hashlib.sha512(f.read()).hexdigest()\n') |
| 49 | + f.write(' if hash != \'%s\':\n' % hash) |
| 50 | + f.write(' shutil.copy(file, file+\'.orig\')\n') |
| 51 | + f.write(' with open(file, \'wb\') as f:\n') |
| 52 | + f.write(' f.write(base64.b64decode(content))\n') |
| 53 | + f.write(' else:\n') |
| 54 | + f.write(' print(\'already updated: \' + file)\n') |
| 55 | + f.write(' else:\n') |
| 56 | + f.write(' with open(file, \'wb\') as f:\n') |
| 57 | + f.write(' f.write(base64.b64decode(content))\n') |
| 58 | + f.write(' print (\'Done\')\n') |
| 59 | + f.write('elif len(sys.argv) > 1 and sys.argv[1] == \'uninstall\':\n') |
| 60 | + for file in files: |
| 61 | + if os == 'linux' or os == 'macos': |
| 62 | + f.write(' file = curdir+\'/MGLToolsPckgs/%s\'\n' % file) |
| 63 | + elif os == 'windows': |
| 64 | + f.write(' file = curdir+\'/Lib/site-packages/%s\'\n' % file) |
| 65 | + f.write(' if os.path.exists(file+\'.orig\'):\n') |
| 66 | + f.write(' shutil.copy(file+\'.orig\', file)\n') |
| 67 | + f.write(' os.remove(file+\'.orig\')\n') |
| 68 | + f.write(' print (\'Done\')\n') |
| 69 | + f.write('else:\n') |
| 70 | + f.write(' print (\'Please read README.md for usage instructions.\')\n') |
| 71 | + |
| 72 | +if __name__ == "__main__": |
| 73 | + build_installer_py() |
0 commit comments