Merge pull request #4 from me0w00f/V2

Add the feature: cover image.

model/crud.py

@@ -128,9 +128,10 @@ def create_admin(db: Session, AdminReg: schemas.UserReg) -> bool | User:
 
 
 def create_post(post_uuid: str, user_uuid: str, posts_title: str, tags: str, category_id: int,
-                comment: bool, db: Session):
+                comment: bool, cover_url: str, db: Session):
     """
     Create a data of a post in the database.
+    :param cover_url: URL of the cover.
     :param post_uuid: Uuid of the post.
     :param user_uuid: Uuid of the user.
     :param posts_title: Title of the post.
@@ -149,6 +150,7 @@ def create_post(post_uuid: str, user_uuid: str, posts_title: str, tags: str, cat
         author_uuid=user_uuid,
         category_id=category_id,
         tags=tags,
+        cover_url=cover_url,
         comment=comment,
         create_time=date,
         update_time=date
@@ -170,9 +172,10 @@ def create_post(post_uuid: str, user_uuid: str, posts_title: str, tags: str, cat
 
 
 def update_post(post_uuid: str, user_uuid: str, posts_title: str, tags: str, category_id: int,
-                comment: bool, db: Session):
+                comment: bool, cover_url: str, db: Session):
     """
     Update a post.
+    :param cover_url: URL of the cover.
     :param post_uuid: Uuid of the post.
     :param user_uuid: Uuid of the user.
     :param posts_title: Title of the post.
@@ -192,6 +195,7 @@ def update_post(post_uuid: str, user_uuid: str, posts_title: str, tags: str, cat
                 'title': posts_title,
                 'tags': tags,
                 'category_id': category_id,
+                'cover_url': cover_url,
                 'comment': comment,
                 'update_time': update_date
             },

model/models.py

@@ -25,6 +25,7 @@ class Posts(Base):
     title = Column(TEXT, nullable=False)
     author_uuid = Column(VARCHAR(36), nullable=False)
     tags = Column(TEXT, nullable=False)
+    cover_url = Column(VARCHAR(255), nullable=True)
     category_id = Column(INT, ForeignKey('categories.id'), nullable=False)
     comment = Column(BOOLEAN, nullable=False)
     create_time = Column(DATETIME, nullable=False)

routers/posts.py

@@ -27,11 +27,13 @@ def create_a_post(posts_title: str = Form(),
                   tags: str = Form(),
                   category_id: int = Form(),
                   comment: bool = Form(),
+                  cover_url: str = Form(),
                   content_file: UploadFile = File(),
                   token: str = Depends(oauth2Scheme),
                   db: Session = Depends(get_db)):
     """
     * Send a post with a file.
+    * :param cover_url: URL of the cover.
     * :param posts_title: Title of the post.
     * :param tags: Tags of the post.
     * :param category_id: Category of the post.
@@ -71,7 +73,7 @@ def create_a_post(posts_title: str = Form(),
         )
 
     if crud.create_post(post_uuid=post_uuid, posts_title=posts_title, tags=tags,
-                        comment=comment, category_id=category_id, user_uuid=author_uuid, db=db):
+                        comment=comment, category_id=category_id, user_uuid=author_uuid, cover_url=cover_url, db=db):
         return {
             "Status": "Success!"
         }
@@ -80,6 +82,7 @@ def create_a_post(posts_title: str = Form(),
 @router_posts.put('/update')
 def update_a_post(post_uuid: str = Form(),
                   posts_title: str = Form(),
+                  cover_url: str = Form(),
                   tags: str = Form(),
                   category_id: int = Form(),
                   comment: bool = Form(),
@@ -88,6 +91,7 @@ def update_a_post(post_uuid: str = Form(),
                   db: Session = Depends(get_db)):
     """
     * Update a post with new information and a file.
+    * :param cover_url: URL of the cover.
     * :param post_uuid: Uuid of the post.
     * :param posts_title: Title of the post.
     * :param tags: Tags of the post.
@@ -132,7 +136,7 @@ def update_a_post(post_uuid: str = Form(),
         )
 
     if crud.update_post(post_uuid=post_uuid, user_uuid=author_uuid, posts_title=posts_title,
-                        tags=tags, category_id=category_id, comment=comment, db=db):
+                        tags=tags, category_id=category_id, comment=comment, cover_url=cover_url, db=db):
         return {
             "Status": "Success!"
         }

routers/resources.py

@@ -47,6 +47,7 @@ def get_posts(page: int, db: Session = Depends(get_db)):
             'title': x.title,
             'author_uuid': x.author_uuid,
             'author_name': author.nick_name,
+            'cover_url': x.cover_url,
             'tags': x.tags,
             'category_id': x.category_id,
             'category': category_name,
@@ -83,6 +84,7 @@ def get_single_post(post_uuid: str, db: Session = Depends(get_db)):
         "post_uuid": db_post_info.post_uuid,
         "title": db_post_info.title,
         "tags": db_post_info.tags,
+        "cover_url": db_post_info.cover_url,
         "author_uuid": db_post_info.author_uuid,
         "author": author.nick_name,
         "category_id": db_post_info.category_id,