52 lines
1.3 KiB
Plaintext
52 lines
1.3 KiB
Plaintext
|
#!/bin/python3
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import logging
|
||
|
|
import requests
|
||
|
|
import json
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
logging.basicConfig(format='%(levelname)s\t- %(message)s')
|
||
|
|
logger = logging.getLogger()
|
||
|
|
logger.setLevel(logging.INFO)
|
||
|
|
|
||
|
|
state = {}
|
||
|
|
|
||
|
|
# session file - document name - output file(optional)
|
||
|
|
def getDoc(args):
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
|
||
|
|
parser.add_argument('session', nargs='?', default=None)
|
||
|
|
parser.add_argument('name', nargs='?', default=None)
|
||
|
|
parser.add_argument('output', nargs='?', default=None)
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
if not args.session or not args.name:
|
||
|
|
logger.error("Need session file and document's name.")
|
||
|
|
sys.exit(-1)
|
||
|
|
|
||
|
|
if (not os.path.isfile(args.session)):
|
||
|
|
logger.error("File '" + args.session + "' not found.")
|
||
|
|
sys.exit(-1)
|
||
|
|
|
||
|
|
doc = {'session' : args.session, 'name' : args.name}
|
||
|
|
|
||
|
|
try:
|
||
|
|
req = requests.post(f'http://{state['REP_ADDRESS']}/subject/', json=json.dumps(doc))
|
||
|
|
req.raise_for_status()
|
||
|
|
except requests.exceptions.RequestException as errex:
|
||
|
|
logger.error("Failed to obtain response from server.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if args.output:
|
||
|
|
with open(args.output, 'w') as f:
|
||
|
|
f.write(req)
|
||
|
|
|
||
|
|
else:
|
||
|
|
sys.stdout.write(req)
|
||
|
|
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
getDoc(sys.argv[1:])
|