40 - Extending the Reflex Local Auth User Model

codingforentrepreneurs · codingforentrepreneurs · commit a3ba31a93ed8 · 2024-06-13T17:37:11.000-06:00
diff --git a/full_stack_python/auth/__init__.py b/full_stack_python/auth/__init__.py
@@ -1,5 +1,8 @@
+from .models import UserInfo
 from . import pages
 
+
 __all__ = [
-    'pages'
+    'pages',
+    'UserInfo'
 ]
diff --git a/full_stack_python/auth/models.py b/full_stack_python/auth/models.py
@@ -0,0 +1,28 @@
+from datetime import datetime
+import reflex as rx
+
+import sqlalchemy
+from sqlmodel import Field
+
+from .. import utils
+
+class UserInfo(rx.Model, table=True):
+    email: str
+    user_id: int = Field(foreign_key='localuser.id')
+    created_at: datetime = Field(
+        default_factory=utils.timing.get_utc_now,
+        sa_type=sqlalchemy.DateTime(timezone=True),
+        sa_column_kwargs={
+            'server_default': sqlalchemy.func.now()
+        },
+        nullable=False
+    )
+    updated_at: datetime = Field(
+        default_factory=utils.timing.get_utc_now,
+        sa_type=sqlalchemy.DateTime(timezone=True),
+        sa_column_kwargs={
+            'onupdate': sqlalchemy.func.now(),
+            'server_default': sqlalchemy.func.now()
+        },
+        nullable=False
+    )
