added lol api functionality for spectator-v4 and summoner-v4

diff --git a/skgyorugo/lol_api/__init__.py b/skgyorugo/lol_api/__init__.py
new file mode 100644 (file)
index 0000000..0bf6605
--- /dev/null
+++ b/skgyorugo/lol_api/__init__.py
@@ -0,0 +1,12 @@
+import urllib3
+from dataclasses import dataclass
+import os
+import json
+import logging
+from typing import Optional
+
+BASE_URL = "https://euw1.api.riotgames.com"
+
+KEY = os.getenv("LOL_STREAM_HELPER")
+
+HEADER = {"X-Riot-Token": KEY}

diff --git a/skgyorugo/lol_api/match_v5.py b/skgyorugo/lol_api/match_v5.py
new file mode 100644 (file)
index 0000000..5341c9f
--- /dev/null
+++ b/skgyorugo/lol_api/match_v5.py
@@ -0,0 +1,14 @@
+from lol_api import *
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+formatter = logging.Formatter("[%(levelname)s] %(asctime)s: %(name)s; %(message)s")
+
+file_handler = logging.FileHandler('/var/log/aptbot/logs.log')
+file_handler.setFormatter(formatter)
+
+logger.handlers = []
+logger.addHandler(file_handler)
+
+

diff --git a/skgyorugo/lol_api/spectator_v4.py b/skgyorugo/lol_api/spectator_v4.py
new file mode 100644 (file)
index 0000000..3724297
--- /dev/null
+++ b/skgyorugo/lol_api/spectator_v4.py
@@ -0,0 +1,113 @@
+from lol_api import *
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+formatter = logging.Formatter("[%(levelname)s] %(asctime)s: %(name)s; %(message)s")
+
+file_handler = logging.FileHandler('/var/log/aptbot/logs.log')
+file_handler.setFormatter(formatter)
+
+logger.handlers = []
+logger.addHandler(file_handler)
+
+
+@dataclass
+class BannedChampion:
+    pick_turn: int
+    champion_id: int
+    team_id: int
+
+@dataclass
+class Perks:
+    perk_ids: list[int]
+    perk_style: int
+    perk_sub_style: int
+
+@dataclass
+class CurrentGameParticipant:
+    champion_id: int
+    perks: Perks
+    profile_icon_id: int
+    bot: bool
+    team_id: int
+    summoner_name: str
+    summoner_id: str
+    spell1_id: int
+    spell2_id: int
+
+@dataclass
+class GameInfo:
+    game_id: int
+    game_type: str
+    game_start_time: int
+    map_id: int
+    game_length: int
+    platform_id: str
+    game_mode: str
+    banned_champions: list[BannedChampion]
+    game_queue_config_id: int
+    observers: str
+    participants: list[CurrentGameParticipant]
+
+def get_spectator_info_from_summoner_id(summoner_id: str) -> Optional[GameInfo]:
+    endpoint = f"/lol/spectator/v4/active-games/by-summoner/{summoner_id}"
+    url = BASE_URL + endpoint
+    http = urllib3.PoolManager()
+    r = http.request(
+        "GET",
+        url,
+        headers=HEADER,
+    )
+    if r.status == 404:
+        logger.info(f"Summoner with summoner id: {summoner_id} wasn't found in game. Status code {r.status}")
+        return None
+    if r.status != 200:
+        logger.warning(f"Couldn't retrieve summoner with summoner id: {summoner_id}. Status code {r.status}")
+        return None
+    data = json.loads(r.data.decode("utf-8"))
+
+    banned_champions: list[BannedChampion] = []
+    for banned in data["bannedChampions"]:
+        banned_champions.append(
+            BannedChampion(
+                banned["pickTurn"],
+                banned["championId"],
+                banned["teamId"],
+            )
+        )
+
+    participants: list[CurrentGameParticipant] = []
+    for participant in data["participants"]:
+        perks = Perks(
+            [perk_id for perk_id in participant["perks"]["perkIds"]],
+            participant["perks"]["perkStyle"],
+            participant["perks"]["perkSubStyle"]
+        )
+        participants.append(
+            CurrentGameParticipant(
+                participant["championId"],
+                perks,
+                participant["profileIconId"],
+                participant["bot"],
+                participant["teamId"],
+                participant["summonerName"],
+                participant["summonerId"],
+                participant["spell1Id"],
+                participant["spell2Id"],
+            )
+        )
+
+    return GameInfo(
+        data["gameId"],
+        data["gameType"],
+        data["gameStartTime"],
+        data["mapId"],
+        data["gameLength"],
+        data["platformId"],
+        data["gameMode"],
+        banned_champions,
+        data["gameQueueConfigId"],
+        data["observers"]["encryptionKey"],
+        participants,
+    )

