From 86ae085e2de2b9e9ff78333c1b093d98feba1ad7 Mon Sep 17 00:00:00 2001 From: xnought Date: Thu, 16 Nov 2023 19:00:36 -0800 Subject: [PATCH] fix: format --- backend/src/server.py | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/backend/src/server.py b/backend/src/server.py index 949ea1e9..f5a5bfaf 100644 --- a/backend/src/server.py +++ b/backend/src/server.py @@ -16,13 +16,21 @@ def get_all_entries(): """ with Database() as db: try: - entries_sql = db.execute_return("""SELECT id, name, filePDBAlphaFold, length, mass FROM proteins""") + entries_sql = db.execute_return( + """SELECT id, name, filePDBAlphaFold, length, mass FROM proteins""" + ) log.warn(entries_sql) # if we got a result back if entries_sql is not None: return [ - ProteinEntry(id=str(entry[0]), name=entry[1], filePDBAlphaFold=entry[2], length=entry[3], mass=entry[4]) + ProteinEntry( + id=str(entry[0]), + name=entry[1], + filePDBAlphaFold=entry[2], + length=entry[3], + mass=entry[4], + ) for entry in entries_sql ] except Exception as e: @@ -46,7 +54,13 @@ def get_protein_entry(protein_id: str): # if we got a result back if entry_sql is not None and len(entry_sql) != 0: # return the only entry - return ProteinEntry(id=str(entry_sql[0][0]), name=entry_sql[0][1], filePDBAlphaFold=entry_sql[0][2], length=entry_sql[0][3], mass=entry_sql[0][4]) + return ProteinEntry( + id=str(entry_sql[0][0]), + name=entry_sql[0][1], + filePDBAlphaFold=entry_sql[0][2], + length=entry_sql[0][3], + mass=entry_sql[0][4], + ) except Exception as e: log.error(e)