From 8d1db9a1c081e6b76c286484654b19b1019689ec Mon Sep 17 00:00:00 2001
From: Danny Liu <dannyjialiliu@gmail.com>
Date: Wed, 3 Apr 2024 22:36:27 -0700
Subject: [PATCH] feat: api endpoint to receive profile_image_uirl on signup

---
 backend/apps/web/models/auths.py  | 5 +++--
 backend/apps/web/models/users.py  | 6 +++---
 backend/apps/web/routers/auths.py | 2 +-
 3 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/backend/apps/web/models/auths.py b/backend/apps/web/models/auths.py
index 75637700d..5b262d631 100644
--- a/backend/apps/web/models/auths.py
+++ b/backend/apps/web/models/auths.py
@@ -82,6 +82,7 @@ class SignupForm(BaseModel):
     name: str
     email: str
     password: str
+    profile_image_url: str
 
 
 class AuthsTable:
@@ -90,7 +91,7 @@ class AuthsTable:
         self.db.create_tables([Auth])
 
     def insert_new_auth(
-        self, email: str, password: str, name: str, role: str = "pending"
+        self, email: str, password: str, name: str, profile_image_url: str, role: str = "pending"
     ) -> Optional[UserModel]:
         log.info("insert_new_auth")
 
@@ -101,7 +102,7 @@ class AuthsTable:
         )
         result = Auth.create(**auth.model_dump())
 
-        user = Users.insert_new_user(id, name, email, role)
+        user = Users.insert_new_user(id, name, email, profile_image_url, role)
 
         if result and user:
             return user
diff --git a/backend/apps/web/models/users.py b/backend/apps/web/models/users.py
index 255c701df..62613581d 100644
--- a/backend/apps/web/models/users.py
+++ b/backend/apps/web/models/users.py
@@ -30,7 +30,7 @@ class UserModel(BaseModel):
     name: str
     email: str
     role: str = "pending"
-    profile_image_url: str = "/user.png"
+    profile_image_url: str
     timestamp: int  # timestamp in epoch
 
 
@@ -57,7 +57,7 @@ class UsersTable:
         self.db.create_tables([User])
 
     def insert_new_user(
-        self, id: str, name: str, email: str, role: str = "pending"
+        self, id: str, name: str, email: str, profile_image_url: str, role: str = "pending",
     ) -> Optional[UserModel]:
         user = UserModel(
             **{
@@ -65,7 +65,7 @@ class UsersTable:
                 "name": name,
                 "email": email,
                 "role": role,
-                "profile_image_url": "/user.png",
+                "profile_image_url": profile_image_url,
                 "timestamp": int(time.time()),
             }
         )
diff --git a/backend/apps/web/routers/auths.py b/backend/apps/web/routers/auths.py
index d881ec746..a1931b009 100644
--- a/backend/apps/web/routers/auths.py
+++ b/backend/apps/web/routers/auths.py
@@ -146,7 +146,7 @@ async def signup(request: Request, form_data: SignupForm):
         )
         hashed = get_password_hash(form_data.password)
         user = Auths.insert_new_auth(
-            form_data.email.lower(), hashed, form_data.name, role
+            form_data.email.lower(), hashed, form_data.name, form_data.profile_image_url, role
         )
 
         if user: