Merge pull request #368 from Hpero4/master

添加大乱斗Buff功能 (#314)

app/common/util.py

@@ -1,11 +1,16 @@
 import json
+import os
+import re
+from functools import lru_cache, wraps
 
 import requests
 import base64
 import subprocess
 import psutil
+import win32api
 
-from app.common.config import cfg, VERSION
+from app.common.config import cfg, VERSION, LOCAL_PATH
+from app.common.logger import logger
 
 
 class Github:
@@ -56,7 +61,6 @@ def __get_ver_info(self):
 
         return json_data.get(VERSION, {})
 
-
     def getNotice(self):
         url = f'{self.githubApi}/repos/{self.user}/{self.repositories}/contents/document/notice.md'
 
@@ -75,6 +79,158 @@ def getNotice(self):
         }
 
 
+class AramHome:
+    """
+    Power by: 大乱斗之家
+    Site: http://www.jddld.com
+    """
+    ARAM_CFG_PATH = f"{LOCAL_PATH}/AramBuff.json"
+    URL = "http://www.jddld.com"
+    TAG = "AramHome"
+    APPID = 1
+    APP_SECRET = "PHPCMFBBC77AF8E8FA5"
+    data = None
+
+    @staticmethod
+    def needData(func):
+        """
+        TODO 如何优雅地使用它? -- By Hpero4
+        """
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            if AramHome.data is None:
+                m = AramHome()
+                m.__loadData()
+            return func(*args, **kwargs)
+
+        return wrapper
+
+    @classmethod
+    async def checkAndUpdate(cls):
+        m = cls()
+        if m.__checkUpdate():
+            await m.__update()
+
+    @classmethod
+    def getInfoByHeroId(cls, heroid: str):
+        return cls.getInfoByField("heroid", heroid)
+
+    @classmethod
+    def getInfoByHeroName(cls, name: str):
+        return cls.getInfoByField("name", name)
+
+    @classmethod
+    def getInfoByCatName(cls, name: str):
+        return cls.getInfoByField("catname", name)
+
+    @classmethod
+    def getInfoByCatNameLoose(cls, name: str):
+        return cls.getInfoByFieldLoose("catname", name)
+
+    @classmethod
+    # @needData
+    @lru_cache(maxsize=None)
+    def getInfoByField(cls, field: str, val: str):
+        """
+        通过字段值匹配一个item
+
+        找不到返回 None
+        """
+        if cls.data is None:
+            m = cls()
+            m.__loadData()
+        for item in cls.data.values():
+            if item[field] == val:
+                return item
+        return None
+
+    @classmethod
+    # @needData
+    @lru_cache(maxsize=None)
+    def getInfoByFieldLoose(cls, field: str, val: str):
+        """
+        通过字段值匹配一个item
+
+        宽松匹配, 只要val在field内都认为找到
+
+        找不到返回 None
+        """
+        if cls.data is None:
+            m = cls()
+            m.__loadData()
+        for item in cls.data.values():
+            if val in item[field]:
+                return item
+        return None
+
+    def __loadData(self):
+        with open(self.ARAM_CFG_PATH, 'r') as f:
+            AramHome.data = json.loads(f.read())
+
+    async def __update(self):
+        logger.info("update info", self.TAG)
+        url = f"{self.URL}/index.php"
+        params = {
+            'appid': self.APPID,
+            'appsecret': self.APP_SECRET,
+            's': 'news',
+            'c': 'search',
+            'api_call_function': 'module_list',
+            'pagesize': '200'  # FIXME 超过200个英雄会拿不完 -- By Hpero4
+        }
+        data = requests.get(url, params=params, proxies=None, verify=False).json()  # 它不需要代理
+        if data.get("code") == 1:
+            with open(self.ARAM_CFG_PATH, "w") as f:
+                data = data.get("data")
+                json.dump(data, f)
+                AramHome.data = data
+        else:
+            logger.warning(f"update err: {data}", self.TAG)
+
+    def __checkUpdate(self):
+        """
+        检查缓存的数据与当前版本是否匹配, 若不匹配尝试更新
+
+        尽可能在游戏启动后再调用, 否则当存在多个客户端时, cfg.lolFolder不一定是准确的(外服国服可能版本不同)
+
+        @return :
+            True -> 需要更新
+            False -> 无需更新
+
+        TODO: 暂未提供历史版本数据查询接口
+        """
+        gamePath = cfg.get(cfg.lolFolder)
+
+        assert gamePath  # 必须有, 否则就是调用逻辑有问题 -- By Hpero4
+
+        # 特判一下国服 -- By Hpero4
+        gamePath = gamePath.replace("/TCLS", "")
+
+        lolExe = f"{gamePath}/Game/League of Legends.exe"
+        # 判断一下, 客户端特殊? 为啥会没有LOL的主程序 -- By Hpero4
+        if not os.path.exists(lolExe):
+            raise FileNotFoundError(lolExe)
+
+        fileInfo = getFileProperties(lolExe).get("StringFileInfo", {})
+        lolVer = fileInfo.get("ProductVersion") or fileInfo.get("FileVersion")
+
+        assert lolVer
+
+        # 缩短至大版本号
+        lolVer = re.search(r"\d+\.\d+", lolVer).group(0)
+
+        # 检查一下版本号是否相同, 如果一样, 就不更新了
+        if os.path.exists(self.ARAM_CFG_PATH):
+            with open(self.ARAM_CFG_PATH, "r") as f:
+                data = json.loads(f.read())
+                AramHome.data = data
+                dataVer = re.search(r"\d+\.\d+", data.get("banben", "")).group(0)
+                if dataVer and dataVer == lolVer:
+                    return False
+
+        return True
+
+
 github = Github()
 
 
