56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
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()
|