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)
def getQuestion(question_id: int) -> dict:
conn = connectToDb()
cursor = conn.cursor()
cursor.execute("SELECT * FROM questions WHERE id=%s", (question_id,))
question = cursor.fetchone()
question['creation_date'] = question['creation_date'].replace(microsecond=0).replace(tzinfo=None)
cursor.close()
conn.close()
return question
try:
conn = connectToDb()
cursor = conn.cursor()
cursor.execute("SELECT * FROM questions WHERE id=%s", (question_id,))
question = cursor.fetchone()
if not question:
return abort(404)
question['creation_date'] = question['creation_date'].replace(microsecond=0).replace(tzinfo=None)
return question
finally:
cursor.close()
conn.close()
def getAllQuestions(limit: int = None, offset: int = None) -> dict:
conn = connectToDb()