refactor code

models/group_info.py

@@ -1,30 +1,47 @@
-
 from services.db_context import db
+from typing import List
 
 
 class GroupInfo(db.Model):
-    __tablename__ = 'group_info'
+    __tablename__ = "group_info"
 
     group_id = db.Column(db.BigInteger(), nullable=False, primary_key=True)
     group_name = db.Column(db.Unicode(), nullable=False, default="")
     max_member_count = db.Column(db.Integer(), nullable=False, default=0)
     member_count = db.Column(db.Integer(), nullable=False, default=0)
 
-    _idx1 = db.Index('group_info_idx1', 'group_id', unique=True)
+    _idx1 = db.Index("group_info_idx1", "group_id", unique=True)
 
     @classmethod
-    async def get_group_info(cls, group_id: int) -> 'GroupInfo':
-        query = cls.query.where(
-            cls.group_id == group_id
-        )
+    async def get_group_info(cls, group_id: int) -> "GroupInfo":
+        """
+        说明：
+            获取群信息
+        参数：
+            :param group_id: 群号
+        """
+        query = cls.query.where(cls.group_id == group_id)
         return await query.gino.first()
 
     @classmethod
-    async def add_group_info(cls, group_id: int, group_name: str, max_member_count: int, member_count: int) -> bool:
+    async def add_group_info(
+        cls, group_id: int, group_name: str, max_member_count: int, member_count: int
+    ) -> bool:
+        """
+        说明：
+            添加群信息
+        参数：
+            :param group_id: 群号
+            :param group_name: 群名称
+            :param max_member_count: 群员最大数量
+            :param member_count: 群员数量
+        """
         try:
-            group = await cls.query.where(
-                cls.group_id == group_id
-            ).with_for_update().gino.first()
+            group = (
+                await cls.query.where(cls.group_id == group_id)
+                .with_for_update()
+                .gino.first()
+            )
             if group:
                 await cls.update(
                     group_id=group_id,
@@ -45,12 +62,23 @@ class GroupInfo(db.Model):
 
     @classmethod
     async def delete_group_info(cls, group_id: int) -> bool:
+        """
+        说明：
+            删除群信息
+        参数：
+            :param group_id: 群号
+        """
         try:
-            await cls.delete.where(
-                cls.group_id == group_id
-            ).gino.status()
+            await cls.delete.where(cls.group_id == group_id).gino.status()
             return True
         except Exception:
             return False
 
-
+    @classmethod
+    async def get_all_group(cls) -> List["GroupInfo"]:
+        """
+        说明：
+            获取所有群对象
+        """
+        query = await cls.query.gino.all()
+        return query