Sl 142 criacao de cursos backend

api/src/core/db.py

@@ -24,7 +24,7 @@
 
 async def init_db():
     SQLModel.metadata.create_all(engine)
-
+    #TODO CHECK THIS
     with engine.connect() as conn:
         try:
             conn.execute(
@@ -39,6 +39,54 @@
                     "ADD COLUMN IF NOT EXISTS passing_score INTEGER NOT NULL DEFAULT 80"
                 )
             )
+            conn.execute(
+                text(
+                    "ALTER TABLE user_progress "
+                    "ADD COLUMN IF NOT EXISTS start_date TIMESTAMP WITHOUT TIME ZONE"
+                )
+            )
+            conn.execute(
+                text(
+                    "ALTER TABLE user_progress "
+                    "ADD COLUMN IF NOT EXISTS deadline TIMESTAMP WITHOUT TIME ZONE"
+                )
+            )
+            conn.execute(
+                text(
+                    "ALTER TABLE user_progress "
+                    "ADD COLUMN IF NOT EXISTS overdue BOOLEAN NOT NULL DEFAULT FALSE"
+                )
+            )
+            conn.execute(
+                text(
+                    "ALTER TABLE user_progress "
+                    "ADD COLUMN IF NOT EXISTS cert_valid_days INTEGER NOT NULL DEFAULT 365"
+                )
+            )
+            conn.execute(
+                text(
+                    "ALTER TABLE user_progress "
+                    "ADD COLUMN IF NOT EXISTS cert_expires_at TIMESTAMP WITHOUT TIME ZONE"
+                )
+            )
+            conn.execute(
+                text(
+                    "ALTER TABLE user_progress "
+                    "ADD COLUMN IF NOT EXISTS expired BOOLEAN NOT NULL DEFAULT FALSE"
+                )
+            )
+            conn.execute(
+                text(
+                    "ALTER TABLE user_progress "
+                    "ADD COLUMN IF NOT EXISTS status VARCHAR NOT NULL DEFAULT 'SCHEDULED'"
+                )
+            )
+            conn.execute(
+                text(
+                    "ALTER TABLE user_progress "
+                    "ADD COLUMN IF NOT EXISTS notified_at TIMESTAMP WITHOUT TIME ZONE"
+                )
+            )
             conn.commit()
         except Exception:
             conn.rollback()

api/src/core/mongo.py

@@ -50,6 +50,12 @@ def get_modules_collection() -> AsyncIOMotorCollection:
     return db[settings.MONGODB_COLLECTION_MODULES]
 
 
+def get_courses_collection() -> AsyncIOMotorCollection:
+    client = _get_client()
+    db = client[settings.MONGODB_DB]
+    return db[settings.MONGODB_COLLECTION_COURSES]
+
+
 async def close_mongo_client() -> None:
     """Close the Mongo client when the app shuts down."""
     global _client

api/src/core/settings.py

@@ -35,6 +35,7 @@ class Settings(BaseSettings):
   MONGODB_COLLECTION_CONTENT: str = "content_pieces"
   MONGODB_COLLECTION_CONTENT_FOLDERS: str = "content_folders"
   MONGODB_COLLECTION_MODULES: str = "modules"
+  MONGODB_COLLECTION_COURSES: str = "courses"
   MONGODB_INLINE_FILE_MAX_BYTES: int = 8 * 1024 * 1024
 
   # Garage object storage (S3-compatible)

api/src/main.py

@@ -18,6 +18,8 @@
     sending_profile,
     modules,
     phishing_kit,
+    courses,
+    progress,
 )
 from src.core.db import init_db
 from src.core.mongo import close_mongo_client
@@ -81,3 +83,5 @@ async def health_check():
 app.include_router(sending_profile.router, prefix="/api", tags=["sending-profiles"])
 app.include_router(modules.router, prefix="/api", tags=["modules"])
 app.include_router(phishing_kit.router, prefix="/api", tags=["phishing-kits"])
+app.include_router(courses.router, prefix="/api", tags=["courses"])
+app.include_router(progress.router, prefix="/api", tags=["progress"])

api/src/models/__init__.py

@@ -74,6 +74,7 @@
     ComplianceQuizPayload,
     CompliancePolicyResponse,
     ComplianceQuizResponse,
+    CourseEnrollmentPayload,
 )
 from .module import (
     Difficulty,
@@ -92,6 +93,15 @@
     ModuleOut,
     PaginatedModules,
 )
+from .course import (
+    CourseDifficulty,
+    CourseCreate,
+    CourseUpdate,
+    CoursePatch,
+    CourseOut,
+    PaginatedCourses,
+)
+from .user_progress import UserProgress, AssignmentStatus
 
 
 __all__ = [
@@ -170,6 +180,7 @@
     "ComplianceQuizPayload",
     "CompliancePolicyResponse",
     "ComplianceQuizResponse",
+    "CourseEnrollmentPayload",
     # Learning Module
     "Difficulty",
     "QuestionType",
@@ -186,4 +197,13 @@
     "ModulePatch",
     "ModuleOut",
     "PaginatedModules",
+    # Course
+    "CourseDifficulty",
+    "CourseCreate",
+    "CourseUpdate",
+    "CoursePatch",
+    "CourseOut",
+    "PaginatedCourses",
+    "UserProgress",
+    "AssignmentStatus",
 ]

api/src/models/course/__init__.py

@@ -0,0 +1,17 @@
+from .document import (
+    CourseDifficulty,
+    CourseCreate,
+    CourseUpdate,
+    CoursePatch,
+    CourseOut,
+    PaginatedCourses,
+)
+
+__all__ = [
+    "CourseDifficulty",
+    "CourseCreate",
+    "CourseUpdate",
+    "CoursePatch",
+    "CourseOut",
+    "PaginatedCourses",
+]

api/src/models/course/document.py

