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