[FEATURE][WS-1698] add UserServiceStatus to user schema

MoBagel · Oct 11, 2022 · 8e69ad8 · 8e69ad8
1 parent 5048436
commit 8e69ad8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/awesome_sso/service/user/schema.py b/awesome_sso/service/user/schema.py
@@ -114,6 +114,11 @@ class Language(str, Enum):
     ZH_TW = "zh-TW"
 
 
+class UserServiceStatus(str, Enum):
+    ACTIVE = "active"
+    TRIAL_ENDED = "trial_ended"
+
+
 class AwesomeUser(Document):
     name: str
     email: EmailStr

diff --git a/awesome_sso/service/user/sync_user.py b/awesome_sso/service/user/sync_user.py
@@ -2,7 +2,7 @@
 from awesome_exception.exceptions import InternalServerError, Unauthorized
 
 from awesome_sso.service.settings import Settings
-from awesome_sso.service.user.schema import AwesomeUserType
+from awesome_sso.service.user.schema import AwesomeUserType, UserServiceStatus
 
 
 async def sync_user(user: AwesomeUserType):
@@ -35,7 +35,9 @@ async def sync_user(user: AwesomeUserType):
             {
                 "service_name": service["service_name"],
                 "external_domain": service["external_domain"],
-                "status": service["status"] if "status" in service else "trial_ended"
+                "status": service["status"]
+                if "status" in service
+                else UserServiceStatus.TRIAL_ENDED,
             }
         )
     config_values["services"] = services_info