@@ -0,0 +1,71 @@
+"""
+Pydantic models for the Course document stored in MongoDB.
+
+Collection  : courses
+Primary key : _id  (ObjectId, serialised as `id: str` in API responses)
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+from enum import StrEnum
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class CourseDifficulty(StrEnum):
+    EASY   = "Easy"
+    MEDIUM = "Medium"
+    HARD   = "Hard"
+
+
+# ── Request models ────────────────────────────────────────────────────────────
+
+class CourseCreate(BaseModel):
+    title:         str             = Field(..., min_length=1, max_length=200)
+    description:   str             = Field(..., max_length=1000)
+    category:      str             = Field(..., min_length=1, max_length=100)
+    difficulty:    CourseDifficulty
+    expected_time: str             = Field(..., min_length=1, max_length=50)
+    cover_image:   Optional[str]   = Field(None, description="content_piece_id of the cover image")
+    modules:       list[str]       = Field(default_factory=list, description="Ordered list of module ObjectId hex strings")
+
+
+class CourseUpdate(CourseCreate):
+    """Full replacement — same shape as create."""
+
+
+class CoursePatch(BaseModel):
+    """Partial update — all fields optional."""
+    title:         Optional[str]             = None
+    description:   Optional[str]             = None
+    category:      Optional[str]             = None
+    difficulty:    Optional[CourseDifficulty] = None
+    expected_time: Optional[str]             = None
+    cover_image:   Optional[str]             = None
+    modules:       Optional[list[str]]       = None
+
+
+# ── Response model ────────────────────────────────────────────────────────────
+
+class CourseOut(BaseModel):
+    id:              str
+    title:           str
+    description:     str
+    category:        str
+    difficulty:      CourseDifficulty
+    expected_time:   str
+    cover_image:     Optional[str]   = None   # content_piece_id — frontend resolves presigned URL
+    modules:         list[str]       = Field(default_factory=list)
+    created_by:      Optional[str]   = None
+    created_at:      datetime
+    updated_at:      datetime
+
+
+class PaginatedCourses(BaseModel):
+    items: list[CourseOut]
+    total: int
+    page:  int
+    limit: int
+    pages: int

api/src/models/email_sending/table.py

@@ -24,7 +24,7 @@ class EmailSendingStatus(StrEnum):
 
 class EmailSending(SQLModel, table=True):
     id: Optional[int] = Field(default=None, primary_key=True)
-    user_id: str = Field(foreign_key="useres.keycloak_id")
+    user_id: str = Field(foreign_key="users.keycloak_id")
     scheduled_date: datetime
     status: EmailSendingStatus = Field(default=EmailSendingStatus.SCHEDULED)
     email_to: str

api/src/models/module/document.py

@@ -59,6 +59,7 @@ class RichContentBlock(BaseModel):
     order:      int                    = Field(..., ge=0)
     media_type: RichMediaType
     url:        str                    = ""
+    content_id: str                    = ""
     caption:    str                    = ""
 
 

api/src/models/org_manager/__init__.py

@@ -1,6 +1,7 @@
 from .schemas import (
     OrgUserCreate,
     OrgGroupCreate,
+    CourseEnrollmentPayload,
     CompliancePolicyPayload,
     QuizQuestionPayload,
     ComplianceQuizPayload,
@@ -11,6 +12,7 @@
 __all__ = [
     "OrgUserCreate",
     "OrgGroupCreate",
+    "CourseEnrollmentPayload",
     "CompliancePolicyPayload",
     "QuizQuestionPayload",
     "ComplianceQuizPayload",

api/src/models/org_manager/schemas.py

@@ -3,6 +3,13 @@
 from pydantic import BaseModel
 
 
+class CourseEnrollmentPayload(BaseModel):
+    course_ids: list[str]
+    start_date: datetime | None = None
+    deadline: datetime | None = None
+    cert_valid_days: float = 365.0
+
+
 class OrgUserCreate(BaseModel):
     username: str
     name: str

api/src/models/user/table.py

@@ -6,7 +6,7 @@
 
 
 class User(SQLModel, table=True):
-    __tablename__ = "useres"
+    __tablename__ = "users"
     keycloak_id: str = Field(primary_key=True)
     email: str
     is_org_manager: bool = Field(default=False, nullable=False)

api/src/models/user_progress/__init__.py

@@ -0,0 +1,3 @@
+from .table import UserProgress, AssignmentStatus
+
+__all__ = ["UserProgress", "AssignmentStatus"]

api/src/models/user_progress/table.py

@@ -0,0 +1,33 @@
+from typing import Optional
+from datetime import datetime
+from sqlmodel import SQLModel, Field, Column
+from sqlalchemy import String, JSON
+from sqlalchemy.dialects import postgresql
+from enum import StrEnum
+
+class AssignmentStatus(StrEnum):
+    SCHEDULED = "SCHEDULED"
+    ACTIVE = "ACTIVE"
+    OVERDUE = "OVERDUE"
+    COMPLETED = "COMPLETED"
+    RENEWAL_REQUIRED = "RENEWAL_REQUIRED"
+
+
+class UserProgress(SQLModel, table=True):
+    __tablename__ = "user_progress"
+    user_id: str = Field(primary_key=True)
+    course_id: str = Field(primary_key=True)
+    progress_data: dict = Field(default={}, sa_column=Column(JSON().with_variant(postgresql.JSONB(), "postgresql")))
+    completed_sections: list[str] = Field(default=[], sa_column=Column(JSON().with_variant(postgresql.ARRAY(String), "postgresql")))
+    total_completed_tasks: int = Field(default=0)
+    is_certified: bool = Field(default=False)
+    start_date: datetime | None = Field(default=None)
+    deadline: datetime | None = Field(default=None)
+    cert_valid_days: float = Field(default=365.0)
+    cert_expires_at: datetime | None = Field(default=None)
+    overdue: bool = Field(default=False)
+    expired: bool = Field(default=False)
+    status: AssignmentStatus = Field(default=AssignmentStatus.SCHEDULED)
+    realm_name: Optional[str] = Field(default=None, index=True)
+    notified_at: Optional[datetime] = Field(default=None)
+    updated_at: datetime = Field(default_factory=datetime.utcnow)