From 81f2e05a4fdaca067bafe36098cdecfb51d9deda Mon Sep 17 00:00:00 2001 From: Kohulan Date: Mon, 13 Mar 2023 13:52:04 +0100 Subject: [PATCH] fix: all flake8 errors --- app/dependencies.py | 2 +- app/main.py | 2 +- app/routers/chem.py | 20 +++++++++++++------- app/routers/compose.py | 2 +- app/routers/converters.py | 2 +- 5 files changed, 17 insertions(+), 11 deletions(-) diff --git a/app/dependencies.py b/app/dependencies.py index f9344c8..a376df2 100644 --- a/app/dependencies.py +++ b/app/dependencies.py @@ -1 +1 @@ -from fastapi import Header, HTTPException +# from fastapi import Header, HTTPException diff --git a/app/main.py b/app/main.py index ff10ea5..85fef24 100644 --- a/app/main.py +++ b/app/main.py @@ -1,4 +1,4 @@ -from fastapi import Depends, FastAPI +from fastapi import FastAPI # from .config import settings from .routers import converters, chem, compose diff --git a/app/routers/chem.py b/app/routers/chem.py index 60fc259..a496aeb 100644 --- a/app/routers/chem.py +++ b/app/routers/chem.py @@ -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 ( @@ -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: @@ -80,6 +80,7 @@ async def smiles_search(smiles: Optional[str]): ) rows = paginate(curs.fetchall()) return rows +""" @router.post("/standardise") @@ -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)): diff --git a/app/routers/compose.py b/app/routers/compose.py index 3ac8810..c55ee46 100644 --- a/app/routers/compose.py +++ b/app/routers/compose.py @@ -1,4 +1,4 @@ -from fastapi import APIRouter, Depends, HTTPException +from fastapi import APIRouter router = APIRouter( prefix="/compose", diff --git a/app/routers/converters.py b/app/routers/converters.py index 88fffd3..5be16e3 100644 --- a/app/routers/converters.py +++ b/app/routers/converters.py @@ -1,4 +1,4 @@ -from fastapi import APIRouter, Depends, HTTPException +from fastapi import APIRouter router = APIRouter( prefix="/converters",