base db created

.gitignore

@@ -1,3 +1,3 @@
-__pycache__
+__pycache__/
 venv/
 .env

.idea/.gitignore

@@ -0,0 +1,6 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/dataSources.xml

@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="DataSourceManagerImpl" format="xml" multifile-model="true">
+    <data-source source="LOCAL" name="[email protected]" uuid="1721f11c-8c2c-4f79-8cb4-ae21650c04d7">
+      <driver-ref>postgresql</driver-ref>
+      <synchronize>true</synchronize>
+      <jdbc-driver>org.postgresql.Driver</jdbc-driver>
+      <jdbc-url>jdbc:postgresql://ep-solitary-dew-a4ljqdib-pooler.us-east-1.aws.neon.tech:5432/neondb</jdbc-url>
+      <working-dir>$ProjectFileDir$</working-dir>
+    </data-source>
+  </component>
+</project>

.idea/data_source_mapping.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="DataSourcePerFileMappings">
+    <file url="file://$APPLICATION_CONFIG_DIR$/consoles/db/1721f11c-8c2c-4f79-8cb4-ae21650c04d7/console.sql" value="1721f11c-8c2c-4f79-8cb4-ae21650c04d7" />
+  </component>
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/yuvabe-app-backend.iml" filepath="$PROJECT_DIR$/.idea/yuvabe-app-backend.iml" />
+    </modules>
+  </component>
+</project>

.idea/sqldialects.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="SqlDialectMappings">
+    <file url="PROJECT" dialect="PostgreSQL" />
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

.idea/yuvabe-app-backend.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="DBE_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

src/core/database.py

@@ -1,10 +1,21 @@
-from sqlmodel import SQLModel, create_engine
-from dotenv import load_dotenv
-from . import models
 import os
 
+from dotenv import load_dotenv
+from sqlmodel import SQLModel, create_engine
+
+from src.core import models as core_models
+from src.feed import models as feed_models
+
 load_dotenv()
 
-engine = create_engine(os.getenv('DATABASE_URL'),echo=True)
+engine = create_engine(os.getenv("DATABASE_URL"), echo=True)
+
+
+def init_db():
+    SQLModel.metadata.create_all(engine)
+
 
-SQLModel.metadata.create_all(engine)
+if __name__ == "__main__":
+    print("Table creating")
+    init_db()
+    print("Table Created successfully!")

src/core/insert_sample.py

@@ -1,40 +0,0 @@
-from sqlmodel import Session
-from datetime import date
-import uuid
-
-from src.core.database import engine 
-from src.core.models import Users, Roles, Teams 
-
-
-role = Roles(
-    id=uuid.uuid4(),
-    name="User"
-)
-
-team = Teams(
-    id=uuid.uuid4(),
-    name="Health Squad"
-)
-
-user = Users(
-    id=uuid.uuid4(),
-    email_id="[email protected]",
-    password="hashed_password_here",
-    user_name="tilak",
-    dob=date(2000, 5, 20),
-    address="Bangalore, India",
-    role_id=role.id,
-    emotion_trend={"happy": 10, "sad": 2},
-    habit_trend={"exercise": 5, "sleep": 8},
-    profile_picture="https://example.com/image.jpg",
-    post_id=uuid.uuid4()
-)
-
-
-with Session(engine) as session:
-    session.add(role)
-    session.add(team)
-    session.add(user)
-    session.commit()
-
-print("Sample data inserted successfully!")

src/core/models.py

@@ -1,25 +1,71 @@
-from sqlmodel import SQLModel, Field
-from datetime import date
-from sqlalchemy.dialects.postgresql import JSONB
 import uuid
+from datetime import date, datetime
+from enum import Enum
+from typing import List, Optional
+
+from sqlalchemy import CheckConstraint, UniqueConstraint
+from sqlmodel import Field, Relationship, SQLModel
+
+
+class AssetStatus(str, Enum):
+    ACTIVE = "Active"
+    UNAVAILABLE = "Unavailable"
+    ON_REQUEST = "On Request"
+    IN_SERVICE = "In Service"
+
 
 class Users(SQLModel, table=True):
