From 2e88bc879cc87f9bc8121abfcb9ca501bf4dd2d4 Mon Sep 17 00:00:00 2001
From: Stefano Pigozzi <ste.pigozzi@gmail.com>
Date: Thu, 9 Jul 2020 04:13:54 +0200
Subject: [PATCH] Add more features to steam

---
 royalpack/commands/steampowered.py | 24 ++++++++++++++++++++++--
 royalpack/tables/steam.py          | 16 ++++++++++++++++
 2 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/royalpack/commands/steampowered.py b/royalpack/commands/steampowered.py
index 81275c09..05c4c3be 100644
--- a/royalpack/commands/steampowered.py
+++ b/royalpack/commands/steampowered.py
@@ -23,8 +23,12 @@ class SteampoweredCommand(rc.Command):
 
     @staticmethod
     def _display(account: Steam):
-        string = f"ℹ️ [b]{account.persona_name}[/b]\n" \
-                 f"{account.profile_url}\n" \
+        string = f"ℹ️ [url={account.profile_url}]{account.persona_name}[/url]\n" \
+                 f"[b]Level {account.account_level}[/b]\n" \
+                 f"\n" \
+                 f"Owned games: [b]{account.owned_games_count}[/b]\n" \
+                 f"Most played 2 weeks: [url=https://store.steampowered.com/app/{account.most_played_game_2weeks}]{account.most_played_game_2weeks}[/url]\n" \
+                 f"Most played forever: [url=https://store.steampowered.com/app/{account.most_played_game_forever}]{account.most_played_game_forever}[/url]\n" \
                  f"\n" \
                  f"SteamID: [c]{account.steamid.as_32}[/c]\n" \
                  f"SteamID2: [c]{account.steamid.as_steam2}[/c]\n" \
@@ -50,6 +54,22 @@ class SteampoweredCommand(rc.Command):
         account.primary_clan_id = r["primaryclanid"]
         account.account_creation_date = datetime.datetime.fromtimestamp(r["timecreated"])
 
+        # noinspection PyProtectedMember
+        response = await self._call(self._api.IPlayerService.GetSteamLevel_v1, steamid=account._steamid)
+        account.account_level = response["response"]["player_level"]
+
+        # noinspection PyProtectedMember
+        response = await self._call(self._api.IPlayerService.GetOwnedGames_v1,
+                                    steamid=account._steamid,
+                                    include_appinfo=False,
+                                    include_played_free_games=True,
+                                    include_free_sub=False,
+                                    appids_filter=None)
+        account.owned_games_count = response["response"]["game_count"]
+        if response["response"]["game_count"] >= 0:
+            account.most_played_game_2weeks = sorted(response["response"]["games"], key=lambda g: -g.get("playtime_2weeks", 0))[0]["appid"]
+            account.most_played_game_forever = sorted(response["response"]["games"], key=lambda g: -g.get("playtime_forever", 0))[0]["appid"]
+
     async def run(self, args: rc.CommandArgs, data: rc.CommandData) -> None:
         author = await data.get_author()
         if len(args) > 0:
diff --git a/royalpack/tables/steam.py b/royalpack/tables/steam.py
index bbc0d4f3..3e2b2447 100644
--- a/royalpack/tables/steam.py
+++ b/royalpack/tables/steam.py
@@ -43,6 +43,22 @@ class Steam:
     def account_creation_date(self):
         return Column(DateTime)
 
+    @declared_attr
+    def account_level(self):
+        return Column(Integer, nullable=False, default=0)
+
+    @declared_attr
+    def owned_games_count(self):
+        return Column(Integer, nullable=False, default=0)
+
+    @declared_attr
+    def most_played_game_2weeks(self):
+        return Column(Integer, nullable=False, default=753)
+
+    @declared_attr
+    def most_played_game_forever(self):
+        return Column(Integer, nullable=False, default=753)
+
     def json(self):
         return {
             "steamid2": self.steamid.as_steam2,