home base api completed

src/core/models.py

@@ -23,7 +23,7 @@ class Users(SQLModel, table=True):
     dob: Optional[date] = None
     address: Optional[str] = None
     profile_picture: Optional[str] = None
-    created_at: datetime = Field(default=datetime.now)
+    created_at: datetime = Field(default_factory=datetime.now)
     asset: List["Assets"] = Relationship(back_populates="user")
 
 

src/core/temp_feed_db.py

@@ -0,0 +1,144 @@
+from datetime import date
+
+from sqlmodel import Session
+
+from src.core.database import engine
+from src.core.models import Assets, EmotionLogs, Roles, Teams, Users, UserTeamsRole
+from src.feed.models import Comments, Likes, Posts
+
+
+def seed_users(session: Session):
+    users = [
+        Users(
+            email_id="tilak@example.com",
+            password="hashed_pass1",
+            user_name="Tilak",
+            dob=date(2001, 5, 21),
+            address="Chennai",
+            profile_picture="tilak.png",
+        ),
+        Users(
+            email_id="arun@example.com",
+            password="hashed_pass2",
+            user_name="Arun",
+            dob=date(2000, 8, 15),
+            address="Bangalore",
+            profile_picture="arun.png",
+        ),
+    ]
+    session.add_all(users)
+    session.commit()
+    print("Users added.")
+    return users
+
+
+def seed_teams(session: Session):
+    teams = [
+        Teams(name="Development"),
+        Teams(name="Marketing"),
+        Teams(name="Design"),
+    ]
+    session.add_all(teams)
+    session.commit()
+    print("Teams added.")
+    return teams
+
+
+def seed_roles(session: Session):
+    roles = [
+        Roles(name="Admin"),
+        Roles(name="Member"),
+        Roles(name="Lead"),
+    ]
+    session.add_all(roles)
+    session.commit()
+    print("Roles added.")
+    return roles
+
+
+def seed_user_teams_roles(session: Session, users, teams, roles):
+    mappings = [
+        UserTeamsRole(user_id=users[0].id, team_id=teams[0].id, role_id=roles[0].id),
+        UserTeamsRole(user_id=users[1].id, team_id=teams[1].id, role_id=roles[1].id),
+    ]
+    session.add_all(mappings)
+    session.commit()
+    print("User-Team-Role mappings added.")
+
+
+def seed_assets(session: Session, users):
+    assets = [
+        Assets(user_id=users[0].id, name="MacBook Pro", type="Laptop"),
+        Assets(user_id=users[1].id, name="Dell Monitor", type="Monitor"),
+    ]
+    session.add_all(assets)
+    session.commit()
+    print("Assets added.")
+    return assets
+
+
+def seed_emotion_logs(session: Session, users):
+    logs = [
+        EmotionLogs(user_id=users[0].id, morning_emotion=8, evening_emotion=6),
+        EmotionLogs(user_id=users[1].id, morning_emotion=7, evening_emotion=8),
+    ]
+    session.add_all(logs)
+    session.commit()
+    print("Emotion logs added.")
+
+
+def seed_posts(session: Session, users):
+    posts = [
+        Posts(
+            user_id=users[0].id,
+            caption="New sprint kickoff!",
+            image="sprint.png",
+        ),
+        Posts(
+            user_id=users[1].id,
+            caption="Design updates rolling out soon!",
+            image="design.png",
+        ),
+    ]
+    session.add_all(posts)
+    session.commit()
+    print("Posts added.")
+    return posts
+
+
+def seed_likes(session: Session, users, posts):
+    likes = [
+        Likes(user_id=users[0].id, post_id=posts[1].id),
+        Likes(user_id=users[1].id, post_id=posts[0].id),
+    ]
+    session.add_all(likes)
+    session.commit()
+    print("Likes added.")
+
+
+def seed_comments(session: Session, users, posts):
+    comments = [
+        Comments(user_id=users[0].id, post_id=posts[1].id, comment="Looks great!"),
+        Comments(user_id=users[1].id, post_id=posts[0].id, comment="Can’t wait!"),
+    ]
+    session.add_all(comments)
+    session.commit()
+    print("Comments added.")
+
+
+def run_all_seeds():
+    with Session(engine) as session:
+        users = seed_users(session)
+        teams = seed_teams(session)
+        roles = seed_roles(session)
+        seed_user_teams_roles(session, users, teams, roles)
+        seed_assets(session, users)
+        seed_emotion_logs(session, users)
+        posts = seed_posts(session, users)
+        seed_likes(session, users, posts)
+        seed_comments(session, users, posts)
+        print("All data seeded successfully!")
+
+
+if __name__ == "__main__":
+    run_all_seeds()

src/feed/models.py

@@ -27,7 +27,7 @@ class Posts(SQLModel, table=True):
     category: PostCategory = Field(default=PostCategory.GLOBAL)
     caption: Optional[str] = None
     image: Optional[str] = None
-    created_at: datetime = Field(default=datetime.now, nullable=False)
+    created_at: datetime = Field(default_factory=datetime.now, nullable=False)
     edited_at: datetime = Field(default_factory=datetime.now)
 
 
@@ -37,7 +37,7 @@ class Comments(SQLModel, table=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)
+    created_at: datetime = Field(default_factory=datetime.now, nullable=False)
 
 
 class Likes(SQLModel, table=True):
@@ -46,4 +46,4 @@ class Likes(SQLModel, table=True):
     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)