-    id: uuid.UUID | None = Field(default=None, primary_key=True)
-    email_id : str = Field(unique=True)
-    password : str 
-    user_name : str
-    dob : date 
-    address : str
-    role_id : uuid.UUID
-    emotion_trend : dict = Field(sa_type=JSONB)
-    habit_trend : dict = Field(sa_type=JSONB)
-    profile_picture : str
-    post_id : uuid.UUID
-
-class Teams(SQLModel, table = True):
-    id: uuid.UUID | None = Field(default=None, primary_key=True)
-    name : str
-
-class Roles(SQLModel, table = True):
-    id : uuid.UUID | None = Field(default=None , primary_key=True)
-    name : str
+    __tablename__ = "users"
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    email_id: str = Field(unique=True, nullable=False)
+    password: str = Field(nullable=False)
+    user_name: str = Field(nullable=False)
+    dob: Optional[date] = None
+    address: Optional[str] = None
+    profile_picture: Optional[str] = None
+    created_at: datetime = Field(default=datetime.now)
+    asset: List["Assets"] = Relationship(back_populates="user")
+
+
+class Teams(SQLModel, table=True):
+    __tablename__ = "teams"
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    name: str = Field(unique=True, nullable=False)
+
+
+class Roles(SQLModel, table=True):
+    __tablename__ = "roles"
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    name: str = Field(unique=True, nullable=False)
+
+
+class UserTeamsRole(SQLModel, table=True):
+    __tablename__ = "user_teams_role"
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    team_id: uuid.UUID = Field(foreign_key="teams.id", nullable=False)
+    role_id: uuid.UUID = Field(foreign_key="roles.id", nullable=False)
+
+
+class Assets(SQLModel, table=True):
+    __tablename__ = "assets"
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    name: str = Field(nullable=False)
+    type: str = Field(nullable=False)
+    status: AssetStatus = Field(default=AssetStatus.UNAVAILABLE)
+    user: "Users" = Relationship(back_populates="asset")
+
+
+class EmotionLogs(SQLModel, table=True):
+    __tablename__ = "emotion_logs"
+    __table_args__ = (
+        UniqueConstraint("user_id", "log_date"),
+        CheckConstraint("morning_emotion BETWEEN 1 AND 10 or morning_emotion IS NULL"),
+        CheckConstraint("evening_emotion BETWEEN 1 AND 10 or evening_emotion IS NULL"),
+    )
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    morning_emotion: Optional[int] = Field(default=None, ge=1, le=10)
+    evening_emotion: Optional[int] = Field(default=None, ge=1, le=10)
+    log_date: date = Field(default_factory=date.today)

src/feed/models.py

@@ -1,2 +1,49 @@
 import uuid
-import sqlmodel
+from datetime import datetime
+from enum import Enum
+from typing import Optional
+
+from sqlalchemy import UniqueConstraint
+from sqlmodel import Field, SQLModel
+
+
+class PostType(str, Enum):
+    BIRTHDAY = "Birthday"
+    NOTICE = "Notice"
+    BANNER = "Banner"
+    JOB_REQUEST = "Job Request"
+
+
+class PostCategory(str, Enum):
+    TEAM = "Team"
+    GLOBAL = "Global"
+
+
+class Posts(SQLModel, table=True):
+    __tablename__ = "posts"
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    type: PostType = Field(default=PostType.NOTICE)
+    category: PostCategory = Field(default=PostCategory.GLOBAL)
+    caption: Optional[str] = None
+    image: Optional[str] = None
+    created_at: datetime = Field(default=datetime.now, nullable=False)
+    edited_at: datetime = Field(default_factory=datetime.now)
+
+
+class Comments(SQLModel, table=True):
+    __tablename__ = "comments"
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    post_id: uuid.UUID = Field(foreign_key="posts.id", nullable=False)
+    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    comment: str = Field(nullable=False)
+    created_at: datetime = Field(default=datetime.now, nullable=False)
+
+
+class Likes(SQLModel, table=True):
+    __tablename__ = "likes"
+    __table_args__ = (UniqueConstraint("user_id", "post_id"),)
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    post_id: uuid.UUID = Field(foreign_key="posts.id", nullable=False)
+    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    liked_at: datetime = Field(default=datetime.now, nullable=False)