gold_readbag and russian

models/redbag_user.py

@@ -0,0 +1,65 @@
+
+from services.db_context import db
+
+
+class RedbagUser(db.Model):
+    __tablename__ = 'redbag_users'
+
+    id = db.Column(db.Integer(), primary_key=True)
+    user_qq = db.Column(db.BigInteger(), nullable=False)
+    group_id = db.Column(db.BigInteger(), nullable=False)
+    send_redbag_count = db.Column(db.Integer(), default=0)
+    get_redbag_count = db.Column(db.Integer(), default=0)
+    spend_gold = db.Column(db.Integer(), default=0)
+    get_gold = db.Column(db.Integer(), default=0)
+
+    _idx1 = db.Index('redbag_group_users_idx1', 'user_qq', 'group_id', unique=True)
+
+    @classmethod
+    async def add_redbag_data(cls, user_qq: int, group_id: int, itype: str, money: int):
+        query = cls.query.where(
+            (cls.user_qq == user_qq) & (cls.group_id == group_id)
+        )
+        user = await query.with_for_update().gino.first() or await cls.create(
+                    user_qq=user_qq,
+                    group_id=group_id,
+                )
+        if itype == 'get':
+            await user.update(
+                get_redbag_count=user.get_redbag_count + 1,
+                get_gold=user.get_gold + money,
+            ).apply()
+        else:
+            await user.update(
+                send_redbag_count=user.send_redbag_count + 1,
+                spend_gold=user.spend_gold + money,
+            ).apply()
+
+    @classmethod
+    async def ensure(cls, user_qq: int, group_id: int) -> bool:
+        query = cls.query.where(
+            (cls.user_qq == user_qq) & (cls.group_id == group_id)
+        )
+        user = await query.gino.first() or await cls.create(
+            user_qq=user_qq,
+            group_id=group_id,
+        )
+        return user
+
+    @classmethod
+    async def get_user_all(cls, group_id: int = None) -> list:
+        if not group_id:
+            query = await cls.query.gino.all()
+        else:
+            query = await cls.query.where(
+                (cls.group_id == group_id)
+            ).gino.all()
+        return query
+
+
+
+
+
+
+
+