abort if question is not found

This commit is contained in:
mystie 2025-03-28 19:51:08 +03:00
parent 1f8979cf58
commit dd27fc8e09

View file

@ -137,14 +137,22 @@ def connectToDb():
return psycopg.connect(f"postgresql://{dbUser}:{dbPass}@{dbHost}/{dbName}", row_factory=dict_row) return psycopg.connect(f"postgresql://{dbUser}:{dbPass}@{dbHost}/{dbName}", row_factory=dict_row)
def getQuestion(question_id: int) -> dict: def getQuestion(question_id: int) -> dict:
try:
conn = connectToDb() conn = connectToDb()
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute("SELECT * FROM questions WHERE id=%s", (question_id,)) cursor.execute("SELECT * FROM questions WHERE id=%s", (question_id,))
question = cursor.fetchone() question = cursor.fetchone()
if not question:
return abort(404)
question['creation_date'] = question['creation_date'].replace(microsecond=0).replace(tzinfo=None) question['creation_date'] = question['creation_date'].replace(microsecond=0).replace(tzinfo=None)
return question
finally:
cursor.close() cursor.close()
conn.close() conn.close()
return question
def getAllQuestions(limit: int = None, offset: int = None) -> dict: def getAllQuestions(limit: int = None, offset: int = None) -> dict:
conn = connectToDb() conn = connectToDb()