You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
4.3 KiB
102 lines
4.3 KiB
import ssl
|
|
import discord
|
|
from discord import app_commands
|
|
import os, sys
|
|
import aiohttp
|
|
from datetime import datetime
|
|
from discord.ext import tasks
|
|
from player import *
|
|
import traceback
|
|
from exceptions import *
|
|
|
|
#Скрыть сообщение если надо ephemeral=True
|
|
class DiscordClient(discord.Client):
|
|
ANY_INPUT = "ссылка на стим | имя игрока | стимид"
|
|
discord2steam_cache = {}
|
|
backend_url = ""
|
|
secret_key = ""
|
|
stats = {}
|
|
show_stats_prev = 0
|
|
|
|
def __init__(self, backend_url, secret_key):
|
|
self.backend_url = backend_url
|
|
self.secret_key = secret_key
|
|
###################################################
|
|
super().__init__(intents=discord.Intents.default())
|
|
self.tree = app_commands.CommandTree(self)
|
|
self.load_extensions(['user_ext', 'admin_ext', 'other_ext'])
|
|
###################################################
|
|
self.setup_events()
|
|
|
|
def load_extensions(self, extensions_path):
|
|
if type(extensions_path) == str:
|
|
extensions_path = [extensions_path]
|
|
for path in extensions_path:
|
|
print(f"Load extensions from: {path}")
|
|
sys.path.insert(0, path)
|
|
for extension in os.listdir(path):
|
|
extension, ext = os.path.splitext(extension)
|
|
if ext != ".py":
|
|
continue
|
|
print(f"Loading: {extension}")
|
|
__import__(extension).Extension(self)
|
|
sys.path.pop(0)
|
|
|
|
async def setup_hook(self):
|
|
self.stats = await self.GetStats()
|
|
print("sync tree")
|
|
if os.getenv("MAIN_DISCORD_SERVER_ID",""):
|
|
await self.tree.sync(guild=discord.Object(int(os.getenv("MAIN_DISCORD_SERVER_ID"))))
|
|
await self.tree.sync()
|
|
|
|
def setup_events(self):
|
|
@self.event
|
|
async def on_ready():
|
|
print(f'Logged in as {self.user} (ID: {self.user.id})')
|
|
|
|
@self.tree.error
|
|
async def on_app_command_error(interaction, error):
|
|
if isinstance(error.original, CannotCastToSteamID):
|
|
return await interaction.response.send_message("Не возможно найти такой профиль, попробуй написать иные данные!", ephemeral=True)
|
|
elif isinstance(error.original, LowPermition):
|
|
#Добавить проверку на модератора или в бекенде добавить другой код ответа
|
|
return await interaction.response.send_message("Это не для тебя и не для таких как ты сделано...\n", ephemeral=True)
|
|
elif isinstance(error.original, NotFoundPlayerOnServer):
|
|
return await interaction.response.send_message("Игрок не найден на серверах", ephemeral=True)
|
|
elif isinstance(error.original, UnknownBackendResponse):
|
|
return await interaction.response.send_message("Ошибка на стороне сервера в исполнении говнокода, стоит подождать или позвать помощь", ephemeral=False)
|
|
elif isinstance(error.original, discord.errors.NotFound):
|
|
return await interaction.response.send_message("Слишком долгий ответ со стороны сервера, причины:\n1) Возможно бекенд сдох\n2)Cлишком долгий незапланированный ответ с сервера\n3)Стоит позвать помощь", ephemeral=True)
|
|
traceback.print_exc()
|
|
|
|
async def GetSteam64OfDiscord(self, user, no_cache = False):
|
|
if user.id in self.discord2steam_cache and not no_cache:
|
|
return self.discord2steam_cache[user.id]
|
|
|
|
async with aiohttp.ClientSession(cookies={"secretkey":self.secret_key}) as session:
|
|
async with session.get(f"{self.backend_url}/api/discord?discord_id={user.id}", ssl=False) as response:
|
|
steamid_response = await response.json()
|
|
if steamid_response != None:
|
|
self.discord2steam_cache[user.id] = steamid_response["steam64"]
|
|
else:
|
|
raise NeedDiscordAuthOfSteam
|
|
|
|
return self.discord2steam_cache[user.id]
|
|
|
|
async def GetPlayer(self, profile, requester_steam64, load_profile = True):
|
|
player = Player(profile, requester_steam64, self.stats)
|
|
await player.GetSteamID()
|
|
if load_profile:
|
|
await player.LoadProfile()
|
|
return player
|
|
|
|
async def GetStats(self):
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(f"{os.getenv('BACKEND_URL')}/api/stats", ssl=False) as response:
|
|
return await response.json()
|
|
|
|
if __name__ == "__main__":
|
|
DiscordClient(
|
|
os.getenv("BACKEND_URL"),
|
|
os.getenv("BACKEND_SECRETKEY")
|
|
).run(os.getenv("DISCORD_TOKEN"))
|