From d3c7b517043f6d759b454a9e14cd2b3b9639888a Mon Sep 17 00:00:00 2001 From: duvallj Date: Sat, 2 Feb 2019 16:34:48 -0500 Subject: Make 1F335 slightly happier --- update.py | 61 ++++++++++++++++++++++++++++++------------------------------- 1 file changed, 30 insertions(+), 31 deletions(-) (limited to 'update.py') diff --git a/update.py b/update.py index a04df5c..b2d1039 100755 --- a/update.py +++ b/update.py @@ -1,15 +1,12 @@ #!/usr/bin/env python3 import argparse -import textwrap import os import sys import hashlib import shutil import requests -import colorama -from termcolor import colored parser = argparse.ArgumentParser( description="A Simple Git-Based Modpack Manager", @@ -17,8 +14,8 @@ parser = argparse.ArgumentParser( epilog='''\ Available commands: install : Downloads mods listed in downloads.txt and populates the mods folder specified in pack-location.txt - apply_updates : Using the latest downloads in latest-urls.txt, repopulates downloads.txt to reflect the most recent mod versions - check_updates : Compares downloads.txt and latest-urls.txt to see if any mods can be updated + apply_updates : Using the urls in mods.txt, repopulates downloads.txt to reflect the most recent mod versions + check_updates : Compares downloads.txt and mods.txt to see if any mods can be updated ''') parser.add_argument('command', @@ -29,10 +26,10 @@ parser.add_argument('filename', nargs='?', default="mods.txt", help="Optional filename to specify latest mods (default: mods.txt)") -parser.add_argument('--url-file', +parser.add_argument('--version-file', type=str, default="downloads.txt", - help="Optional custom URL file to download mods from (default: downloads.txt)") + help="Optional custom version file to download mods from (default: downloads.txt)") parser.add_argument('--pack-location', type=str, help="Optional custom modpack folder location (default: read from pack-location.txt)") @@ -50,9 +47,9 @@ def read_file(fil): # Apply updates to the actual mod pack def install(args): - print(colored("Updating pack...", 'green', attrs=['bold'])) + print("Updating pack...") # (fname, checksum, url) - mods = read_file(args.url_file) + mods = read_file(args.version_file) names = [mod[0] for mod in mods] for mod in mods: @@ -61,30 +58,30 @@ def install(args): hashlib.sha1(open(mod_path, 'rb').read()).hexdigest() == mod[1]: print("Skipping {mod[0]}, already up to date".format(mod=mod)) else: - print(colored('Installing {mod[0]} from {mod[2]}...'.format(mod=mod), attrs=['bold'])) + print('Installing {mod[0]} from {mod[2]}...'.format(mod=mod)) download_obj = requests.get(mod[2], stream=True) with open(mod_path, "wb") as write_file: shutil.copyfileobj(download_obj.raw, write_file) print("Done!") print() - print(colored("Removing old mods...", 'green', attrs=['bold'])) + print("Removing old mods...") for jar in os.listdir(args.pack_location): if jar not in names and os.path.splitext(jar)[1] == ".jar": os.remove(os.path.join(args.pack_location, jar)) - print(colored("Removing '{jar}'".format(jar=jar), attrs=['bold'])) + print("Removing '{jar}'".format(jar=jar)) print() - print(colored("Finished installing mods!", 'green', attrs=['bold'])) + print("Finished installing mods!") -# Using the latest urls, update downloads.txt to match the urls and have the correct sha1 +# Using the latest urls, update downloads.txt to match and have the correct sha1 def apply_updates(args): - print(colored("Populating URL File...", 'green', attrs=['bold'])) + print("Populating URL File...") mods = read_file(args.filename) - print(colored("Getting new versions of all mods...", attrs=['bold'])) - with open(args.url_file, 'w') as f: + print("Getting new versions of all mods...") + with open(args.version_file, 'w') as f: f.write('# Format: \n') for mod in mods: print("Fetching {mod[0]}...".format(mod=mod)) @@ -93,25 +90,28 @@ def apply_updates(args): f.write('{mod[0]} {hsh} {resp.url}\n'.format(mod=mod, hsh=hsh, resp=resp)) print() print("Done!") - print(colored("Updates applied to downloads.txt", 'green', attrs=['bold'])) - print(colored("[!] No mods were installed. To update your mods folder, run 'update.py install'", 'white', 'on_red', attrs=['bold', 'underline'])) + print("Updates applied to {args.version_file}".format(args=args)) + print("[!] No mods were installed. To update your mods folder, run 'update.py install'") + # Find if any updates are available def check_updates(args): - print(colored("Checking for updates to mods...", 'green', attrs=['bold'])) + print("Checking for updates to mods...") latest = read_file(args.filename) - old = read_file(args.url_file) + old = read_file(args.version_file) old_urls = [mod[2] for mod in old] - print(colored("Checking updates...", attrs=['bold'])) + print("Checking updates...") for mod in latest: - print('\033[2K',end="") # ANSI code to clear line - print("Checking for updates to {mod[0]}...".format(mod=mod),end="\r") + print("Checking for updates to {mod[0]}...".format(mod=mod), end="") + sys.stdout.flush() # takes care of line-buffered terminals resp = requests.get(mod[1]) - if not resp.url in old_urls: - print(colored(" -> Found update for {mod[0]}: {resp.url.split('/')[-1]}".format(mod=mod, resp=resp), attrs=['bold'])) + if resp.url in old_urls: + print(" No updates") + else: + print(" Found update: {resp.url.split('/')[-1]}".format(resp=resp)) - print('\033[2K' + colored("Finished checking for updates!", attrs=['bold'])) + print("Finished checking for updates!") COMMAND_MAP = { 'install': install, @@ -120,7 +120,6 @@ COMMAND_MAP = { } if __name__ == "__main__": - colorama.init() args = parser.parse_args() if not args.pack_location: @@ -129,16 +128,16 @@ if __name__ == "__main__": args.pack_location = f.read().strip() if not os.path.exists(args.pack_location): - print(colored("Error: mod folder \"" + args.pack_location + "\" does not exist.", 'red', attrs=['bold'])) + print("Error: mod folder \"" + args.pack_location + "\" does not exist.") parser.print_help() sys.exit(1) elif not os.path.isdir(args.pack_location): - print(colored("Error: mod folder \"" + args.pack_location + "\" is not actually a folder.", 'red', attrs=['bold'])) + print("Error: mod folder \"" + args.pack_location + "\" is not actually a folder.") parser.print_help() sys.exit(1) if not (args.command in COMMAND_MAP): - print(colored("Error: command \"" + args.command + "\" does not exist", 'red', attrs=['bold'])) + print("Error: command \"" + args.command + "\" does not exist") parser.print_help() sys.exit(1) -- cgit v1.2.3