diff --git a/epic.py b/epic.py new file mode 100644 index 0000000..f1f198e --- /dev/null +++ b/epic.py @@ -0,0 +1,99 @@ +import os +import sys +from requests import post, get +from pytz import timezone +from datetime import datetime +from pagermaid.listener import listener +from pagermaid.utils import alias_command + + +@listener(is_plugin=True, outgoing=True, command=alias_command("epic"), + description="获取 Epic 喜加一信息") +async def epic(context): + await context.edit("获取中 . . .") + epic_url = "https://www.epicgames.com/store/backend/graphql-proxy" + headers = { + "Referer": "https://www.epicgames.com/store/zh-CN/", + "Content-Type": "application/json; charset=utf-8", + } + data = { + "query": + "query searchStoreQuery($allowCountries: String, $category: String, $count: Int, $country: String!, " + "$keywords: String, $locale: String, $namespace: String, $sortBy: String, $sortDir: String, $start: Int, " + "$tag: String, $withPrice: Boolean = false, $withPromotions: Boolean = false) {\n Catalog {\n " + "searchStore(allowCountries: $allowCountries, category: $category, count: $count, country: $country, " + "keywords: $keywords, locale: $locale, namespace: $namespace, sortBy: $sortBy, sortDir: $sortDir, " + "start: $start, tag: $tag) {\n elements {\n title\n id\n namespace\n description\n effectiveDate\n " + "keyImages {\n type\n url\n }\n seller {\n id\n name\n }\n productSlug\n urlSlug\n url\n items {\n id\n " + "namespace\n }\n customAttributes {\n key\n value\n }\n categories {\n path\n }\n price(country: " + "$country) @include(if: $withPrice) {\n totalPrice {\n discountPrice\n originalPrice\n voucherDiscount\n " + "discount\n currencyCode\n currencyInfo {\n decimals\n }\n fmtPrice(locale: $locale) {\n originalPrice\n " + "discountPrice\n intermediatePrice\n }\n }\n lineOffers {\n appliedRules {\n id\n endDate\n " + "discountSetting {\n discountType\n }\n }\n }\n }\n promotions(category: $category) @include(if: " + "$withPromotions) {\n promotionalOffers {\n promotionalOffers {\n startDate\n endDate\n discountSetting {" + "\n discountType\n discountPercentage\n }\n }\n }\n upcomingPromotionalOffers {\n promotionalOffers {\n " + "startDate\n endDate\n discountSetting {\n discountType\n discountPercentage\n }\n }\n }\n }\n }\n paging " + "{\n count\n total\n }\n }\n }\n}\n", + "variables": { + "allowCountries": "CN", + "category": "freegames", + "count": 1000, + "country": "CN", + "locale": "zh-CN", + "sortBy": "effectiveDate", + "sortDir": "asc", + "withPrice": True, + "withPromotions": True + } + } + try: + res = post(epic_url, headers=headers, json=data, timeout=10.0) + resJson = res.json() + games = resJson["data"]["Catalog"]["searchStore"]["elements"] + except Exception as e: + return await context.edit("请求 Epic Store API 错误:" + str(sys.exc_info()[0]) + "\n" + str(e)) + if not games: + return await context.edit("Epic 可能又抽风啦,请稍后再试(") + else: + for game in games: + try: + game_name = game["title"] + game_corp = game["seller"]["name"] + game_price = game["price"]["totalPrice"]["fmtPrice"]["originalPrice"] + game_promotions = game["promotions"]["promotionalOffers"] + upcoming_promotions = game["promotions"]["upcomingPromotionalOffers"] + if not game_promotions and upcoming_promotions: + continue # 促销即将上线,跳过 + else: + game_thumbnail, game_dev, game_pub = None, None, None + for image in game["keyImages"]: + game_thumbnail = image["url"] if image["type"] == "Thumbnail" else None + for pair in game["customAttributes"]: + game_dev = pair["value"] if pair["key"] == "developerName" else game_corp + game_pub = pair["value"] if pair["key"] == "publisherName" else game_corp + game_desp = game["description"] + end_date_iso = game["promotions"]["promotionalOffers"][0]["promotionalOffers"][0]["endDate"][:-1] + end_date = datetime.fromisoformat(end_date_iso).replace( + tzinfo=timezone('UTC')).astimezone(timezone('Asia/Chongqing')).strftime("%Y-%m-%d %H:%M:%S") + # API 返回不包含游戏商店 URL,此处自行拼接,可能出现少数游戏 404 + game_url = f"https://www.epicgames.com/store/zh-CN/p/{game['productSlug'].replace('/home', '')}" + msg = f"**FREE now :: {game_name} ({game_price})**\n\n{game_desp}\n\n" + msg += f"游戏由 {game_pub} 发售," if game_dev == game_pub else f"游戏由 {game_dev} 开发、{game_pub} 出版," + msg += f"将在 **{end_date}** 结束免费游玩,戳下面的链接领取吧~\n{game_url}" + + if game_thumbnail: + r = get(game_thumbnail, timeout=10.0) + with open("epic.jpg", "wb") as code: + code.write(r.content) + await context.client.send_file(context.chat_id, "epic.jpg", caption=msg, force_document=False) + else: + await context.respond(msg) + except (TypeError, IndexError): + pass + except Exception as e: + await context.respond("获取 Epic 信息错误:" + str(sys.exc_info()[0]) + "\n" + str(e)) + try: + os.remove("epic.jpg") + except FileNotFoundError: + pass + await context.delete() diff --git a/list.json b/list.json index cb4d1a7..301527f 100644 --- a/list.json +++ b/list.json @@ -779,6 +779,16 @@ "supported": true, "des-short": "搜索消息插件", "des": "查找最近50条消息中包含指定关键字的消息,并推送相应链接人名及信息至Log频道。\n指令:-findmessage" + }, + { + "name": "epic", + "version": "1.00", + "section": "chat", + "maintainer": "xtaodada", + "size": "5.80 kb", + "supported": true, + "des-short": "获取 Epic 喜加一信息", + "des": "获取 Epic 喜加一信息。\n指令:-epic" } ] }