Update Live-streaming code (auto-daily features)

2026-08-15 16:49:17 +08:00
commit 06ff69e945
350 changed files with 62343 additions and 0 deletions
@@ -0,0 +1,77 @@
+import sqlite3
+import tempfile
+import unittest
+from pathlib import Path
+
+from app.danmu_queue import bilibili_gift_cny_values
+from app.stats_store import StatsStore
+
+
+class BilibiliGiftValueTests(unittest.TestCase):
+    def test_gold_coin_conversion_uses_one_thousand_per_cny(self):
+        unit_value, total_value = bilibili_gift_cny_values("gold", 100, 1)
+        self.assertAlmostEqual(unit_value, 0.1)
+        self.assertAlmostEqual(total_value, 0.1)
+
+    def test_multi_quantity_conversion_preserves_total(self):
+        unit_value, total_value = bilibili_gift_cny_values("gold", 5000, 2)
+        self.assertAlmostEqual(unit_value, 2.5)
+        self.assertAlmostEqual(total_value, 5.0)
+
+    def test_silver_coin_has_no_cny_value(self):
+        self.assertEqual(bilibili_gift_cny_values("silver", 5000, 1), (0.0, 0.0))
+
+    def test_historical_repair_is_idempotent_and_rebuilds_aggregate(self):
+        with tempfile.TemporaryDirectory() as temp_dir:
+            path = Path(temp_dir) / "statistics.sqlite3"
+            connection = sqlite3.connect(path)
+            StatsStore._migrate(connection)
+            fields = {
+                "event_id": "gift-1",
+                "platform": "bilibili",
+                "platform_user_id": "123",
+                "occurred_at_utc": "2026-07-28T00:00:00Z",
+                "business_date": "2026-07-28",
+                "gift_id": "1",
+                "gift_name": "灯牌",
+                "quantity": 1,
+                "unit_value": 10.0,
+                "total_value": 10.0,
+                "currency": "CNY",
+                "payload_json": '{"coin_type":"gold","raw_total_coin":100,"value_rule":"gold_battery_10_to_cny_1_v1"}',
+            }
+            StatsStore._insert(connection, "gift_events", fields)
+            StatsStore._aggregate_gift(connection, fields)
+            legacy_fields = dict(fields)
+            legacy_fields.update({
+                "event_id": "gift-legacy",
+                "occurred_at_utc": "2026-07-28T00:01:00Z",
+                "payload_json": '{"coin_type":"gold","raw_total_coin":100,"value_migration":"bilibili_coin_to_cny_v1"}',
+            })
+            StatsStore._insert(connection, "gift_events", legacy_fields)
+            StatsStore._aggregate_gift(connection, legacy_fields)
+            connection.commit()
+
+            first = StatsStore._repair_gift_value_history(connection)
+            second = StatsStore._repair_gift_value_history(connection)
+
+            event = connection.execute(
+                "SELECT unit_value, total_value, json_extract(payload_json, '$.value_rule') "
+                "FROM gift_events WHERE event_id='gift-1'"
+            ).fetchone()
+            aggregate = connection.execute(
+                "SELECT quantity, total_value FROM gift_aggregates"
+            ).fetchone()
+            connection.close()
+
+        self.assertEqual(first, 2)
+        self.assertEqual(second, 0)
+        self.assertAlmostEqual(event[0], 0.1)
+        self.assertAlmostEqual(event[1], 0.1)
+        self.assertEqual(event[2], "bilibili_gold_coin_1000_to_cny_1_v2")
+        self.assertEqual(aggregate[0], 2)
+        self.assertAlmostEqual(aggregate[1], 0.2)
+
+
+if __name__ == "__main__":
+    unittest.main()