diff --git a/requirements.txt b/requirements.txt index b4352ef..36b6aad 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,6 @@ fastapi pydantic sqlalchemy -uvicorn[standard] \ No newline at end of file +uvicorn[standard] +psycopg2-binary +fastapi_utils diff --git a/src/enums.py b/src/enums.py index 036aabd..b9e9f43 100644 --- a/src/enums.py +++ b/src/enums.py @@ -1,11 +1,11 @@ -from enum import Enum +from fastapi_utils.enums import StrEnum -class MinigameEnum(str, Enum): +class MinigameEnum(StrEnum): SpellingBee = 'SpellingBee' Hangman = 'Hangman' JustSign = 'JustSign' -class CourseEnum(str, Enum): +class CourseEnum(StrEnum): Fingerspelling = 'Fingerspelling' - Animals = 'Animals' \ No newline at end of file + Animals = 'Animals' diff --git a/src/models.py b/src/models.py index d3aaf93..522af50 100644 --- a/src/models.py +++ b/src/models.py @@ -1,5 +1,6 @@ from sqlalchemy import Boolean, Column, ForeignKey, Integer, String, Float from sqlalchemy.orm import relationship +from enum import Enum from enums import MinigameEnum, CourseEnum from database import Base @@ -32,4 +33,4 @@ class CourseProgress(Base): progress_value = Column(Float, nullable=False) course = Column(Enum(CourseEnum), nullable=False) owner = Column(Integer, ForeignKey("users.user_id")) - \ No newline at end of file +