294 lines
11 KiB
Python
294 lines
11 KiB
Python
import calendar
|
||
import random
|
||
from datetime import timedelta
|
||
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.utils._build_image import BuildImage
|
||
|
||
from ..config import g_bIsDebug
|
||
from ..dbService import g_pDBService
|
||
from ..json import g_pJsonManager
|
||
from ..tool import g_pToolManager
|
||
from .database import CSqlManager
|
||
|
||
|
||
class CUserSignDB(CSqlManager):
|
||
@classmethod
|
||
async def initDB(cls):
|
||
# userSignLog 表结构，每条为一次签到事件
|
||
userSignLog = {
|
||
"uid": "TEXT NOT NULL", # 用户ID
|
||
"signDate": "DATE NOT NULL", # 签到日期
|
||
"isSupplement": "TINYINT NOT NULL DEFAULT 0", # 是否补签
|
||
"exp": "INT NOT NULL DEFAULT 0", # 当天签到经验
|
||
"point": "INT NOT NULL DEFAULT 0", # 当天签到金币
|
||
"createdAt": "DATETIME NOT NULL DEFAULT (datetime(CURRENT_TIMESTAMP, 'localtime'))", # 创建时间
|
||
"PRIMARY KEY": "(uid, signDate)",
|
||
}
|
||
|
||
# userSignSummary 表结构，每用户一行用于缓存签到状态
|
||
userSignSummary = {
|
||
"uid": "TEXT PRIMARY KEY NOT NULL", # 用户ID
|
||
"totalSignDays": "INT NOT NULL DEFAULT 0", # 累计签到天数
|
||
"currentMonth": "CHAR(7) NOT NULL DEFAULT ''", # 当前月份（如2025-05）
|
||
"monthSignDays": "INT NOT NULL DEFAULT 0", # 本月签到次数
|
||
"lastSignDate": "DATE DEFAULT NULL", # 上次签到日期
|
||
"continuousDays": "INT NOT NULL DEFAULT 0", # 连续签到天数
|
||
"supplementCount": "INT NOT NULL DEFAULT 0", # 补签次数
|
||
"updatedAt": "DATETIME NOT NULL DEFAULT (datetime(CURRENT_TIMESTAMP, 'localtime'))", # 更新时间
|
||
}
|
||
|
||
await cls.ensureTableSchema("userSignLog", userSignLog)
|
||
await cls.ensureTableSchema("userSignSummary", userSignSummary)
|
||
|
||
@classmethod
|
||
async def getUserSignRewardByDate(cls, uid: str, date: str) -> tuple[int, int]:
|
||
"""根据指定日期获取用户签到随机奖励
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
date (str): 用户签到日期 示例：2025-05-27
|
||
|
||
Returns:
|
||
tuple[int, int]: 经验、金币
|
||
"""
|
||
try:
|
||
async with cls._transaction():
|
||
async with cls.m_pDB.execute(
|
||
"SELECT exp, point FROM userSignLog WHERE uid=? AND signDate=?",
|
||
(uid, date),
|
||
) as cursor:
|
||
row = await cursor.fetchone()
|
||
|
||
if row is None:
|
||
return 0, 0
|
||
|
||
exp = row["exp"]
|
||
point = row["point"]
|
||
|
||
return exp, point
|
||
except Exception as e:
|
||
logger.warning("获取用户签到数据失败", e=e)
|
||
return 0, 0
|
||
|
||
@classmethod
|
||
async def getUserSignCountByDate(cls, uid: str, monthStr: str) -> int:
|
||
"""根据日期查询用户签到总天数
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
monthStr (str): 需要查询的日期 示例: 2025-05
|
||
|
||
Returns:
|
||
int: 查询月总签到天数
|
||
"""
|
||
try:
|
||
sql = "SELECT COUNT(*) FROM userSignLog WHERE uid=? AND signDate LIKE ?"
|
||
param = f"{monthStr}-%"
|
||
async with cls.m_pDB.execute(sql, (uid, param)) as cursor:
|
||
row = await cursor.fetchone()
|
||
return row[0] if row else 0
|
||
except Exception as e:
|
||
logger.warning("统计用户月签到次数失败", e=e)
|
||
return 0
|
||
|
||
@classmethod
|
||
async def hasSigned(cls, uid: str, signDate: str) -> bool:
|
||
"""判断指定日期是否已签到
|
||
|
||
Args:
|
||
uid (int): 用户ID
|
||
signDate (str): 日期字符串 'YYYY-MM-DD'
|
||
|
||
Returns:
|
||
bool: True=已签到，False=未签到
|
||
"""
|
||
try:
|
||
sql = "SELECT 1 FROM userSignLog WHERE uid=? AND signDate=? LIMIT 1"
|
||
async with cls.m_pDB.execute(sql, (uid, signDate)) as cursor:
|
||
row = await cursor.fetchone()
|
||
return row is not None
|
||
except Exception as e:
|
||
logger.warning("查询是否已签到失败", e=e)
|
||
return False
|
||
|
||
@classmethod
|
||
async def sign(cls, uid: str, signDate: str = "") -> int:
|
||
"""签到
|
||
|
||
Args:
|
||
uid (int): 用户ID
|
||
signDate (str): 日期字符串 'YYYY-MM-DD' 不传默认当前系统日期
|
||
|
||
Returns:
|
||
bool: 0: 签到失败 1: 签到成功 2: 重复签到
|
||
"""
|
||
try:
|
||
if not signDate:
|
||
signDate = g_pToolManager.dateTime().date().today().strftime("%Y-%m-%d")
|
||
|
||
if await cls.hasSigned(uid, signDate):
|
||
return 2
|
||
|
||
todayStr = g_pToolManager.dateTime().date().today().strftime("%Y-%m-%d")
|
||
isSupplement = 0 if signDate == todayStr else 1
|
||
|
||
expMax, expMin, pointMax, pointMin = [
|
||
g_pJsonManager.m_pSign.get(key, default)
|
||
for key, default in (
|
||
("exp_max", 50),
|
||
("exp_min", 5),
|
||
("point_max", 2000),
|
||
("point_min", 200),
|
||
)
|
||
]
|
||
|
||
exp = random.randint(expMin, expMax)
|
||
point = random.randint(pointMin, pointMax)
|
||
vipPoint = 0
|
||
|
||
async with cls._transaction():
|
||
await cls.m_pDB.execute(
|
||
"INSERT INTO userSignLog (uid, signDate, isSupplement, exp, point) VALUES (?, ?, ?, ?, ?)",
|
||
(uid, signDate, isSupplement, exp, point),
|
||
)
|
||
|
||
cursor = await cls.m_pDB.execute(
|
||
"SELECT * FROM userSignSummary WHERE uid=?", (uid,)
|
||
)
|
||
row = await cursor.fetchone()
|
||
|
||
currentMonth = signDate[:7]
|
||
if row:
|
||
monthSignDays = (
|
||
row["monthSignDays"] + 1
|
||
if row["currentMonth"] == currentMonth
|
||
else 1
|
||
)
|
||
totalSignDays = row["totalSignDays"]
|
||
lastDate = row["lastSignDate"]
|
||
prevDate = (
|
||
g_pToolManager.dateTime().strptime(signDate, "%Y-%m-%d")
|
||
- timedelta(days=1)
|
||
).strftime("%Y-%m-%d")
|
||
continuousDays = (
|
||
row["continuousDays"] + 1 if lastDate == prevDate else 1
|
||
)
|
||
supplementCount = (
|
||
row["supplementCount"] + 1
|
||
if isSupplement
|
||
else row["supplementCount"]
|
||
)
|
||
await cls.m_pDB.execute(
|
||
"""
|
||
UPDATE userSignSummary
|
||
SET totalSignDays=totalSignDays+1,
|
||
currentMonth=?,
|
||
monthSignDays=?,
|
||
lastSignDate=?,
|
||
continuousDays=?,
|
||
supplementCount=?
|
||
WHERE uid=?
|
||
""",
|
||
(
|
||
currentMonth,
|
||
monthSignDays,
|
||
signDate,
|
||
continuousDays,
|
||
supplementCount,
|
||
uid,
|
||
),
|
||
)
|
||
else:
|
||
totalSignDays = 1
|
||
await cls.m_pDB.execute(
|
||
"""
|
||
INSERT INTO userSignSummary
|
||
(uid, totalSignDays, currentMonth, monthSignDays, lastSignDate, continuousDays, supplementCount)
|
||
VALUES (?, ?, ?, ?, ?, ?, ?)
|
||
""",
|
||
(
|
||
uid,
|
||
1,
|
||
currentMonth,
|
||
1,
|
||
signDate,
|
||
1,
|
||
1 if isSupplement else 0,
|
||
),
|
||
)
|
||
|
||
# 计算累签奖励
|
||
reward = g_pJsonManager.m_pSign["continuou"].get(f"{totalSignDays}", None)
|
||
|
||
if reward:
|
||
point += reward.get("point", 0)
|
||
exp += reward.get("exp", 0)
|
||
vipPoint = reward.get("vipPoint", 0)
|
||
|
||
plant = reward.get("plant", {})
|
||
|
||
if plant:
|
||
for key, value in plant.items():
|
||
await g_pDBService.userSeed.addUserSeedByUid(uid, key, value)
|
||
|
||
if g_bIsDebug:
|
||
exp += 9999
|
||
|
||
# 向数据库更新
|
||
currentExp = await g_pDBService.user.getUserExpByUid(uid)
|
||
await g_pDBService.user.updateUserExpByUid(uid, currentExp + exp)
|
||
|
||
currentPoint = await g_pDBService.user.getUserPointByUid(uid)
|
||
await g_pDBService.user.updateUserPointByUid(uid, currentPoint + point)
|
||
|
||
if vipPoint > 0:
|
||
currentVipPoint = await g_pDBService.user.getUserVipPointByUid(uid)
|
||
await g_pDBService.user.updateUserVipPointByUid(
|
||
uid, currentVipPoint + vipPoint
|
||
)
|
||
|
||
return 1
|
||
except Exception as e:
|
||
logger.warning("执行签到失败", e=e)
|
||
return 0
|
||
|
||
@classmethod
|
||
async def drawSignCalendarImage(cls, uid: str, year: int, month: int):
|
||
# 绘制签到图，自动提取数据库中该用户该月的签到天数
|
||
cellSize = 80
|
||
padding = 40
|
||
titleHeight = 80
|
||
cols = 7
|
||
rows = 6
|
||
width = cellSize * cols + padding * 2
|
||
height = cellSize * rows + padding * 2 + titleHeight
|
||
|
||
img = BuildImage(width, height, color=(255, 255, 255))
|
||
await img.text((padding, 20), f"{year}年{month}月签到表", font_size=36)
|
||
|
||
firstWeekday, totalDays = calendar.monthrange(year, month)
|
||
monthStr = f"{year:04d}-{month:02d}"
|
||
try:
|
||
sql = "SELECT signDate FROM userSignLog WHERE uid=? AND signDate LIKE ?"
|
||
async with cls.m_pDB.execute(sql, (uid, f"{monthStr}-%")) as cursor:
|
||
rows = await cursor.fetchall()
|
||
signedDays = set(int(r[0][-2:]) for r in rows if r[0][-2:].isdigit())
|
||
except Exception as e:
|
||
logger.warning("绘制签到图时数据库查询失败", e=e)
|
||
signedDays = set()
|
||
|
||
for day in range(1, totalDays + 1):
|
||
index = day + firstWeekday - 1
|
||
row = index // cols
|
||
col = index % cols
|
||
x1 = padding + col * cellSize
|
||
y1 = padding + titleHeight + row * cellSize
|
||
x2 = x1 + cellSize - 10
|
||
y2 = y1 + cellSize - 10
|
||
color = (112, 196, 112) if day in signedDays else (220, 220, 220)
|
||
await img.rectangle((x1, y1, x2, y2), fill=color, outline="black", width=2)
|
||
await img.text((x1 + 10, y1 + 10), str(day), font_size=24)
|
||
|
||
return img
|