🎨 使用Python包管理和模块访问

🎨 重构数据库结构, 使用集中式服务定位器与数据库交互
🎨 更改部分代码逻辑, 减小内存消耗

utils/darw/drawFarm.py

@@ -0,0 +1,426 @@
+from zhenxun.configs.config import Config
+from zhenxun.services.log import logger
+from zhenxun.utils._build_image import BuildImage
+from zhenxun.utils.image_utils import ImageTemplate
+from zhenxun.utils.platform import PlatformUtils
+
+from ...core.dbService import g_pDBService
+from .. import config, getJsonManager, getToolManager
+
+
+class CDarwFarm:
+    @classmethod
+    async def drawFarmByUid(cls, uid: str) -> bytes:
+        """绘制用户农场
+
+        Args:
+            uid (str): 用户UID
+
+        Returns:
+            bytes: 返回绘制结果
+        """
+        img, soilSize, soilPos, grass = await cls.createBaseCanvas()
+
+        player = await getToolManager().getPlayerByUid(uid)
+        if not player:
+            return img.pic2bytes()
+
+        soilUnlock = int(player.user.get("soil", 3))
+
+        isFirstExpansion = True  # 首次添加扩建图片
+        isFirstRipe = True
+
+        for index in range(0, 30):
+            x = soilPos[str(index + 1)]["x"]
+            y = soilPos[str(index + 1)]["y"]
+
+            # 已解锁格子
+            if index < soilUnlock:
+                isRipe = await cls.drawUnlockedTile(img, uid, index + 1, x, y, soilSize)
+                if isRipe and isFirstRipe:
+                    await cls.pasteRipeOnce(img, x, y, soilSize)
+                    isFirstRipe = False
+            else:
+                addedExpansion = await cls.drawLockedTile(
+                    img, x, y, soilSize, grass, isFirstExpansion
+                )
+                if addedExpansion:
+                    isFirstExpansion = False
+
+        await cls.drawUserInfo(img, player, uid)
+        await cls.applyDefinitionResize(img)
+
+        return img.pic2bytes()
+
+    @classmethod
+    async def createBaseCanvas(cls) -> tuple[BuildImage, tuple, dict, BuildImage]:
+        """创建基础画布并返回 (img, soilSize, soilPos, grass)
+
+        Returns:
+            tuple: (主画布, 土地尺寸, 土地坐标映射, 草地图)
+        """
+        img = BuildImage(
+            background=config.g_sResourcePath / "background/background.jpg"
+        )
+
+        soil = await getJsonManager().getSoil()
+        soilSize = soil["size"]
+        soilPos = soil["soil"]
+
+        grass = BuildImage(background=config.g_sResourcePath / "soil/草土地.png")
+        await grass.resize(0, soilSize[0], soilSize[1])
+
+        return img, soilSize, soilPos, grass
+
+    @classmethod
+    async def drawUnlockedTile(
+        cls, img: BuildImage, uid: str, soilIndex: int, x: int, y: int, soilSize
+    ) -> bool:
+        """绘制已解锁土地格（包含土地贴图与作物），返回该格是否可收获。
+
+        Args:
+            img (BuildImage): 主画布。
+            uid (str): 用户UID。
+            soilIndex (int): 土地索引（从1开始）。
+            x (int), y (int): 贴图位置。
+            soilSize: 土地贴图尺寸（[w,h] 或 (w,h)）。
+
+        Returns:
+            bool: 是否有可收获作物（isRipe）。
+        """
+        soilUrl = "soil/普通土地.png"
+        soilInfo = await g_pDBService.userSoil.getUserSoil(uid, soilIndex)
+
+        if soilInfo:
+            soilLevel = soilInfo.get("soilLevel", 0)
+            if soilLevel == 1:
+                soilUrl = "soil/红土地.png"
+            elif soilLevel == 2:
+                soilUrl = "soil/黑土地.png"
+            elif soilLevel == 3:
+                soilUrl = "soil/金土地.png"
+
+        soil = BuildImage(background=config.g_sResourcePath / soilUrl)
+        await soil.resize(0, soilSize[0], soilSize[1])
+        await img.paste(soil, (x, y))
+
+        isPlant, plant, isRipe, offsetX, offsetY = await cls.drawSoilPlant(
+            uid, soilIndex
+        )
+        if isPlant and plant is not None:
+            await img.paste(
+                plant,
+                (
+                    x + soilSize[0] // 2 - plant.width // 2 + offsetX,
+                    y + soilSize[1] // 2 - plant.height // 2 + offsetY,
+                ),
+            )
+
+        return bool(isRipe)
+
+    @classmethod
+    async def drawLockedTile(
+        cls,
+        img: BuildImage,
+        x: int,
+        y: int,
+        soilSize,
+        grass: BuildImage,
+        isFirstExpansion: bool,
+    ) -> bool:
+        """绘制未解锁格子并在首次出现时绘制扩建图标。
+
+        Returns:
+            bool: 是否在此格绘制了扩建图标（仅首次返回 True）。
+        """
+        await img.paste(grass, (x, y))
+
+        if isFirstExpansion:
+            expansion = BuildImage(
+                background=config.g_sResourcePath / "background/expansion.png"
+            )
+            await expansion.resize(0, 69, 69)
+            await img.paste(
+                expansion,
+                (
+                    x + soilSize[0] // 2 - expansion.width // 2,
+                    y + soilSize[1] // 2 - expansion.height,
+                ),
+            )
+            return True
+
+        return False
+
+    @classmethod
+    async def pasteRipeOnce(cls, img: BuildImage, x: int, y: int, soilSize):
+        """首次发现成熟作物时贴上成熟提示图（ripe）。"""
+        ripe = BuildImage(background=config.g_sResourcePath / "background/ripe.png")
+        await img.paste(
+            ripe,
+            (x + soilSize[0] // 2 - ripe.width // 2, y - ripe.height // 2),
+        )
+
+    @classmethod
+    async def drawUserInfo(cls, img: BuildImage, player, uid: str):
+        """绘制左上角用户信息（头像、框、昵称、经验、等级、货币等）。
+        TODO 需要详细的绘制头像框
+
+        Args:
+            img (BuildImage): 主画布。
+            player: 用户对象（包含 user 字段）。
+            uid (str): 用户UID。
+        """
+        image = await PlatformUtils.get_user_avatar(uid, "qq")
+        if image:
+            avatar = BuildImage(background=image)
+            await avatar.resize(0, 140, 150)
+            await img.paste(avatar, (125, 85))
+
+        frame = BuildImage(background=config.g_sResourcePath / "background/frame.png")
+        await img.paste(frame, (75, 44))
+
+        nameImg = await BuildImage.build_text_image(
+            player.user["name"], size=24, font_color=(77, 35, 4)
+        )
+        await img.paste(nameImg, (300, 92))
+
+        level = await player.getUserLevel()
+        beginX = 309
+        endX = 627
+        width = int((level[2] / level[1]) * (endX - beginX)) if level[1] else 0
+        await img.rectangle((beginX, 188, beginX + width, 222), (171, 194, 41))
+
+        # 经验
+        expImg = await BuildImage.build_text_image(
+            f"{level[2]} / {level[1]}", size=24, font_color=(102, 120, 19)
+        )
+        await img.paste(expImg, (390, 193))
+
+        # 等级
+        levelImg = await BuildImage.build_text_image(
+            str(level[0]), size=32, font_color=(214, 111, 1)
+        )
+        await img.paste(levelImg, (660, 187))
+
+        # 农场币
+        pointImg = await BuildImage.build_text_image(
+            str(player.user.get("point", 0)), size=24, font_color=(253, 253, 253)
+        )
+        await img.paste(pointImg, (330, 255))
+
+        # 点券
+        bondsImg = await BuildImage.build_text_image(
+            "0", size=24, font_color=(253, 253, 253)
+        )
+        await img.paste(bondsImg, (570, 255))
+
+    @classmethod
+    async def applyDefinitionResize(cls, img: BuildImage):
+        """根据配置调整绘制清晰度（original/medium/hight/其他）。"""
+        definition = Config.get_config("zhenxun_plugin_farm", "绘制农场清晰度")
+        if definition == "medium":
+            await img.resize(0.6)
+        elif definition == "hight":
+            await img.resize(0.8)
+        elif definition == "original":
+            pass
+        else:
+            await img.resize(0.4)
+
+    @classmethod
+    async def drawDetailFarmByUid(cls, uid: str) -> list:
+        """绘制用户详细信息
+
+        Args:
+            uid (str): 用户Uid
+
+        Returns:
+            list:
+        """
+
+        info = []
+        farm = await cls.drawFarmByUid(uid)
+        player = await getToolManager().getPlayerByUid(uid)
+        if not player:
+            return info
+
+        info.append(BuildImage.open(farm))
+
+        dataList = []
+        columnName = [
+            "-",
+            "土地ID",
+            "土地等级",
+            "作物名称",
+            "成熟时间",
+            "土地状态",
+            "被偷数量",
+            "剩余产出",
+        ]
+
+        icon = ""
+        soilNumber = player.user.get("soil", 3)
+
+        for i in range(1, soilNumber + 1):
+            soilInfo = await g_pDBService.userSoil.getUserSoil(uid, i)
+
+            if not soilInfo:
+                continue
+
+            match soilInfo.get("soilLevel", 0):
+                case 1:
+                    name = "红土地.png"
+                case 2:
+                    name = "黑土地.png"
+                case 3:
+                    name = "金土地.png"
+                case _:
+                    name = "普通土地.png"
+            iconPath = config.g_sResourcePath / "soil" / name
+
+            if iconPath.exists():
+                icon = (iconPath, 33, 33)
+
+            plantName = soilInfo.get("plantName", "-")
+
+            if plantName == "-":
+                matureTime = "-"
+                soilStatus = "-"
+                totalNumber = "-"
+                plantNumber = "-"
+            else:
+                matureTime = (
+                    getToolManager()
+                    .dateTime()
+                    .fromtimestamp(int(soilInfo.get("matureTime", 0)))
+                    .strftime("%Y-%m-%d %H:%M:%S")
+                )
+                soilStatus = await g_pDBService.userSoil.getUserSoilStatus(uid, i)
+
+                totalNumber = await g_pDBService.userSteal.getTotalStolenCount(uid, i)
+                planInfo = await g_pDBService.plant.getPlantByName(plantName)
+
+                if not planInfo:
+                    plantNumber = "None"
+                else:
+                    plantNumber = f"{planInfo['harvest'] - totalNumber}"
+
+            dataList.append(
+                [
+                    icon,
+                    i,
+                    await g_pDBService.userSoil.getSoilLevelText(soilInfo["soilLevel"]),
+                    plantName,
+                    matureTime,
+                    soilStatus,
+                    totalNumber,
+                    plantNumber,
+                ]
+            )
+
+            if len(dataList) >= 15:
+                result = await ImageTemplate.table_page(
+                    "土地详细信息",
+                    "",
+                    columnName,
+                    dataList,
+                )
+
+                info.append(result.copy())
+                dataList.clear()
+
+            if i >= soilNumber:
+                result = await ImageTemplate.table_page(
+                    "土地详细信息",
+                    "",
+                    columnName,
+                    dataList,
+                )
+
+                info.append(result.copy())
+                dataList.clear()
+
+        return info
+
+    @classmethod
+    async def drawSoilPlant(
+        cls, uid: str, soilIndex: int
+    ) -> tuple[bool, BuildImage, bool, int, int]:
+        """绘制植物资源
+
+        Args:
+            uid (str): 用户Uid
+            soilIndex (int): 土地索引 从1开始
+
+        Returns:
+            tuple[bool, BuildImage]: [绘制是否成功，资源图片, 是否成熟]
+        """
+
+        plant = None
+        soilInfo = await g_pDBService.userSoil.getUserSoil(uid, soilIndex)
+
+        if not soilInfo:
+            return False, None, False, 0, 0  # type: ignore
+
+        # 是否枯萎
+        if int(soilInfo.get("wiltStatus", 0)) == 1:
+            plant = BuildImage(background=config.g_sResourcePath / "plant/basic/9.png")
+            await plant.resize(0, 150, 212)
+            return True, plant, False, 0, 0
+
+        # 获取作物详细信息
+        plantInfo = await g_pDBService.plant.getPlantByName(soilInfo["plantName"])
+        if not plantInfo:
+            logger.error(f"绘制植物资源失败: {soilInfo['plantName']}")
+            return False, None, False, 0, 0  # type: ignore
+
+        offsetX = plantInfo.get("officX", 0)
+        offsetY = plantInfo.get("officY", 0)
+        offsetW = plantInfo.get("officW", 0)
+        offsetH = plantInfo.get("officH", 0)
+
+        currentTime = getToolManager().dateTime().now().timestamp()
+        phaseList = await g_pDBService.plant.getPlantPhaseByName(soilInfo["plantName"])
+
+        # 如果当前时间大于成熟时间 说明作物成熟
+        if currentTime >= soilInfo["matureTime"]:
+            plant = BuildImage(
+                background=config.g_sResourcePath
+                / f"plant/{soilInfo['plantName']}/{len(phaseList)}.png"
+            )
+
+            return True, plant, True, offsetX, offsetY
+        else:
+            # 如果是多阶段作物 且没有成熟 #早期思路 多阶段作物 直接是倒数第二阶段图片
+            # if soilInfo["harvestCount"] >= 1:
+            #     plant = BuildImage(
+            #         background=g_sResourcePath
+            #         / f"plant/{soilInfo['plantName']}/{plantInfo['phase'] - 1}.png"
+            #     )
+
+            #     return True, plant, False, offsetX, offsetY
+
+            # 如果没有成熟 则根据当前阶段进行绘制
+            elapsedTime = currentTime - soilInfo["plantTime"]
+            currentStage = currentStage = sum(
+                1 for thr in phaseList if elapsedTime >= thr
+            )
+
+            if currentStage <= 0:
+                if not plantInfo["general"]:
+                    plant = BuildImage(
+                        background=config.g_sResourcePath
+                        / f"plant/{soilInfo['plantName']}/0.png"
+                    )
+                else:
+                    plant = BuildImage(
+                        background=config.g_sResourcePath / "plant/basic/0.png"
+                    )
+                await plant.resize(0, 35 + offsetW, 58 + offsetH)
+            else:
+                plant = BuildImage(
+                    background=config.g_sResourcePath
+                    / f"plant/{soilInfo['plantName']}/{currentStage}.png"
+                )
+
+        return True, plant, False, offsetX, offsetY