Skip to content

Commit

Permalink
fix: all flake8 errors
Browse files Browse the repository at this point in the history
  • Loading branch information
Kohulan committed Mar 13, 2023
1 parent 066d163 commit 81f2e05
Show file tree
Hide file tree
Showing 5 changed files with 17 additions and 11 deletions.
2 changes: 1 addition & 1 deletion app/dependencies.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
from fastapi import Header, HTTPException
# from fastapi import Header, HTTPException
2 changes: 1 addition & 1 deletion app/main.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from fastapi import Depends, FastAPI
from fastapi import FastAPI

# from .config import settings
from .routers import converters, chem, compose
Expand Down
20 changes: 13 additions & 7 deletions app/routers/chem.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,10 @@
from fastapi import Request, APIRouter
from typing import Optional
from rdkit import Chem
import base64
import requests
from urllib.request import urlopen
from urllib.parse import urlsplit
import mimetypes

# from ..database import db
# from fastapi_pagination import Page, add_pagination, paginate
from rdkit.Chem.EnumerateStereoisomers import (
Expand Down Expand Up @@ -69,6 +68,7 @@ async def smiles_stereoisomers(smiles: Optional[str]):
return smilesArray


"""
@router.get("/search/{smiles}")
async def smiles_search(smiles: Optional[str]):
if smiles:
Expand All @@ -80,6 +80,7 @@ async def smiles_search(smiles: Optional[str]):
)
rows = paginate(curs.fetchall())
return rows
"""


@router.post("/standardise")
Expand Down Expand Up @@ -176,23 +177,28 @@ async def extract_chemicalinfo(request: Request):
reference = body["reference"]
split = urlsplit(image_path)
filename = "/tmp/" + split.path.split("/")[-1]
if 'img' in body:
if "img" in body:
imgDataURI = body["img"]
if imgDataURI:
response = urlopen(imgDataURI)
with open(filename, 'wb') as f:
with open(filename, "wb") as f:
f.write(response.file.read())
smiles = predict_SMILES(filename)
os.remove(filename)
return JSONResponse(content={ "reference" : reference, "smiles": smiles.split(".")})
return JSONResponse(
content={"reference": reference, "smiles": smiles.split(".")}
)
else:
response = requests.get(image_path)
if response.status_code == 200:
with open(filename, 'wb') as f:
with open(filename, "wb") as f:
f.write(response.content)
smiles = predict_SMILES(filename)
os.remove(filename)
return JSONResponse(content={ "reference" : reference, "smiles": smiles.split(".")})
return JSONResponse(
content={"reference": reference, "smiles": smiles.split(".")}
)


# @app.get("/molecules/", response_model=List[schemas.Molecule])
# def read_molecules(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
Expand Down
2 changes: 1 addition & 1 deletion app/routers/compose.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from fastapi import APIRouter, Depends, HTTPException
from fastapi import APIRouter

router = APIRouter(
prefix="/compose",
Expand Down
2 changes: 1 addition & 1 deletion app/routers/converters.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from fastapi import APIRouter, Depends, HTTPException
from fastapi import APIRouter

router = APIRouter(
prefix="/converters",
Expand Down

0 comments on commit 81f2e05

Please sign in to comment.