Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update data_source.py

Browse Source
This commit is contained in:
HibiKier 2021-10-03 21:40:05 +08:00 committed by GitHub
parent 5fd0667061
commit fedcb37b1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 145 additions and 110 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

123
plugins/epic/data_source.py
View File

@ -1,22 +1,25 @@
from httpx import AsyncClient from httpx import AsyncClient
from datetime import datetime from datetime import datetime
from nonebot.log import logger from nonebot.log import logger
from nonebot.adapters.cqhttp import Bot, Event, Message, MessageEvent, GroupMessageEvent from nonebot.adapters.cqhttp import (
Bot,
Event,
GroupMessageEvent,
)
from configs.config import NICKNAME from configs.config import NICKNAME
# 获取所有 Epic Game Store 促销游戏 # 获取所有 Epic Game Store 促销游戏
# 方法参考：RSSHub /epicgames 路由 # 方法参考：RSSHub /epicgames 路由
# https://github.com/DIYgod/RSSHub/blob/master/lib/routes/epicgames/index.js # https://github.com/DIYgod/RSSHub/blob/master/lib/routes/epicgames/index.js
async def get_Epicgame(): async def get_epic_game():
epic_url = "https://www.epicgames.com/store/backend/graphql-proxy" epic_url = "https://www.epicgames.com/store/backend/graphql-proxy"
headers = { headers = {
"Referer": "https://www.epicgames.com/store/zh-CN/", "Referer": "https://www.epicgames.com/store/zh-CN/",
"Content-Type": "application/json; charset=utf-8", "Content-Type": "application/json; charset=utf-8",
} }
data = { data = {
"query": "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",
"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": { "variables": {
"allowCountries": "CN", "allowCountries": "CN",
"category": "freegames", "category": "freegames",
@ -26,14 +29,14 @@ async def get_Epicgame():
"sortBy": "effectiveDate", "sortBy": "effectiveDate",
"sortDir": "asc", "sortDir": "asc",
"withPrice": True, "withPrice": True,
"withPromotions": True "withPromotions": True,
} },
} }
async with AsyncClient(headers=headers) as client: async with AsyncClient(headers=headers) as client:
try: try:
res = await client.post(epic_url, json=data, timeout=10.0) res = await client.post(epic_url, json=data, timeout=10.0)
resJson = res.json() res_json = res.json()
games = resJson['data']['Catalog']['searchStore']['elements'] games = res_json["data"]["Catalog"]["searchStore"]["elements"]
return games return games
except Exception as e: except Exception as e:
logger.error(str(e)) logger.error(str(e))
@ -43,33 +46,39 @@ async def get_Epicgame():
# 获取 Epic Game Store 免费游戏信息 # 获取 Epic Game Store 免费游戏信息
# 处理免费游戏的信息方法借鉴 pip 包 epicstore_api 示例 # 处理免费游戏的信息方法借鉴 pip 包 epicstore_api 示例
# https://github.com/SD4RK/epicstore_api/blob/master/examples/free_games_example.py # https://github.com/SD4RK/epicstore_api/blob/master/examples/free_games_example.py
async def get_Epicfree(bot: Bot, event: Event): async def get_epic_free(bot: Bot, event: Event):
games = await get_Epicgame() games = await get_epic_game()
if not games: if not games:
return "Epic 可能又抽风啦，请稍后再试（", 404 return "Epic 可能又抽风啦，请稍后再试（", 404
else: else:
msg = ""
msg_list = [] msg_list = []
for game in games: for game in games:
game_name = game['title'] game_name = game["title"]
game_corp = game['seller']['name'] game_corp = game["seller"]["name"]
game_price = game['price']['totalPrice']['fmtPrice']['originalPrice'] game_price = game["price"]["totalPrice"]["fmtPrice"]["originalPrice"]
# 赋初值以避免 local variable referenced before assignment # 赋初值以避免 local variable referenced before assignment
game_dev, game_pub, game_thumbnail = (None, None, None) game_dev, game_pub, game_thumbnail = (None, None, None)
try: try:
game_promotions = game['promotions']['promotionalOffers'] game_promotions = game["promotions"]["promotionalOffers"]
upcoming_promotions = game['promotions']['upcomingPromotionalOffers'] upcoming_promotions = game["promotions"]["upcomingPromotionalOffers"]
if not game_promotions and upcoming_promotions: if not game_promotions and upcoming_promotions:
# 促销暂未上线，但即将上线 # 促销暂未上线，但即将上线
promotion_data = upcoming_promotions[0]['promotionalOffers'][0] promotion_data = upcoming_promotions[0]["promotionalOffers"][0]
start_date_iso, end_date_iso = (promotion_data['startDate'][:-1], start_date_iso, end_date_iso = (
promotion_data['endDate'][:-1]) promotion_data["startDate"][:-1],
promotion_data["endDate"][:-1],
)
# 删除字符串中最后一个 "Z" 使 Python datetime 可处理此时间 # 删除字符串中最后一个 "Z" 使 Python datetime 可处理此时间
start_date = datetime.fromisoformat(start_date_iso).strftime( start_date = datetime.fromisoformat(start_date_iso).strftime(
'%b.%d %H:%M') "%b.%d %H:%M"
end_date = datetime.fromisoformat(end_date_iso).strftime('%b.%d %H:%M') )
end_date = datetime.fromisoformat(end_date_iso).strftime(
"%b.%d %H:%M"
)
if isinstance(event, GroupMessageEvent): if isinstance(event, GroupMessageEvent):
_message = '\n由 {} 公司发行的游戏 {} ({}) 在 UTC 时间 {} 即将推出免费游玩，预计截至 {}。'.format(game_corp, game_name, game_price, start_date, end_date) _message = "\n由 {} 公司发行的游戏 {} ({}) 在 UTC 时间 {} 即将推出免费游玩，预计截至 {}。".format(
game_corp, game_name, game_price, start_date, end_date
)
data = { data = {
"type": "node", "type": "node",
"data": { "data": {
@ -80,30 +89,49 @@ async def get_Epicfree(bot: Bot, event: Event):
} }
msg_list.append(data) msg_list.append(data)
else: else:
msg = '\n由 {} 公司发行的游戏 {} ({}) 在 UTC 时间 {} 即将推出免费游玩，预计截至 {}。'.format( msg = "\n由 {} 公司发行的游戏 {} ({}) 在 UTC 时间 {} 即将推出免费游玩，预计截至 {}。".format(
game_corp, game_name, game_price, start_date, end_date) game_corp, game_name, game_price, start_date, end_date
)
msg_list.append(msg) msg_list.append(msg)
else: else:
for image in game['keyImages']: for image in game["keyImages"]:
if image['type'] == 'Thumbnail': if image["type"] == "Thumbnail":
game_thumbnail = image['url'] game_thumbnail = image["url"]
for pair in game['customAttributes']: for pair in game["customAttributes"]:
if pair['key'] == 'developerName': if pair["key"] == "developerName":
game_dev = pair['value'] game_dev = pair["value"]
if pair['key'] == 'publisherName': if pair["key"] == "publisherName":
game_pub = pair['value'] game_pub = pair["value"]
# 如 game['customAttributes'] 未找到则均使用 game_corp 值 # 如 game['customAttributes'] 未找到则均使用 game_corp 值
game_dev = game_dev if game_dev != None else game_corp game_dev = game_dev if game_dev is not None else game_corp
game_pub = game_pub if game_pub != None else game_corp game_pub = game_pub if game_pub is not None else game_corp
game_desp = game['description'] game_desp = game["description"]
end_date_iso = game['promotions']['promotionalOffers'][0]['promotionalOffers'][0]['endDate'][:-1] end_date_iso = game["promotions"]["promotionalOffers"][0][
end_date = datetime.fromisoformat(end_date_iso).strftime('%b.%d %H:%M') "promotionalOffers"
][0]["endDate"][:-1]
end_date = datetime.fromisoformat(end_date_iso).strftime(
"%b.%d %H:%M"
)
# API 返回不包含游戏商店 URL，此处自行拼接，可能出现少数游戏 404 请反馈 # API 返回不包含游戏商店 URL，此处自行拼接，可能出现少数游戏 404 请反馈
game_url_part = (game['productSlug'].replace('/home', '')) if ('/home' in game['productSlug']) else game['productSlug'] game_url_part = (
game_url = 'https://www.epicgames.com/store/zh-CN/p/{}'.format( (game["productSlug"].replace("/home", ""))
game_url_part) if ("/home" in game["productSlug"])
else game["productSlug"]
)
game_url = "https://www.epicgames.com/store/zh-CN/p/{}".format(
game_url_part
)
if isinstance(event, GroupMessageEvent): if isinstance(event, GroupMessageEvent):
_message = '[CQ:image,file={}]\n\nFREE now :: {} ({})\n{}\n此游戏由 {} 开发、{} 发行，将在 UTC 时间 {} 结束免费游玩，戳链接速度加入你的游戏库吧~\n{}\n'.format(game_thumbnail, game_name, game_price, game_desp, game_dev, game_pub,end_date, game_url) _message = "[CQ:image,file={}]\n\nFREE now :: {} ({})\n{}\n此游戏由 {} 开发、{} 发行，将在 UTC 时间 {} 结束免费游玩，戳链接速度加入你的游戏库吧~\n{}\n".format(
game_thumbnail,
game_name,
game_price,
game_desp,
game_dev,
game_pub,
end_date,
game_url,
)
data = { data = {
"type": "node", "type": "node",
"data": { "data": {
@ -114,9 +142,16 @@ async def get_Epicfree(bot: Bot, event: Event):
} }
msg_list.append(data) msg_list.append(data)
else: else:
msg = '[CQ:image,file={}]\n\nFREE now :: {} ({})\n{}\n此游戏由 {} 开发、{} 发行，将在 UTC 时间 {} 结束免费游玩，戳链接速度加入你的游戏库吧~\n{}\n'.format( msg = "[CQ:image,file={}]\n\nFREE now :: {} ({})\n{}\n此游戏由 {} 开发、{} 发行，将在 UTC 时间 {} 结束免费游玩，戳链接速度加入你的游戏库吧~\n{}\n".format(
game_thumbnail, game_name, game_price, game_desp, game_dev, game_pub, game_thumbnail,
end_date, game_url) game_name,
game_price,
game_desp,
game_dev,
game_pub,
end_date,
game_url,
)
msg_list.append(msg) msg_list.append(msg)
except TypeError as e: except TypeError as e:
# logger.info(str(e)) # logger.info(str(e))