#!/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:])