Skip to content

Commit

Permalink
feat: linted code, merged mol end point and added extra query paramet…
Browse files Browse the repository at this point in the history
…er to choose generator
  • Loading branch information
CS76 committed Mar 16, 2023
1 parent 81a6d3e commit 32cc977
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 18 deletions.
36 changes: 20 additions & 16 deletions app/routers/chem.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,30 +31,33 @@
responses={404: {"description": "Not found"}},
)


@router.get("/")
async def chem_index():
return {
"module" : "chem",
"message" : "Successful",
"status" : 200
}
return {"module": "chem", "message": "Successful", "status": 200}


@router.get("/mol")
async def smiles_mol(smiles: str):
async def smiles_mol(smiles: str, generator: Optional[str] = "cdk"):
"""
Convert smiles to mol block:
- **smiles**: required (query parameter)
- **generator**: optional (defaults: cdk)
"""
if smiles:
m = Chem.MolFromSmiles(smiles)
return Chem.MolToMolBlock(m)
if generator:
if generator == "cdk":
return getCDKSDGMol(smiles)
else:
m = Chem.MolFromSmiles(smiles)
return Chem.MolToMolBlock(m)
else:
return None



@router.get("/cannonicalsmiles")
async def smiles_mol(smiles: str):
async def smiles_cannonicalise(smiles: str):
"""
Cannonicalise smiles:
Expand All @@ -65,10 +68,10 @@ async def smiles_mol(smiles: str):
return Chem.MolToSmiles(m)
else:
return None


@router.get("/inchi")
async def smiles_mol(smiles: str):
async def smiles_inchi(smiles: str):
"""
Convert smiles to InChI:
Expand All @@ -79,9 +82,10 @@ async def smiles_mol(smiles: str):
return Chem.inchi.MolToInchi(m)
else:
return None



@router.get("/inchikey")
async def smiles_mol(smiles: str):
async def smiles_inchikey(smiles: str):
"""
Convert smiles to InChIKey:
Expand Down Expand Up @@ -174,7 +178,7 @@ async def smiles_iupac(smiles: str):
return iupac


@router.post("/smiles")
@router.get("/smiles")
async def iupac_smiles(iupac: Optional[str], selfies: Optional[str]):
"""
Generate smiles from IUPAC name or selfies:
Expand All @@ -193,7 +197,7 @@ async def iupac_smiles(iupac: Optional[str], selfies: Optional[str]):
async def nplikeliness_score(smiles: str):
"""
Generate natural product likeliness score based on RDKit implementation
- **smiles**: required (query)
"""
if smiles:
Expand Down
4 changes: 2 additions & 2 deletions app/routers/converters.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,5 +9,5 @@


@router.get("/")
async def chem_index():
return {"message": "Hello Converters Router!"}
async def converters_index():
return {"module": "converters", "message": "Successful", "status": 200}

0 comments on commit 32cc977

Please sign in to comment.