import argparse
import requests
import sys

URL = "https://biosig.lab.uq.edu.au/dynamut2/api/prediction_mm"

def main(args):
    pdb_file = args.pdb_file
    mutations_list = args.mutations_list

    files_to_submit = {"pdb_file": open(pdb_file,"rb"),"mutations_list": open(mutations_file,"rb")}

    req = requests.post(URL, files=pdb_to_submit)
    print(req.json())
    return True

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Job submission for single prediction - DynaMut2')
    parser.add_argument('--pdb_file', dest='pdb_file', type=str, help='PDB file', required=False, default=None)
    parser.add_argument('--mutations_list', dest='mutations_list', type=str, help='Mutation list file. One entry per line. Mutations must be separated by a semi-colon (;). E.g.: A E346K;A T118P')

    args = parser.parse_args()
    if args.pdb_file == None:
        print('usage: main.py [-h] [--pdb_file PDB_FILE]')
        print('[--mutations_list MUTATION]')
        print('main.py: error: missing arguments: Please provide --pdb_file')
        sys.exit(1)
    main(args)