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.
88 lines
3.8 KiB
Python
88 lines
3.8 KiB
Python
import os, argparse, sys, re
|
|
from mwclient import Site
|
|
from functions import print_colormsg
|
|
|
|
p = argparse.ArgumentParser(description='Upload files from a directory, with metadata values to the wiki.\n'
|
|
'Note that any value 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()
|
|
|
|
|
|
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
|
|
|
|
site = Site(host=args.host, path=args.path)
|
|
# login
|
|
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
|
|
|
|
|
|
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()
|
|
|
|
# TODO os.listdir(args.dir) ORDER!
|
|
# TODO PartOf from os.listdir(args.dir) that fit on to condition
|
|
# TODO Part base on os.lisdir order!
|
|
for _file in os.listdir(args.dir):
|
|
if (os.path.splitext(_file)[-1]).lower() not in ['.jpg', '.jpeg', '.png']:
|
|
print_colormsg(f'File: {_file} is not a .jpg .jpeg or .png hence will Not be upload', level='warning')
|
|
else:
|
|
print_colormsg(_file, level='ok')
|
|
|
|
if args.dry == True:
|
|
print(args)
|
|
sys.exit()
|
|
|
|
|
|
smw_prop_val = '''{{ImageMetadata
|
|
|Title={title}
|
|
|Date={date}
|
|
|Part={part}
|
|
|Partof={partof}
|
|
|Creator={creator}
|
|
}}
|
|
[[Template:ImageMetadata]]
|
|
'''
|
|
|
|
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:
|
|
site.upload(file=_file, filename=filename, description='img_smw_prop_val', ignore=True)
|
|
|