feat: updated emotion tag

alembic/versions/d9b4df655a55_updated_emotion_tag.py

@@ -0,0 +1,45 @@
+"""updated emotion tag
+
+Revision ID: d9b4df655a55
+Revises: fec3872d7eba
+Create Date: 2025-12-04 14:34:22.373838
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'd9b4df655a55'
+down_revision: Union[str, Sequence[str], None] = 'fec3872d7eba'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    emotion_enum = sa.Enum(
+        'JOYFUL', 'HAPPY', 'CALM', 'NEUTRAL', 'ANXIOUS', 'SAD', 'FRUSTRATED',
+        name='emotion_enum'
+    )
+    emotion_enum.create(op.get_bind(), checkfirst=True)
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('emotion_logs', sa.Column('morning_emotion', sa.Enum('JOYFUL', 'HAPPY', 'CALM', 'NEUTRAL', 'ANXIOUS', 'SAD', 'FRUSTRATED', name='emotion_enum'), nullable=True))
+    op.add_column('emotion_logs', sa.Column('evening_emotion', sa.Enum('JOYFUL', 'HAPPY', 'CALM', 'NEUTRAL', 'ANXIOUS', 'SAD', 'FRUSTRATED', name='emotion_enum'), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('emotion_logs', 'evening_emotion')
+    op.drop_column('emotion_logs', 'morning_emotion')
+    emotion_enum = sa.Enum(
+        'JOYFUL', 'HAPPY', 'CALM', 'NEUTRAL', 'ANXIOUS', 'SAD', 'FRUSTRATED',
+        name='emotion_enum'
+    )
+    emotion_enum.drop(op.get_bind(), checkfirst=True)
+    # ### end Alembic commands ###

src/core/models.py

@@ -7,8 +7,9 @@ from typing import List, Optional
 
 
 from sqlalchemy.dialects.postgresql import UUID
-from sqlalchemy import CheckConstraint, UniqueConstraint, ForeignKey, SAEnum
+from sqlalchemy import CheckConstraint, UniqueConstraint, ForeignKey
 from sqlmodel import Field, Relationship, SQLModel
+from sqlalchemy import Enum as SQLEnum
 
 
 class AssetStatus(str, Enum):
@@ -95,15 +96,15 @@ class Assets(SQLModel, table=True):
     name: str = Field(nullable=False)
     type: str = Field(nullable=False)
     status: AssetStatus = Field(default=AssetStatus.UNAVAILABLE)
-    user: "Users" = Relationship(back_populatealmosts="asset")
+    user: "Users" = Relationship(back_populates="asset")
 
 
 class EmotionLogs(SQLModel, table=True):
     __tablename__ = "emotion_logs"
-    __table_args__ = (
-        UniqueConstraint("user_id", "log_date"),
-    )
+    __table_args__ = (UniqueConstraint("user_id", "log_date"),)
+
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+
     user_id: uuid.UUID = Field(
         sa_column=Column(
             UUID(as_uuid=True),
@@ -111,12 +112,13 @@ class EmotionLogs(SQLModel, table=True):
             nullable=False,
         )
     )
+
     morning_emotion: Optional[Emotion] = Field(
-        default=None,
-        sa_column=Column(SAEnum(Emotion, native_enum=False), nullable=True)
-    )
+    sa_column=Column(SQLEnum(Emotion, name="emotion_enum", native_enum=True), nullable=True)
+)
     evening_emotion: Optional[Emotion] = Field(
-        default=None,
-        sa_column=Column(SAEnum(Emotion, native_enum=False), nullable=True)
-    )
+    sa_column=Column(SQLEnum(Emotion, name="emotion_enum", native_enum=True), nullable=True)
+)
+
     log_date: date = Field(default_factory=date.today)
+

src/home/schemas.py

@@ -1,7 +1,7 @@
 from datetime import date
 from typing import List, Optional
 from pydantic import BaseModel
-from src.core.enums import Emotion
+from src.core.models import Emotion
 
 class EmotionLogCreate(BaseModel):
     user_id: str