refactor code

models/sigin_group_user.py

@@ -4,7 +4,7 @@ from services.db_context import db
 
 
 class SignGroupUser(db.Model):
-    __tablename__ = 'sign_group_users'
+    __tablename__ = "sign_group_users"
 
     id = db.Column(db.Integer(), primary_key=True)
     user_qq = db.Column(db.BigInteger(), nullable=False)
@@ -13,13 +13,19 @@ class SignGroupUser(db.Model):
     checkin_count = db.Column(db.Integer(), nullable=False)
     checkin_time_last = db.Column(db.DateTime(timezone=True), nullable=False)
     impression = db.Column(db.Numeric(scale=3, asdecimal=False), nullable=False)
-    add_probability = db.Column(db.Numeric(scale=3, asdecimal=False), nullable=False, default=0)
-    specify_probability = db.Column(db.Numeric(scale=3, asdecimal=False), nullable=False, default=0)
+    add_probability = db.Column(
+        db.Numeric(scale=3, asdecimal=False), nullable=False, default=0
+    )
+    specify_probability = db.Column(
+        db.Numeric(scale=3, asdecimal=False), nullable=False, default=0
+    )
 
-    _idx1 = db.Index('sign_group_users_idx1', 'user_qq', 'belonging_group', unique=True)
+    _idx1 = db.Index("sign_group_users_idx1", "user_qq", "belonging_group", unique=True)
 
     @classmethod
-    async def ensure(cls, user_qq: int, belonging_group: int, for_update: bool = False) -> 'SignGroupUser':
+    async def ensure(
+        cls, user_qq: int, belonging_group: int, for_update: bool = False
+    ) -> "SignGroupUser":
         query = cls.query.where(
             (cls.user_qq == user_qq) & (cls.belonging_group == belonging_group)
         )
@@ -35,14 +41,18 @@ class SignGroupUser(db.Model):
         )
 
     @classmethod
-    async def query_impression_all(cls, belonging_group: int) -> 'list,list':
+    async def get_all_impression(cls, belonging_group: int) -> "list, list":
+        """
+        说明：
+            获取该群所有用户 id 及对应 好感度
+        参数：
+            :param belonging_group: 群号
+        """
         impression_list = []
         user_qq_list = []
         user_group = []
         if belonging_group:
-            query = cls.query.where(
-                (cls.belonging_group == belonging_group)
-            )
+            query = cls.query.where(cls.belonging_group == belonging_group)
         else:
             query = cls.query
         for user in await query.gino.all():
@@ -50,4 +60,3 @@ class SignGroupUser(db.Model):
             user_qq_list.append(user.user_qq)
             user_group.append(user.belonging_group)
         return user_qq_list, impression_list, user_group
-