+    liked_at: datetime = Field(default_factory=datetime.now, nullable=False)

src/home/router.py

@@ -0,0 +1,39 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlmodel import Session
+
+from src.core.database import engine
+
+from .schemas import EmotionLogCreate, EmotionLogResponse, HomeResponse
+from .service import add_or_update_emotion, get_emotions, get_home_data
+
+router = APIRouter(prefix="/home", tags=["Home"])
+
+
+def get_session():
+    with Session(engine) as session:
+        yield session
+
+
+@router.get("/{user_id}", response_model=HomeResponse)
+def fetch_home_data(user_id: str, session: Session = Depends(get_session)):
+    try:
+        return get_home_data(user_id, session)
+    except ValueError as e:
+        raise HTTPException(status_code=404, detail=str(e))
+
+
+@router.post("/emotion", response_model=EmotionLogResponse)
+def create_or_update_emotion(
+    data: EmotionLogCreate, session: Session = Depends(get_session)
+):
+    record = add_or_update_emotion(data, session)
+    return EmotionLogResponse(
+        log_date=record.log_date,
+        morning_emotion=record.morning_emotion,
+        evening_emotion=record.evening_emotion,
+    )
+
+
+@router.get("/emotion/{user_id}", response_model=list[EmotionLogResponse])
+def get_user_emotions(user_id: str, session: Session = Depends(get_session)):
+    return get_emotions(user_id, session)

src/home/schemas.py

@@ -0,0 +1,27 @@
+# pylint: disable=no-name-in-module
+# pylint: disable=no-self-argument
+
+from datetime import date
+from typing import List, Optional
+
+from pydantic import BaseModel
+
+
+class EmotionLogCreate(BaseModel):
+    user_id: str
+    morning_emotion: Optional[int] = None
+    evening_emotion: Optional[int] = None
+    log_date: date
+
+
+class EmotionLogResponse(BaseModel):
+    log_date: date
+    morning_emotion: Optional[int]
+    eveing_emotion: Optional[int]
+
+
+class HomeResponse(BaseModel):
+    user_id: str
+    user_name: str
+    philosophy_text: str
+    recent_emotions: List[EmotionLogResponse]

src/home/service.py

@@ -1,2 +1,85 @@
-from typing import List
-from uuid import UUID
+from datetime import date, timedelta
+
+from sqlmodel import Session, select
+
+from src.core.models import EmotionLogs, Users
+
+from .schemas import EmotionLogCreate, EmotionLogResponse, HomeResponse
+
+PHILOSOPHY_TEXT = "Your mind is your greatest asset — train it daily."
+
+
+def get_home_data(user_id: str, session: Session) -> HomeResponse:
+    # Fetch user info
+    user = session.exec(select(Users).where(Users.id == user_id)).first()
+    if not user:
+        raise ValueError("User not found")
+
+    # Fetch last 7 days of emotion logs
+    seven_days_ago = date.today() - timedelta(days=7)
+    emotion_logs = session.exec(
+        select(EmotionLogs)
+        .where(EmotionLogs.user_id == user_id)
+        .where(EmotionLogs.log_date >= seven_days_ago)
+        .order_by(EmotionLogs.log_date)
+    ).all()
+
+    emotion_responses = [
+        EmotionLogResponse(
+            log_date=log.log_date,
+            morning_emotion=log.morning_emotion,
+            evening_emotion=log.evening_emotion,
+        )
+        for log in emotion_logs
+    ]
+
+    return HomeResponse(
+        user_id=str(user.id),
+        user_name=user.user_name,
+        philosophy_text=PHILOSOPHY_TEXT,
+        recent_emotions=emotion_responses,
+    )
+
+
+def add_or_update_emotion(data: EmotionLogCreate, session: Session):
+    existing_log = session.exec(
+        select(EmotionLogs)
+        .where(EmotionLogs.user_id == data.user_id)
+        .where(EmotionLogs.log_date == data.log_date)
+    ).first()
+
+    if existing_log:
+        # Update existing record
+        if data.morning_emotion is not None:
+            existing_log.morning_emotion = data.morning_emotion
+        if data.evening_emotion is not None:
+            existing_log.evening_emotion = data.evening_emotion
+    else:
+        new_log = EmotionLogs(
+            user_id=data.user_id,
+            morning_emotion=data.morning_emotion,
+            evening_emotion=data.evening_emotion,
+            log_date=data.log_date,
+        )
+        session.add(new_log)
+
+    session.commit()
+    session.refresh(existing_log or new_log)
+    return existing_log or new_log
+
+
+def get_emotions(user_id: str, session: Session):
+    logs = session.exec(
+        select(EmotionLogs)
+        .where(EmotionLogs.user_id == user_id)
+        .order_by(EmotionLogs.log_date.desc())
+    ).all()
+
+    return [
+        EmotionLogResponse(
+            log_date=log.log_date,
+            morning_emotion=log.morning_emotion,
+            evening_emotion=log.evening_emotion,
+        )
+        for log in logs
+    ]

src/main.py

@@ -1,7 +1,12 @@
 from fastapi import FastAPI
 
-app = FastAPI()
+from src.home.router import router as home_router
+
+app = FastAPI(title="Yuvabe App API")
+
+app.include_router(home_router, prefix="/home", tags=["Home"])
+
 
 @app.get("/")
 def root():
-    return {"message": "Hello, FastAPI!"}
+    return {"message": "API is running fine!"}