import argparse
import requests
import sys

URL = "http://biosig.unimelb.edu.au/csm_ab/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 - CSM-AB')
    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()
    if args.job_id == None:
        print('usage: get.py [-h] [--job_id JOB_ID]')
        print('get.py: error: missing arguments: Please provide --job_id')
        sys.exit(1)

    main(args)

import argparse
import requests
import sys

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

def main(args):
    pdb_file = args.pdb_file
    pdb_accession = args.pdb_accession
    if pdb_accession:
        data = {"pdb_accession":pdb_accession}
        req = requests.post(URL, data=data)
        print(req.json())
    else:
        pdb_to_submit = {"pdb_file": pdb_file}
        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 - CSM-AB')
    parser.add_argument('--pdb_file', dest='pdb_file', type=argparse.FileType('r'), help='PDB file')
    parser.add_argument('--pdb_accession', dest='pdb_accession', type=str, help='Type PDB-ID')

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