You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
4.9 KiB
Python
114 lines
4.9 KiB
Python
import os, argparse, sys, re
|
|
from mwclient import (Site,
|
|
errors)
|
|
from jinja2 import Template
|
|
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('--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('--dir', required=True,
|
|
help='Required. Full path of the image directory, that you wish to upload')
|
|
|
|
p.add_argument('--title', required=True,
|
|
help='Required. Must not exist yet in the wiki.')
|
|
p.add_argument('--date', required=True,
|
|
help='Required. Format: yyyy/mm/dd '
|
|
'For dates without day or month use 01 as default '
|
|
'ie. 1986: --date "1986/01/01" '
|
|
'March 1985: --date "1984/05/01"')
|
|
p.add_argument('--creator', required=False, action='append', default=[''],
|
|
help='Multiple values should be SEPARATED BY COMMA')
|
|
p.add_argument('--org', required=False, action='append', default=[''],
|
|
help='Organization:Multiple values should be SEPARATED BY '
|
|
'COMMA')
|
|
p.add_argument('--format', required=False, action='append', default=[''],
|
|
help='Multiple values should be SEPARATED BY COMMA')
|
|
p.add_argument('--event', required=False, action='append', default=[''],
|
|
help='Multiple values should be SEPARATED BY COMMA')
|
|
p.add_argument('--topic', required=False, action='append', default=[''],
|
|
help='Multiple values should be SEPARATED BY COMMA')
|
|
|
|
# TODO ADD NEW PROPS
|
|
args = p.parse_args()
|
|
|
|
# login
|
|
site = Site(host=args.host, path=args.path)
|
|
|
|
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()
|
|
|
|
# read template file
|
|
with open(os.path.join(wd, 'templates/smw_infobox_template.jinja')) as tmplt:
|
|
smw_propval_template = Template(tmplt.read())
|
|
|
|
lsimgs = reorder_imgs(dir=args.dir, dry=args.dry)
|
|
dirname = os.path.split(args.dir)[-1].replace(' ', '_')
|
|
dirname = re.sub(r'[\W]', '', dirname) #remove non letters or digits
|
|
# print('lsimgs:', lsimgs, '\n', dirname)
|
|
|
|
for n, _file in enumerate(lsimgs):
|
|
pagename = f'{dirname}-{_file}'
|
|
print_colormsg(pagename, level='ok')
|
|
page = site.pages[_file]
|
|
|
|
if page.exists:
|
|
url = page.imageinfo['descriptionurl']
|
|
print_colormsg(
|
|
f'Already exists in {url} Will NOT be uploaded',
|
|
level='warning')
|
|
else:
|
|
img_smw_prop_val = smw_propval_template.render(
|
|
title=args.title,
|
|
date=args.date,
|
|
part=n + 1,
|
|
partof=len(lsimgs),
|
|
creator=(', ').join(args.creator[1:]),
|
|
organization=(', ').join(args.org[1:]),
|
|
format=(', ').join(args.format[1:]),
|
|
event=(', ').join(args.event[1:]),
|
|
topic=(', ').join(args.topic[1:])
|
|
)
|
|
|
|
_file_path = os.path.join(args.dir, _file)
|
|
if not args.dry:
|
|
pageurl = f'https://{args.host}{args.path}index.php/File:{pagename}'
|
|
with open(_file_path, 'rb') as _f:
|
|
try:
|
|
site.upload(file=_file_path,
|
|
filename=pagename,
|
|
description=img_smw_prop_val,
|
|
ignore=True)
|
|
print(img_smw_prop_val)
|
|
except errors.APIError as e:
|
|
print_colormsg(f'Error: {e.info}\n'
|
|
f'It will not be uploaded',
|
|
level='fail')
|
|
|
|
print(f'See image at {pageurl}')
|
|
else:
|
|
print(img_smw_prop_val)
|
|
|