Snippets

Biosig Lab CSM-AB_single_prediction_API

Created by Yoochan Myung

File get.py Added

  • Ignore whitespace
  • Hide word diff
+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)

File post.py Added

  • Ignore whitespace
  • Hide word diff
+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)
  1. 1
  2. 2
HTTPS SSH

You can clone a snippet to your computer for local editing. Learn more.