@@ -107,7 +263,7 @@ def getLolClientPid(path):
         arr = processes.split()
         try:
             pos = arr.index(b"LeagueClientUx.exe")
-            return int(arr[pos+1])
+            return int(arr[pos + 1])
         except ValueError:
             raise ValueError(f"Subprocess return exception: {processes}")
     else:
@@ -176,3 +332,50 @@ def getPortTokenServerByPid(pid):
             break
 
     return port, token, server
+
+
+def getFileProperties(fname):
+    """
+    读取给定文件的所有属性, 返回一个字典.
+
+    returns : {'FixedFileInfo': {'Signature': -17890115, 'StrucVersion': 65536, 'FileVersionMS': 917513, 'FileVersionLS':
+    38012988, 'ProductVersionMS': 917513, 'ProductVersionLS': 38012988, 'FileFlagsMask': 23, 'FileFlags': 0,
+    'FileOS': 4, 'FileType': 1, 'FileSubtype': 0, 'FileDate': None}, 'StringFileInfo': {'Comments': None,
+    'InternalName': 'League of Legends (TM) Client', 'ProductName': 'League of Legends (TM) Client', 'CompanyName':
+    'Riot Games, Inc.', 'LegalCopyright': 'Copyright (C) 2009', 'ProductVersion': '14.9.580.2108', 'FileDescription':
+    'League of Legends (TM) Client', 'LegalTrademarks': None, 'PrivateBuild': None, 'FileVersion': '14.9.580.2108',
+    'OriginalFilename': 'League of Legends.exe', 'SpecialBuild': None}, 'FileVersion': '14.9.580.2108'}
+
+    """
+
+    propNames = ('Comments', 'InternalName', 'ProductName',
+                 'CompanyName', 'LegalCopyright', 'ProductVersion',
+                 'FileDescription', 'LegalTrademarks', 'PrivateBuild',
+                 'FileVersion', 'OriginalFilename', 'SpecialBuild')
+
+    props = {'FixedFileInfo': None, 'StringFileInfo': None, 'FileVersion': None}
+
+    try:
+        fixedInfo = win32api.GetFileVersionInfo(fname, '\\')
+        props['FixedFileInfo'] = fixedInfo
+        props['FileVersion'] = "%d.%d.%d.%d" % (fixedInfo['FileVersionMS'] / 65536,
+                                                fixedInfo['FileVersionMS'] % 65536, fixedInfo['FileVersionLS'] / 65536,
+                                                fixedInfo['FileVersionLS'] % 65536)
+
+        # \VarFileInfo\Translation returns list of available (language, codepage)
+        # pairs that can be used to retreive string info. We are using only the first pair.
+        lang, codepage = win32api.GetFileVersionInfo(fname, '\\VarFileInfo\\Translation')[0]
+
+        # any other must be of the form \StringfileInfo\%04X%04X\parm_name, middle
+        # two are language/codepage pair returned from above
+
+        strInfo = {}
+        for propName in propNames:
+            strInfoPath = u'\\StringFileInfo\\%04X%04X\\%s' % (lang, codepage, propName)
+            strInfo[propName] = win32api.GetFileVersionInfo(fname, strInfoPath)
+
+        props['StringFileInfo'] = strInfo
+    except:
+        return {}
+    else:
+        return props