posts model and schemas, user model updated to include posts relationship

Author: igorbenav

src/app/models/post.py

@@ -0,0 +1,28 @@
+from typing import Optional
+import uuid as uuid_pkg
+from datetime import datetime
+
+from sqlalchemy import String, DateTime, ForeignKey
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from app.core.database import Base
+
+class Post(Base):
+    __tablename__ = "post"
+
+    id: Mapped[int] = mapped_column(
+        "id", autoincrement=True, nullable=False, unique=True, primary_key=True, init=False
+    )
+    created_by_user_id: Mapped[int] = mapped_column(ForeignKey("user.id"))
+    title: Mapped[str] = mapped_column(String(30))
+    text: Mapped[str] = mapped_column(String(63206))
+    media_url: Mapped[str] = mapped_column(String)
+
+    user: Mapped["User"] = relationship(back_populates="posts", lazy="selectin")
+
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime, default_factory=datetime.utcnow
+    )
+    updated_at: Mapped[Optional[datetime]] = mapped_column(default=None)
+    deleted_at: Mapped[Optional[datetime]] = mapped_column(default=None)
+    is_deleted: Mapped[bool] = mapped_column(default=False)

src/app/models/user.py

@@ -1,23 +1,25 @@
-from typing import Optional
+from typing import Optional, List
 import uuid as uuid_pkg
 from datetime import datetime
 
 from sqlalchemy import String, DateTime
-from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from app.core.database import Base
+from app.models.post import Post
 
 class User(Base):
     __tablename__ = "user"
 
     id: Mapped[int] = mapped_column(
         "id", autoincrement=True, nullable=False, unique=True, primary_key=True, init=False
     )
+    
     name: Mapped[str] = mapped_column(String(30))
     username: Mapped[str] = mapped_column(String(20), unique=True, index=True)
     email: Mapped[str] = mapped_column(String(50), unique=True, index=True)
     hashed_password: Mapped[str] = mapped_column(String)
