3
0
mirror of https://github.com/jlu5/PyLink.git synced 2024-11-27 21:19:31 +01:00
PyLink/pylink-contribdl
2017-02-27 07:28:37 -08:00

65 lines
2.0 KiB
Python
Executable File

#!/usr/bin/env python3
"""
Contrib module manager for PyLink IRC Services.
"""
from distutils.util import strtobool
import sys
import os
import os.path
import urllib.request
import traceback
BASE_URL = 'https://github.com/PyLink/pylink-contrib-modules'
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('path', nargs='+', help='specifies the paths to download (e.g. '
'"plugins/helloworld", "protocols/blah", etc.)')
parser.add_argument('--version', '-v', nargs='?', help='specifies the tag (version) to download from',
default='master')
parser.add_argument('--yes', '-y', action='store_true', help='skip yes/no confirmations')
if len(sys.argv) < 2:
parser.print_help()
print()
print('For a list of available contrib modules, see %s' % BASE_URL)
sys.exit(1)
args = parser.parse_args()
print('This will download the following URLs:')
print()
urls = {}
for target in set(args.path):
if not target.startswith(('plugins', 'protocols')):
print('ERROR: Target paths should be in one of the following forms: "plugins/<pluginname>" '
'or "protocols/<modulename>" (got %r)' % target, file=sys.stderr)
sys.exit(1)
target += '.py'
url = '%s/blob/%s/%s' % (BASE_URL, args.version, target)
urls[target] = url
print(url, '=>', '%s' % os.path.join(os.getcwd(), target))
if not args.yes:
text = input('Continue (Y)es/(n)o? ').lower()
if not strtobool(text):
sys.exit(1)
print()
for filename, url in urls.items():
print('Retrieving %s...' % url)
try:
urllib.request.urlretrieve(url, filename)
except:
# If one request errors, move onto the next one.
traceback.print_exc()
continue
print()
print('All downloads finished.')