import time
import aiosqlite
from database import get_conn


class WaterUsageModel:
    @classmethod
    async def create_table(cls):
        async with get_conn() as conn:
            conn.row_factory = aiosqlite.Row
            await conn.execute(
                """
            CREATE TABLE IF NOT EXISTS water_usage (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                room_id INTEGER,
                room_name TEXT,
                left_water REAL,
                left_money REAL,
                left_free_water REAL,
                left_free_money REAL,
                water_price REAL,
                mon_time INTEGER,
                created_at INTEGER UNIQUE
            )
            """
            )
            await conn.commit()

    @classmethod
    async def insert(cls, data: dict):
        print(data)
        async with get_conn() as conn:
            conn.row_factory = aiosqlite.Row
            await conn.execute(
                """
            INSERT OR IGNORE INTO water_usage (
                room_id, room_name,
                left_water, left_money,
                left_free_water, left_free_money,
                water_price, mon_time, created_at
            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
            """,
                (
                    data["roomId"],
                    data["roomName"],
                    data["leftWater"],
                    data["leftMoney"],
                    data["leftFreeWater"],
                    data["leftFreeMoney"],
                    data["coldWaterPrice"],
                    data["monTime"],
                    int(time.time() * 1000),
                ),
            )
            await conn.commit()