-    
+
     profile_image_url: Mapped[str] = mapped_column(String, default="https://profileimageurl.com")
     uuid: Mapped[uuid_pkg.UUID] = mapped_column(
         default_factory=uuid_pkg.uuid4, primary_key=True, unique=True
@@ -29,3 +31,5 @@ class User(Base):
     deleted_at: Mapped[Optional[datetime]] = mapped_column(default=None)
     is_deleted: Mapped[bool] = mapped_column(default=False)
     is_superuser: Mapped[bool] = mapped_column(default=False)
+
+    posts: Mapped[List[Post]] = relationship(back_populates="user", cascade="all, delete", lazy="selectin", default_factory=list)

src/app/schemas/post.py

@@ -0,0 +1,99 @@
+from typing import Annotated, Optional
+from datetime import datetime
+
+from pydantic import BaseModel, Field, ConfigDict
+
+from app.core.models import UUIDModel, TimestampModel, PersistentDeletion
+
+class PostBase(BaseModel):
+    title: Annotated[
+        str,
+        Field(min_length=2, max_length=30, examples=["This is my post"])
+    ]
+    text: Annotated[
+        str, 
+        Field(min_length=1, max_length=63206, examples=["This is the content of my post."])
+    ]
+    
+
+class Post(TimestampModel, PostBase, UUIDModel, PersistentDeletion):
+    media_url: Annotated[
+        str | None, 
+        Field(
+            pattern=r"^(https?|ftp)://[^\s/$.?#].[^\s]*$",
+            examples=["https://www.postimageurl.com"],
+            default=None
+        ),
+    ]
+    created_by_user_id: int
+
+
+class PostRead(BaseModel):
+    id: int
+    title: Annotated[
+        str,
+        Field(min_length=2, max_length=30, examples=["This is my post"])
+    ]
+    text: Annotated[
+        str, 
+        Field(min_length=1, max_length=63206, examples=["This is the content of my post."])
+    ]
+    media_url: Annotated[
+        str | None, 
+        Field(
+            pattern=r"^(https?|ftp)://[^\s/$.?#].[^\s]*$",
+            examples=["https://www.postimageurl.com"],
+            default=None
+        ),
+    ]
+    created_by_user_id: int
+    created_at: datetime
+
+
+class PostCreate(PostBase):
+    model_config = ConfigDict(extra='forbid')
+    
+    created_by_user_id: int
+    media_url: str | None = None
+
+
+class PostUpdate(PostBase):
+    model_config = ConfigDict(extra='forbid')
+    
+    title: Annotated[
+        str | None,
+        Field(
+            min_length=2, 
+            max_length=30, 
+            examples=["This is my updated post"],
+            default=None
+        )
+    ]
+    text: Annotated[
+        str | None,
+        Field(
+            min_length=1, 
+            max_length=63206, 
+            examples=["This is the updated content of my post."],
+            default=None
+        )
+    ]
+    media_url: Annotated[
+        str | None,
+        Field(
+            pattern=r"^(https?|ftp)://[^\s/$.?#].[^\s]*$",
+            examples=["https://www.postimageurl.com"],
+            default=None
+        )
+    ]
+
+
+class PostUpdateInternal(PostUpdate):
+    updated_at: datetime
+
+
+class PostDelete(BaseModel):
+    model_config = ConfigDict(extra='forbid')
+
+    is_deleted: bool
+    deleted_at: datetime

src/app/schemas/user.py

@@ -8,21 +8,15 @@
 class UserBase(BaseModel):
     name: Annotated[
         str, 
-        Field(
-            min_length=2, max_length=30, examples=["User Userson"]
-        )
+        Field(min_length=2, max_length=30, examples=["User Userson"])
     ]
     username: Annotated[
         str, 
-        Field(
-            min_length=2, max_length=20, pattern=r"^[a-z0-9]+$", examples=["userson"]
-        )
+        Field(min_length=2, max_length=20, pattern=r"^[a-z0-9]+$", examples=["userson"])
     ]
     email: Annotated[
         EmailStr, 
-        Field(
-            examples=["[email protected]"]
-        )
+        Field(examples=["[email protected]"])
     ]
 
 
@@ -39,15 +33,11 @@ class UserRead(BaseModel):
     id: int
     name: Annotated[
         str, 
-        Field(
-            min_length=2, max_length=30, examples=["User Userson"]
-        )
+        Field(min_length=2, max_length=30, examples=["User Userson"])
     ]
     username: Annotated[
         str, 
-        Field(
-            min_length=2, max_length=20, pattern=r"^[a-z0-9]+$", examples=["userson"]
-        )
+        Field(min_length=2, max_length=20, pattern=r"^[a-z0-9]+$", examples=["userson"])
     ]
     profile_image_url: str
 
@@ -57,9 +47,7 @@ class UserCreate(UserBase):
 
     password: Annotated[
         str, 
-        Field(
-            pattern=r"^.{8,}|[0-9]+|[A-Z]+|[a-z]+|[^a-zA-Z0-9]+$", examples=["Str1ngst!"]
-        )
+        Field(pattern=r"^.{8,}|[0-9]+|[A-Z]+|[a-z]+|[^a-zA-Z0-9]+$", examples=["Str1ngst!"])
     ]
 
 

src/migrations/versions/ad940a8c0e9e_.py renamed to src/migrations/versions/211a8b0d0080_.py

@@ -1,8 +1,8 @@
 """empty message
 
-Revision ID: ad940a8c0e9e
+Revision ID: 211a8b0d0080
 Revises: 
-Create Date: 2023-09-28 23:19:52.423224
+Create Date: 2023-10-06 00:46:07.632040
 
 """
 from typing import Sequence, Union
@@ -12,7 +12,7 @@
 
 
 # revision identifiers, used by Alembic.
-revision: str = 'ad940a8c0e9e'
+revision: str = '211a8b0d0080'
 down_revision: Union[str, None] = None
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None