Working on membership stuff

app/models/membership_group_member_models.py

@@ -10,6 +10,7 @@ from app.lib_general import log, logging
 from app.models.common_field_schema import base_fields, default_num_bytes
 from app.models.membership_group_models import Membership_Group_Base
 from app.models.membership_member_models import Membership_Member_Base
+from app.models.product_models import Product_Base
 
 
 class Membership_Group_Member_Base(BaseModel):
@@ -18,11 +19,11 @@ class Membership_Group_Member_Base(BaseModel):
 
     id_random: Optional[str] = Field(
         **base_fields['membership_group_member_id_random'],
-        alias='membership_group_member_id_random',
-        default_factory=lambda:secrets.token_urlsafe(default_num_bytes),
+        alias = 'membership_group_member_id_random',
+        default_factory = lambda:secrets.token_urlsafe(default_num_bytes),
     )
     id: Optional[int] = Field(
-        #alias='membership_group_member_id'
+        alias = 'membership_group_member_id'
     )
 
     membership_group_id_random: Optional[str]
@@ -30,6 +31,9 @@ class Membership_Group_Member_Base(BaseModel):
     membership_member_id_random: Optional[str]
     membership_member_id: Optional[int]
 
+    product_id_random: Optional[str]
+    product_id: Optional[int] # The product they purchased to get this membership
+
     flag: Optional[bool]
     flag_message: Optional[str]
 
@@ -41,6 +45,7 @@ class Membership_Group_Member_Base(BaseModel):
     # Including other related objects
     membership_group: Optional[Membership_Group_Base]
     membership_member: Optional[Membership_Member_Base]
+    product: Optional[Product_Base] # The product used to be a part of the group?
 
     _processed_at: datetime.datetime = PrivateAttr(default_factory=datetime.datetime.now)
 
@@ -80,6 +85,15 @@ class Membership_Group_Member_Base(BaseModel):
             return redis_lookup_id_random(record_id_random=values['membership_member_id_random'], table_name='membership_member')
         return None
 
+    @validator('product_id', always=True)
+    def product_id_lookup(cls, v, values, **kwargs):
+        log.setLevel(logging.WARNING)
+        log.debug(locals())
+
+        if values['product_id_random']:
+            return redis_lookup_id_random(record_id_random=values['product_id_random'], table_name='product')
+        return None
+
     class Config:
         underscore_attrs_are_private = True
         fields = base_fields