#!/usr/bin/env python3 # -*- encoding: utf-8 -*- ''' @File : data_source.py @Time : 2020/11/21 11:13:42 @Author : Kyomotoi @Contact : kyomotoiowo@gmail.com @Github : https://github.com/Kyomotoi @License : Copyright © 2018-2020 Kyomotoi, All Rights Reserved. ''' __author__ = 'kyomotoi' import re import os import time import json import string import random import hashlib import requests from pathlib import Path from zipfile import PyZipFile from typing import Tuple, Dict, List class Generate: """虚拟身份证部分""" GENERATE_DATA_PATH = Path( '.') / 'ATRI' / 'plugins' / 'plugin_utils' / 'main.bin' def infoID(self) -> Tuple[Dict[str, List[str]], Dict[str, str]]: with PyZipFile(os.path.abspath(self.GENERATE_DATA_PATH), "r") as zipFile: with zipFile.open("name.json", "r") as f: name = json.loads(f.read().decode()) with zipFile.open("area.json", "r") as f: area = json.loads(f.read().decode()) return name, area def numberID(self, area: int, sex: int, birth: int) -> str: def checkSum(fullCode: str) -> int or str: assert len(fullCode) == 17 checkSum = sum([((1 << (17 - i)) % 11) * int(fullCode[i]) for i in range(0, 17)]) checkDigit = (12 - (checkSum % 11)) % 11 if checkDigit < 10: return checkDigit else: return "X" # type: ignore orderCode = str(random.randint(10, 99)) sexCode = str(random.randrange(sex, 10, step=2)) fullCode = str(area) + str(birth) + str(orderCode) + str(sexCode) fullCode += str(checkSum(fullCode)) return fullCode class Genshin: """原神部分""" def md5(self, text: str) -> str: """text 转 md5""" md5 = hashlib.md5() md5.update(text.encode()) return md5.hexdigest() def DSGet(self) -> str: mhyVersion = "2.1.0" n = self.md5(mhyVersion) i = str(int(time.time())) r = ''.join(random.sample(string.ascii_lowercase + string.digits, 6)) c = self.md5("salt=" + n + "&t=" + i + "&r=" + r) return i + "," + r + "," + c def GetInfo(self, uid: str) -> str: """请求API""" req = requests.get( url= f"https://api-takumi.mihoyo.com/game_record/genshin/api/index?server=cn_gf01&role_id={uid}", headers={ 'Accept': 'application/json, text/plain, */*', 'DS': self.DSGet(), 'Origin': 'https://webstatic.mihoyo.com', 'x-rpc-app_version': '2.1.0', 'User-Agent': 'Mozilla/5.0 (Linux; Android 9; Unspecified Device) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/39.0.0.0 Mobile Safari/537.36 miHoYoBBS/2.2.0', 'x-rpc-client_type': '4', 'Referer': 'https://webstatic.mihoyo.com/app/community-game-records/index.html?v=6', 'Accept-Encoding': 'gzip, deflate', 'Accept-Language': 'zh-CN,en-US;q=0.8', 'X-Requested-With': 'com.mihoyo.hyperion' }) return req.text def JsonAnalysis(self, JsonText) -> str: """解析数据""" data = json.loads(JsonText) Character_Info = "Roles:\n" Character_List = [] Character_List = data["data"]["avatars"] for i in Character_List: if (i["element"] == "None"): Character_Type = "无属性" elif (i["element"] == "Anemo"): Character_Type = "风属性" elif (i["element"] == "Pyro"): Character_Type = "火属性" elif (i["element"] == "Geo"): Character_Type = "岩属性" elif (i["element"] == "Electro"): Character_Type = "雷属性" elif (i["element"] == "Cryo"): Character_Type = "冰属性" elif (i["element"] == "Hydro"): Character_Type = "水属性" else: Character_Type = "草属性" if (i["name"] == "旅行者"): if (i["image"].find("UI_AvatarIcon_PlayerGirl") != -1): TempText = f'* {i["name"]}：\n' TempText += f' - [萤——妹妹] {i["level"]}级 {Character_Type}\n' elif (i["image"].find("UI_AvatarIcon_PlayerBoy") != -1): TempText = f'* {i["name"]}：\n' TempText += f' - [空——哥哥] {i["level"]}级 {Character_Type}\n' else: TempText = f'* {i["name"]}：\n' TempText += f' - [性别判断失败] {i["level"]}级 {Character_Type}\n' else: TempText = f'* {i["name"]} {i["rarity"]}★角色:\n' TempText += f' - {i["level"]}级 好感度({i["fetter"]})级 {Character_Type}\n' Character_Info = Character_Info + TempText a1 = data["data"]["stats"]["spiral_abyss"] Account_Info = 'Account Info:\n' Account_Info += f'- 活跃天数：{data["data"]["stats"]["active_day_number"]} 天\n' Account_Info += f'- 达成成就：{data["data"]["stats"]["achievement_number"]} 个\n' Account_Info += f'- 获得角色：{data["data"]["stats"]["avatar_number"]}个\n' Account_Info += f'- 深渊螺旋：{"没打" if (data["data"]["stats"]["spiral_abyss"] == "-") else f"打到了{a1}"}\n' Account_Info += f'* 收集：\n' Account_Info += f' - 风神瞳{data["data"]["stats"]["anemoculus_number"]}个 岩神瞳{data["data"]["stats"]["geoculus_number"]}个\n' Account_Info += f'* 解锁：\n' Account_Info += f' - 传送点{data["data"]["stats"]["way_point_number"]}个 秘境{data["data"]["stats"]["domain_number"]}个\n' Account_Info += f'* 共开启宝箱：\n' Account_Info += f' - 普通：{data["data"]["stats"]["common_chest_number"]}个 精致：{data["data"]["stats"]["exquisite_chest_number"]}个\n' Account_Info += f' - 珍贵：{data["data"]["stats"]["luxurious_chest_number"]}个 华丽：{data["data"]["stats"]["precious_chest_number"]}个' return Character_Info + "\r\n" + Account_Info class Roll: """骰娘部分""" def roll_dice(self, par: str) -> str: """掷骰子""" result = 0 proc = '' proc_list = [] p = par.split('+') # 计算每个单独的roll for i in p: args = re.findall(r"(\d{0,10})(?:(d)(\d{1,10}))", i) args = list(args[0]) if not args[0]: args[0] = 1 if int(args[0]) >= 5000 or int(args[2]) >= 5000: return '阿..好大...' for a in range(1, int(args[0]) + 1): rd = random.randint(1, int(args[2])) result = result + rd if len(proc_list) <= 10: proc_list.append(rd) if len(proc_list) <= 10: proc += "+".join(map(str, proc_list)) elif len(proc_list) >= 10: proc += '太长了不展示了就酱' else: proc += str(result) result = f"{par}=({proc})={result}" return str(result)