import argparse
import requests
import sys

URL = "http://biosig.unimelb.edu.au/mmcsm_na/api/prediction_single"

def main(args):
    job_id = args.job_id

    params = {
        "job_id": job_id,
    }

    req = requests.get(URL, data=params)
    print(req.json())
    return True

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Retrieve job results for single prediction - DynaMut2')
    parser.add_argument('--job_id', dest='job_id', type=str, help='Job identifier code generated upon submission', required=False, default=None)

    args = parser.parse_args()
    main(args)

import argparse
import requests
import sys

URL = "http://biosig.unimelb.edu.au/mmcsm_na/api/prediction_single"

def main(args):
    pdb_file = args.pdb_file
    pdb_accession = args.pdb_accession
    mutation_list = args.mutation_list
    na_type = args.na_type

    pdb_to_submit = {"pdb_file": pdb_file}
    params = {
        "pdb_accession": pdb_accession,
        "mutation_list": mutation_list,
        "na_type": na_type
    }

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

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Job submission for single prediction - mmCSM-NA')
    parser.add_argument('--pdb_file', dest='pdb_file', type=argparse.FileType('r'), help='PDB file', required=False, default=None)
    parser.add_argument('--pdb_accession', dest='pdb_accession', type=str, help='4-digit PDB code', required=False, default=None)
    parser.add_argument('--mutation_list', dest='mutation_list', type=str, help='Mutation code AA_FROM RESIDUE_NUMBER AA_TO - e.g.: E346K')
    parser.add_argument('--na_type', dest='na_type', type=str, help='Nucliec acid type')

    args = parser.parse_args()
    if args.pdb_accession == None and args.pdb_file == None:
        print('usage: main.py [-h] [--pdb_file PDB_FILE] [--pdb_accession PDB_ACCESSION]')
        print('[--mutation_list MUTATION_LIST] [--na_type NA_TYPE]')
        print('main.py: error: missing arguments: Please provide --pdb_file or --pdb_accession')
        sys.exit(1)
    main(args)