50 - Foreign Key Relationships & The Models Module

codingforentrepreneurs · codingforentrepreneurs · commit b33e7b780679 · 2024-06-18T12:33:14.000-06:00
diff --git a/full_stack_python/auth/__init__.py b/full_stack_python/auth/__init__.py
@@ -1,6 +0,0 @@
-from .models import UserInfo
-
-
-__all__ = [
-    'UserInfo',
-]
diff --git a/full_stack_python/auth/models.py b/full_stack_python/auth/models.py
diff --git a/full_stack_python/auth/state.py b/full_stack_python/auth/state.py
@@ -3,7 +3,7 @@
 
 import sqlmodel
 
-from .models import UserInfo
+from ..models import UserInfo
 
 
 
diff --git a/full_stack_python/blog/__init__.py b/full_stack_python/blog/__init__.py
@@ -2,14 +2,12 @@
 from .detail import blog_post_detail_page
 from .edit import blog_post_edit_page
 from .list import blog_post_list_page
-from .model import BlogPostModel
 from .state import BlogPostState
 
 __all__= [
     'blog_post_add_page',
     'blog_post_detail_page',
     'blog_post_edit_page',
     'blog_post_list_page',
-    'BlogPostModel',
     'BlogPostState'
 ]
diff --git a/full_stack_python/blog/list.py b/full_stack_python/blog/list.py
@@ -2,10 +2,10 @@
 
 from .. import navigation
 from ..ui.base import base_page
+from ..models import BlogPostModel
+from . import state
 
-from . import state, model
-
-def blog_post_detail_link(child: rx.Component, post: model.BlogPostModel):
+def blog_post_detail_link(child: rx.Component, post: BlogPostModel):
     if post is None:
         return rx.fragment(child)
     post_id = post.id
@@ -18,7 +18,7 @@ def blog_post_detail_link(child: rx.Component, post: model.BlogPostModel):
         href=post_detail_url
     )
 
-def blog_post_list_item(post: model.BlogPostModel):
+def blog_post_list_item(post: BlogPostModel):
     return rx.box(
         blog_post_detail_link(
             rx.heading(post.title),
diff --git a/full_stack_python/blog/model.py b/full_stack_python/blog/model.py
diff --git a/full_stack_python/blog/state.py b/full_stack_python/blog/state.py
@@ -5,7 +5,7 @@
 from sqlmodel import select
 
 from .. import navigation
-from .model import BlogPostModel
+from ..models import BlogPostModel
 
 BLOG_POSTS_ROUTE = navigation.routes.BLOG_POSTS_ROUTE
 if BLOG_POSTS_ROUTE.endswith("/"):
diff --git a/full_stack_python/contact/model.py b/full_stack_python/contact/model.py
diff --git a/full_stack_python/contact/page.py b/full_stack_python/contact/page.py
@@ -1,12 +1,9 @@
 import reflex as rx 
-
-from ..auth.state import SessionState
-from .. import navigation
 from ..ui.base import base_page
+from ..models import ContactEntryModel
+from . import form, state
 
-from . import form, state, model
-
-def contact_entry_list_item(contact: model.ContactEntryModel):
+def contact_entry_list_item(contact: ContactEntryModel):
     return rx.box(
         rx.heading(contact.first_name),
         rx.text("Message:", contact.message),
diff --git a/full_stack_python/contact/state.py b/full_stack_python/contact/state.py
@@ -3,9 +3,10 @@
 import reflex as rx 
 
 from sqlmodel import select
-from .model import ContactEntryModel
+
 
 from ..auth.state import SessionState
+from ..models import ContactEntryModel
 
 class ContactState(SessionState):
     form_data: dict = {}
diff --git a/full_stack_python/models.py b/full_stack_python/models.py
@@ -0,0 +1,89 @@
+from typing import Optional, List
+from datetime import datetime
+import reflex as rx
+from reflex_local_auth.user import LocalUser
+
+import sqlalchemy
+from sqlmodel import Field, Relationship
+
+from . import utils
+
+class UserInfo(rx.Model, table=True):
+    email: str
+    user_id: int = Field(foreign_key='localuser.id')
+    user: LocalUser | None = Relationship() # LocalUser instance
+    posts: List['BlogPostModel'] = Relationship(
+        back_populates='userinfo'
+    )
+    contact_entries: List['ContactEntryModel'] = Relationship(
+        back_populates='userinfo'
+    )
+    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
+    )
+
+
+class BlogPostModel(rx.Model, table=True):
+    # user
+    # id: int -> primary key
+    userinfo_id: int = Field(default=None, foreign_key="userinfo.id")
+    userinfo: Optional['UserInfo'] = Relationship(back_populates="posts")
+    title: str
+    content: str
+    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
+    )
+    publish_active: bool = False
+    publish_date: datetime = Field(
+        default=None,
+        sa_type=sqlalchemy.DateTime(timezone=True),
+        sa_column_kwargs={},
+        nullable=True
+    )
+
+
+
+class ContactEntryModel(rx.Model, table=True):
+    user_id: int | None = None
+    userinfo_id: int = Field(default=None, foreign_key="userinfo.id")
+    userinfo: Optional['UserInfo'] = Relationship(back_populates="contact_entries")
+    first_name: str
+    last_name: str | None = None
+    email: str | None = None # = Field(nullable=True)
+    message: str
+    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
+    )

Original file line number	Diff line number	Diff line change
`@@ -3,7 +3,7 @@`
`3`	`3`
`4`	`4`	`import sqlmodel`
`5`	`5`
`6`		`-from .models import UserInfo`
	`6`	`+from ..models import UserInfo`
`7`	`7`
`8`	`8`
`9`	`9`