feature: user profile creation task added & automatic user profile creation added

Author: DataRohit

migrations/versions/b2f432857772_user_profiles.py renamed to migrations/versions/2399e36ed221_user_profiles.py

@@ -1,8 +1,8 @@
 """user_profiles
 
-Revision ID: b2f432857772
+Revision ID: 2399e36ed221
 Revises: 2e1f47c5da46
-Create Date: 2024-10-09 22:39:33.774071
+Create Date: 2024-10-09 22:47:22.743476
 
 """
 
@@ -12,7 +12,7 @@
 from alembic import op
 from sqlalchemy.dialects import postgresql
 
-revision: str = "b2f432857772"
+revision: str = "2399e36ed221"
 down_revision: Union[str, None] = "2e1f47c5da46"
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None
@@ -23,8 +23,18 @@ def upgrade() -> None:
         "user_profiles",
         sa.Column("uid", sa.UUID(), nullable=False),
         sa.Column("user_uid", sa.UUID(), nullable=False),
-        sa.Column("bio", sa.VARCHAR(), nullable=False),
-        sa.Column("avatar", sa.VARCHAR(), nullable=False),
+        sa.Column(
+            "bio",
+            sa.VARCHAR(),
+            server_default="Tell the Bookly community a little about yourself! You can mention your favorite genres, what kinds of books you're currently reading or offering, and whether you're open to lending, borrowing, or selling books. Help others get to know your reading style and what you're looking for!",
+            nullable=False,
+        ),
+        sa.Column(
+            "avatar",
+            sa.VARCHAR(),
+            server_default="https://api.dicebear.com/9.x/adventurer-neutral/png?seed=Adrian",
+            nullable=False,
+        ),
         sa.Column("created_at", postgresql.TIMESTAMP(), nullable=False),
         sa.Column("updated_at", postgresql.TIMESTAMP(), nullable=False),
         sa.ForeignKeyConstraint(

src/auth/routes.py

@@ -18,6 +18,7 @@
     UserResetPasswordSchema,
 )
 from .service import PasswordResetLogService, TokenBlackListService, UserService
+from .tasks import create_user_profile_task
 from .utils import (
     decode_url_safe_token,
     generate_password_hash,
@@ -117,6 +118,8 @@ async def activate_user(
         {"first_name": user.first_name},
     )
 
+    await create_user_profile_task(user_uid, session)
+
     await token_blacklist_service.blacklist_token(
         activation_token, datetime.fromtimestamp(expires_at), session
     )

src/auth/service.py

@@ -3,6 +3,8 @@
 from sqlalchemy import delete, select
 from sqlmodel.ext.asyncio.session import AsyncSession
 
+from src.profile.models import UserProfile
+
 from .models import PasswordResetLog, TokenBlacklist, User
 from .schemas import UserCreateSchema
 from .utils import generate_password_hash
@@ -58,6 +60,16 @@ async def update_user(
 
         return user
 
+    async def create_user_profile(self, user_uid: str, session: AsyncSession) -> None:
+        user = await session.get(User, user_uid)
+        profile = UserProfile(
+            user_uid=user.uid,
+            avatar=f"https://api.dicebear.com/9.x/adventurer-neutral/png?seed={user.username}",
+        )
+
+        session.add(profile)
+        await session.commit()
+
 
 class TokenBlackListService:
     async def blacklist_token(

src/auth/tasks.py

@@ -1,9 +1,11 @@
 import asyncio
 
 import celery
+from sqlmodel.ext.asyncio.session import AsyncSession
 
+from pkg.celery_app import celery_app
 from pkg.db import get_session
-from src.auth.service import PasswordResetLogService, TokenBlackListService
+from src.auth.service import PasswordResetLogService, TokenBlackListService, UserService
 
 
 @celery.shared_task
@@ -24,3 +26,9 @@ async def async_clear_password_reset_logs():
             await password_reset_log_service.clear_password_reset_logs(session)
 
     asyncio.run(async_clear_password_reset_logs())
+
+
+@celery_app.task
+async def create_user_profile_task(user_uid: str, session: AsyncSession):
+    user_service = UserService()
+    await user_service.create_user_profile(user_uid, session)

src/profile/models.py

@@ -14,8 +14,20 @@ class UserProfile(SQLModel, table=True):
     user_uid: uuid.UUID = Field(
         sa_column=Column(pg.UUID, ForeignKey("users.uid"), nullable=False)
     )
-    bio: str = Field(sa_column=Column(pg.VARCHAR, nullable=False))
-    avatar: str = Field(sa_column=Column(pg.VARCHAR, nullable=False))
+    bio: str = Field(
+        sa_column=Column(
+            pg.VARCHAR,
+            nullable=False,
+            server_default="Tell the Bookly community a little about yourself! You can mention your favorite genres, what kinds of books you're currently reading or offering, and whether you're open to lending, borrowing, or selling books. Help others get to know your reading style and what you're looking for!",
+        )
+    )
+    avatar: str = Field(
+        sa_column=Column(
+            pg.VARCHAR,
+            nullable=False,
+            server_default="https://api.dicebear.com/9.x/adventurer-neutral/png?seed=Adrian",
+        )
+    )
     created_at: datetime = Field(
         sa_column=Column(pg.TIMESTAMP, nullable=False, default=datetime.now)
     )