import os, argparse, sys, re from mwclient import Site from functions import (print_colormsg, reorder_imgs) p = argparse.ArgumentParser(description='Upload files from a directory, with metadata values to the wiki.\n' 'Note that any VALUES CONTAINING ' 'SPACES SHOULD BE BETWEEN QUOTATION MARKS', formatter_class=argparse.ArgumentDefaultsHelpFormatter) # TODO: Add example of command to description p.add_argument('--host', default='hub.xpub.nl/sandbox', help='wiki host') p.add_argument('--path', default='/itchwiki/', help='Wiki path. Should end with /') p.add_argument('--scheme', default='https', help='http or https') p.add_argument('--dry', '-d', action='store_true', help='dry-run: will only print the metadata of each file that will be upload, but does NOT upload') p.add_argument('--title', required=True, help='Metadata **Title** value of publication. Must not exist yet in the wiki.') p.add_argument('--date', required=True, help='Metadata **Date** value of publication. Format yyyy/mm/dd ' 'For dates without day or date use 01 ie. 1986 --date "1986/01/01" March 1985: --date "1984/05/01"') p.add_argument('--creator', required=True, action='append', help='Metadata **Creator** value(s) of publication.' 'Multiple values should be SEPARATED BY COMMA') p.add_argument('--dir', required=True, help='Full path of the image directory, ' 'that you wish to upload') args = p.parse_args() # login site = Site(host=args.host, path=args.path, scheme=args.scheme) wd =os.path.dirname(os.path.abspath(__file__)) # parent working directory with open(os.path.join(wd, 'login.txt'), 'r') as login: # read login user & pwd loginlines = login.read() user, pwd = loginlines.split('\n') site.login(username=user, password=pwd) # login to wiki # metadata checks if os.path.isdir(args.dir) is False: print_colormsg(f'Error: --dir {args.dir} absolute path cannot be found', level='fail') sys.exit() elif not re.match(r'\d{4}\/\d{2}\/\d{2}', args.date): print_colormsg(f'Error: --date {args.date} format should be --date "yyyy/mm/dd"', level='fail') sys.exit() elif len(list(site.ask(f'[[Title::{args.title}]]'))) > 0: print_colormsg(f'Error: --title "{args.title}" already exists in wiki. Provide a different one', level='fail') sys.exit() lsimgs = reorder_imgs(dir=args.dir, dry=args.dry) print('lsimgs:', lsimgs) sys.exit() ## STOP HERE # TODO PartOf from os.listdir(args.dir) that fit on to condition # TODO Part base on os.lisdir order! # todo creator value comma-seperated smw_prop_val = '''{{ImageMetadata |Title={title} |Date={date} |Part={part} |Partof={partof} |Creator={creator} }} [[Template:ImageMetadata]] ''' if args.dry == True: print(args) sys.exit() for n, _file in enumerate(lsimgs): print_colormsg(_file, level='ok') page = site.pages[_file] if page.exists: url = page.imageinfo['descriptionurl'] print_colormsg( f'Already exists in {url} Will be uploaded as new version', level='warning') img_smw_prop_val = smw_prop_val.format(title='cat from hell', date='2020/02/13', part='1', partof='1', creator='test') print(img_smw_prop_val) filename = 'cat2.jpg' page = site.images[filename] img_url = f'https://{args.host}{args.path}/index.php/File:{filename}' if page.exists is True: print(f'{filename} is already upload it, you can see it at {img_url}' ) print('The upload process wont proceed. Please upload all images in folder by hand') else: with open(filename, 'rb') as _file: dirname = (os.path.split(dir)[-1]) site.upload(file=_file, filename=f'{dirname}_{filename}', description='img_smw_prop_val', ignore=True)