allnet: basic playhistory

core/data/schema/arcade.py

@@ -1,7 +1,8 @@
 import re
 from typing import List, Optional
+from datetime import datetime
 
-from sqlalchemy import Column, Table, and_, or_
+from sqlalchemy import Column, Table, and_, or_, UniqueConstraint
 from sqlalchemy.dialects.mysql import insert
 from sqlalchemy.engine import Row
 from sqlalchemy.sql import func, select
@@ -98,8 +99,9 @@ billing_credit: Table = Table(
         "machine",
         Integer,
         ForeignKey("machine.id", ondelete="cascade", onupdate="cascade"),
-        nullable=False, unique=True
+        nullable=False
     ),
+    Column("game_id", CHAR(5), nullable=False),
     Column("chute_type", INTEGER, nullable=False),
     Column("service_type", INTEGER, nullable=False),
     Column("operation_type", INTEGER, nullable=False),
@@ -115,6 +117,25 @@ billing_credit: Table = Table(
     Column("coin_count_slot5", INTEGER, nullable=False),
     Column("coin_count_slot6", INTEGER, nullable=False),
     Column("coin_count_slot7", INTEGER, nullable=False),
+    UniqueConstraint("machine", "game_id", name="machine_billing_credit_uk"),
+    mysql_charset="utf8mb4",
+)
+
+billing_playct: Table = Table(
+    "machine_billing_playcount",
+    metadata,
+    Column("id", BIGINT, primary_key=True, nullable=False),
+    Column(
+        "machine",
+        Integer,
+        ForeignKey("machine.id", ondelete="cascade", onupdate="cascade"),
+        nullable=False, unique=True
+    ),
+    Column("game_id", CHAR(5), nullable=False),
+    Column("year", INTEGER, nullable=False),
+    Column("month", INTEGER, nullable=False),
+    Column("playct", BIGINT, nullable=False, server_default="1"),
+    UniqueConstraint("machine", "game_id", "year", "month", name="machine_billing_playcount_uk"),
     mysql_charset="utf8mb4",
 )
 
@@ -413,12 +434,13 @@ class ArcadeData(BaseData):
             return None
         return result.lastrowid
 
-    async def billing_set_credit(self, machine_id: int, chute_type: int, service_type: int, op_mode: int, coin_rate0: int, coin_rate1: int,
+    async def billing_set_credit(self, machine_id: int, game_id: str, chute_type: int, service_type: int, op_mode: int, coin_rate0: int, coin_rate1: int,
         bonus_adder: int, coin_to_credit_rate: int, coin_count_slot0: int, coin_count_slot1: int, coin_count_slot2: int, coin_count_slot3: int,
         coin_count_slot4: int, coin_count_slot5: int, coin_count_slot6: int, coin_count_slot7: int) -> Optional[int]:
         
         sql = insert(billing_credit).values(
             machine=machine_id,
+            game_id=game_id,
             chute_type=chute_type,
             service_type=service_type,
             operation_type=op_mode,
@@ -460,6 +482,36 @@ class ArcadeData(BaseData):
             return None
         return result.lastrowid
 
+    async def billing_get_credit(self, machine_id: int, game_id: str) -> Optional[Row]:
+        result = await self.execute(billing_credit.select(billing_credit.c.machine == machine_id))
+        if result:
+            return result.fetchone()
+
+    async def billing_add_playcount(self, machine_id: int, game_id: str, playct: int = 1) -> None:
+        now = datetime.now()
+        sql = insert(billing_playct).values(
+            machine=machine_id,
+            game_id=game_id,
+            year=now.year,
+            month=now.month,
+            playct=playct
+        )
+
+        conflict = sql.on_duplicate_key_update(playct=billing_playct.c.playct + playct)
+        result = await self.execute(conflict)
+
+        if result is None:
+            self.logger.error(f"Failed to add playcount for machine {machine_id} running {game_id}")
+    
+    async def billing_get_playcount_3mo(self, machine_id: int, game_id: str) -> Optional[List[Row]]:
+        result = await self.execute(billing_playct.select(and_(
+            billing_playct.c.machine == machine_id,
+            billing_playct.c.game_id == game_id
+        )).order_by(billing_playct.c.year.desc(), billing_playct.c.month.desc()).limit(3))
+
+        if result is not None:
+            return result.fetchall()
+
     def format_serial(
         self, platform_code: str, platform_rev: int, serial_letter: str, serial_num: int, append: int, dash: bool = False
     ) -> str: