rewrite: Restructure to include GUI.

2022-12-23 06:44:32 +02:00
parent 2b93354248
commit f328324747
224 changed files with 8 additions and 0 deletions
--- a/src/modules/shop/data.py
+++ b/src/modules/shop/data.py
@@ -0,0 +1,154 @@
+from enum import Enum
+from cachetools import TTLCache
+
+
+from data import Registry, RowModel, RegisterEnum, WeakCache
+from data.columns import Integer, String, Bool, Timestamp, Column
+
+
+class ShopItemType(Enum):
+    """
+    Schema
+    ------
+    CREATE TYPE ShopItemType AS ENUM (
+        'COLOUR_ROLE'
+    );
+    """
+    COLOUR = 'COLOUR_ROLE',
+
+
+class ShopData(Registry, name='shop'):
+    _ShopItemType = RegisterEnum(ShopItemType)
+
+    class ShopItem(RowModel):
+        """
+        Schema
+        ------
+        CREATE TABLE shop_items(
+            itemid SERIAL PRIMARY KEY,
+            guildid BIGINT NOT NULL,
+            item_type ShopItemType NOT NULL,
+            price INTEGER NOT NULL,
+            purchasable BOOLEAN DEFAULT TRUE,
+            deleted BOOLEAN DEFAULT FALSE,
+            created_at TIMESTAMP DEFAULT (now() at time zone 'utc')
+        );
+        CREATE INDEX guild_shop_items ON shop_items (guildid);
+        """
+        _tablename_ = 'shop_items'
+
+        itemid = Integer(primary=True)
+        guildid = Integer()
+        item_type: Column[ShopItemType] = Column()
+        price = Integer()
+        purchasable = Bool()
+        deleted = Bool()
+        created_at = Timestamp()
+
+    class ColourRole(RowModel):
+        """
+        Schema
+        ------
+        CREATE TABLE shop_items_colour_roles(
+            itemid INTEGER PRIMARY KEY REFERENCES shop_items(itemid) ON DELETE CASCADE,
+            roleid BIGINT NOT NULL
+        );
+        """
+        _tablename_ = 'shop_items_colour_roles'
+
+        itemid = Integer(primary=True)
+        roleid = Integer()
+
+    class ShopItemInfo(RowModel):
+        """
+        A view joining the shop item sub-type information,
+        and including the guild id.
+
+        Schema
+        ------
+        CREATE VIEW shop_item_info AS
+          SELECT
+            *,
+            row_number() OVER (PARTITION BY guildid ORDER BY itemid) AS guild_itemid
+          FROM
+            shop_items
+          LEFT JOIN shop_items_colour_roles USING (itemid)
+          ORDER BY itemid ASC;
+        """
+        _tablename_ = 'shop_item_info'
+        _readonly_ = True
+
+        itemid = Integer(primary=True)
+        guild_itemid = Integer()
+        guildid = Integer()
+        item_type: Column[ShopItemType] = Column()
+        price = Integer()
+        purchasable = Bool()
+        deleted = Bool()
+        created_at = Timestamp()
+        roleid = Integer()
+
+    class MemberInventory(RowModel):
+        """
+        Schema
+        ------
+        CREATE TABLE member_inventory(
+            inventoryid SERIAL PRIMARY KEY,
+            guildid BIGINT NOT NULL,
+            userid BIGINT NOT NULL,
+            transactionid INTEGER REFERENCES coin_transactions(transactionid) ON DELETE SET NULL,
+            itemid INTEGER NOT NULL REFERENCES shop_items(itemid) ON DELETE CASCADE
+        );
+        CREATE INDEX member_inventory_members ON member_inventory(guildid, userid);
+        """
+        _tablename_ = 'member_inventory'
+
+        inventoryid = Integer(primary=True)
+        guildid = Integer()
+        userid = Integer()
+        transactionid = Integer()
+        itemid = Integer()
+
+    class MemberInventoryInfo(RowModel):
+        """
+        Composite view joining the member inventory with shop item information.
+
+        Schema
+        ------
+        CREATE VIEW member_inventory_info AS
+          SELECT
+            inv.inventoryid AS inventoryid,
+            inv.guildid AS guildid,
+            inv.userid AS userid,
+            inv.transactionid AS transactionid,
+            items.itemid AS itemid,
+            items.item_type AS item_type,
+            items.price AS price,
+            items.purchasable AS purchasable,
+            items.deleted AS deleted
+          FROM
+            member_inventory inv
+          LEFT JOIN shop_item_info items USING (itemid)
+          ORDER BY itemid ASC;
+        """
+        _tablename_ = 'member_inventory_info'
+        _readonly_ = True
+
+        inventoryid = Integer(primary=True)
+        guildid = Integer()
+        userid = Integer()
+        transactionid = Integer()
+        itemid = Integer()
+        guild_itemid = Integer()
+        item_type: Column[ShopItemType] = Column()
+        price = Integer()
+        purchasable = Bool()
+        deleted = Bool()
+        roleid = Integer()
+
+        @classmethod
+        async def fetch_inventory_info(cls, guildid, userid) -> list['ShopData.MemberInventoryInfo']:
+            """
+            Fetch the information rows for the given members inventory.
+            """
+            return await cls.fetch_where(guildid=guildid, userid=userid)