From 24f7dd79a66769b7a72970f17aee8f2afac580de Mon Sep 17 00:00:00 2001 From: Navan Chauhan Date: Sun, 12 Jul 2020 21:37:37 +0530 Subject: Added v1 API --- api.py | 40 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 api.py (limited to 'api.py') diff --git a/api.py b/api.py new file mode 100644 index 0000000..436f6fe --- /dev/null +++ b/api.py @@ -0,0 +1,40 @@ +from app import app as flask_app +#app.run(debug=True, host="0.0.0.0", port=8080) + +from fastapi import Body,FastAPI +from fastapi.middleware.wsgi import WSGIMiddleware +from flask import Flask, escape, request +from pydantic import BaseModel + +import mysql.connector as con +mycon = con.connect(host=flask_app.config['DB_HOST'],user=flask_app.config['DB_USER'],password=flask_app.config['DB_PASSWORD'],port=flask_app.config['DB_PORT'],database=flask_app.config['DB_NAME']) +mycursor = mycon.cursor() + +""" +@flask_app.route("/") +def flask_main(): + name = request.args.get("name", "World") + return f"Hello, {escape(name)} from Flask!" +""" + +app = FastAPI() + + +@app.get("/v1") +def API_Version(): + return {"message":"Curie-API v1"} + + + +@app.get("/v1/status/{job_id}") +def get_status(job_id: str): + sqlQuery = 'select id, protein_name, ligand_name, date, description, done from curieweb where id="%s"' % (job_id) + mycursor.execute(sqlQuery) + records = mycursor.fetchall() + if records == []: + return {"message":"Invalid Job ID"} + r = records[0] + return {"job_id":r[0],"Protein Name":r[1],"Ligand Name":r[2],"Submitted On":r[3],"Job Description":r[4],"Job Status":r[5]} + + +app.mount("/", WSGIMiddleware(flask_app)) \ No newline at end of file -- cgit v1.2.3