3
0
mirror of https://github.com/jlu5/PyLink.git synced 2024-11-27 13:09:23 +01:00

Add a contrib module downloader (pylink-contribdl)

Fixes #350.
This commit is contained in:
James Lu 2016-12-05 00:36:11 -08:00
parent f5633329f8
commit c479dd1753

56
pylink-contribdl Executable file
View File

@ -0,0 +1,56 @@
#!/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')
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))
text = input('Continue (Y)es/(n)o? ').lower()
if not strtobool(text):
sys.exit(2)
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.')