diff --git a/skgyorugo/lol_api/summoner_v4.py b/skgyorugo/lol_api/summoner_v4.py
new file mode 100644 (file)
index 0000000..253d83b
--- /dev/null
+++ b/skgyorugo/lol_api/summoner_v4.py
@@ -0,0 +1,71 @@
+from lol_api import *
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+formatter = logging.Formatter("[%(levelname)s] %(asctime)s: %(name)s; %(message)s")
+
+file_handler = logging.FileHandler('/var/log/aptbot/logs.log')
+file_handler.setFormatter(formatter)
+
+logger.handlers = []
+logger.addHandler(file_handler)
+
+
+@dataclass
+class Summoner:
+    summoner_id: str
+    account_id: str
+    puuid: str
+    name: str
+    profile_icon_id: int
+    revision_date: int
+    summoner_level: int
+
+def get_summoner_from_puuid(puuid: str) -> Optional[Summoner]:
+    endpoint = f"/lol/summoner/v4/summoners/by-puuid/{puuid}"
+    url = BASE_URL + endpoint
+    http = urllib3.PoolManager()
+    r = http.request(
+        "GET",
+        url,
+        headers=HEADER,
+    )
+    if r.status != 200:
+        logger.warning(f"Couldn't retrieve summoner with puuid: {puuid}. Status code {r.status}")
+        return None
+    data = json.loads(r.data.decode("utf-8"))
+    return Summoner(
+        data["id"],
+        data["accountId"],
+        data["puuid"],
+        data["name"],
+        data["profileIconId"],
+        data["revisionDate"],
+        data["summonerLevel"]
+    )
+
+
+def get_summoner_from_name(name: str) -> Optional[Summoner]:
+    endpoint = f"/lol/summoner/v4/summoners/by-name/{name}"
+    url = BASE_URL + endpoint
+    http = urllib3.PoolManager()
+    r = http.request(
+        "GET",
+        url,
+        headers=HEADER,
+    )
+    if r.status != 200:
+        logger.warning(f"Couldn't retrieve summoner: {name}. Status code {r.status}")
+        return None
+    data = json.loads(r.data.decode("utf-8"))
+    return Summoner(
+        data["id"],
+        data["accountId"],
+        data["puuid"],
+        data["name"],
+        data["profileIconId"],
+        data["revisionDate"],
+        data["summonerLevel"]
+    )
+

diff --git a/skgyorugo/main.py b/skgyorugo/main.py
index 7a12aa78c66d35943692dff684d742820b7f3247..ba7189e4afa03bdc022ff5300e5fb86acc9d501b 100644 (file)
--- a/skgyorugo/main.py
+++ b/skgyorugo/main.py
@@ -38,11 +38,11 @@ logger.handlers = []
 logger.addHandler(file_handler)
 
 PATH = os.path.dirname(os.path.realpath(__file__))
-logger.info(f"main PATH set to: {PATH}")
+logger.info(f"PATH set to: {PATH}")
 COMMANDS_PATH = os.path.join(PATH, "commands")
-logger.info(f"main COMMANDS_PATH set to: {COMMANDS_PATH}")
+logger.info(f"COMMANDS_PATH set to: {COMMANDS_PATH}")
 AUTO_MESSAGES_PATH = os.path.join(PATH, "auto_messages")
-logger.info(f"main AUTO_MESSAGES_PATH set to: {AUTO_MESSAGES_PATH}")
+logger.info(f"AUTO_MESSAGES_PATH set to: {AUTO_MESSAGES_PATH}")
 
 commands_specs = {}
 commands_modules = {}