Add Terms of Use acceptance feature for User model

app/models.py

@@ -41,6 +41,9 @@ class User(Model):
     is_2fa_required = fields.BooleanField(default=False)
     is_2fa_activated = fields.BooleanField(default=False)
     is_ergon_validation_required = fields.BooleanField(default=False)
+    # Terms of use
+    is_use_terms_accepted = fields.BooleanField(default=False)
+    use_terms_accepted_at = fields.DatetimeField(null=True)
     # Metadata
     is_active = fields.BooleanField(default=True)
     is_superuser = fields.BooleanField(default=False)

app/routers/frontend.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 import asyncio
+import datetime
 from typing import Annotated, List
 from fastapi import APIRouter, Depends, Request
 from fastapi_limiter.depends import RateLimiter
@@ -45,6 +46,7 @@ async def get_user_info(
         "role": user.role.job_title if user.role else None,
         "email": user.email,
         "username": user.username,
+        "is_use_terms_accepted": user.is_use_terms_accepted,
         "cpf": cpf,
     }
 
@@ -150,6 +152,24 @@ async def get_patient_encounters(
         return []
 
 
+@router_request(
+    method="POST",
+    router=router,
+    path="/user/accept-terms/",
+    response_model=bool,
+)
+async def accept_use_terms(
+    user: Annotated[User, Depends(assert_user_is_active)],
+    request: Request,
+) -> List[Encounter]:
+
+    user.is_use_terms_accepted = True
+    user.use_terms_accepted_at = datetime.datetime.now()
+    await user.save()
+
+    return user
+
+
 @router.get("/patient/filter_tags")
 async def get_filter_tags(_: Annotated[User, Depends(assert_user_is_active)]) -> List[str]:
     return [

migrations/app/35_20241029140029_update.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE "user" ADD "use_terms_accepted_at" TIMESTAMPTZ;
+        ALTER TABLE "user" ADD "is_use_terms_accepted" BOOL NOT NULL  DEFAULT False;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE "user" DROP COLUMN "use_terms_accepted_at";
+        ALTER TABLE "user" DROP COLUMN "is_use_terms_accepted";"""