add support for multiple profiles

Author: fourjr

crasync/core.py

@@ -1,6 +1,6 @@
 import aiohttp
 import asyncio
-from .models import Profile, Clan, Clans, Constants
+from .models import Profile, Profiles, Clan, Clans, Constants
 
 
 class Client:
@@ -19,19 +19,38 @@ def __init__(self, session=None):
         self.session = session or aiohttp.ClientSession()
 
 
-    async def get_profile(self, tag):
-        '''Get a profile object using a tag.'''
+    async def get_profile(self, *, tags):
+        '''Get a profile object using tag(s).'''
+        tags = tags.split(',')
+
+        if len(tags) == 1:
+            url = f'{self.BASE}/profile/{tags[0]}'
+
+            async with self.session.get(url) as resp:
+                if resp.status == 200:
+                    data = await resp.json()
+                else:
+                    print('API is down. Please be patient.')
+                    return None
+
+            return Profile(self, data)
 
-        url = f'{self.BASE}/profile/{tag}'
+        taglist = ''
+        i = 0
+        for tag in tags:
+            i += 1
+            taglist += tag[1:]
+            if i != len(tags):
+                taglist += ','
 
+        url = f'{self.BASE}/profile/{taglist}'
         async with self.session.get(url) as resp:
             if resp.status == 200:
                 data = await resp.json()
             else:
                 print('API is down. Please be patient.')
                 return None
-
-        return Profile(self, data)
+        return Profiles(self, data)
 
     async def get_clans(self, *, tags):
         '''Get a clan object using tag(s)'''

crasync/models.py

@@ -172,10 +172,6 @@ def badge_url(self):
 
     def __repr__(self):
         return f'<Clan tag={self.tag}>'
-
-class Clans(Base):
-    def from_data(self, data):
-        self.clans = [Clan(self.client, c) for c in data]
 
 class Profile(Base):
     '''Represents a player profile.
@@ -226,6 +222,14 @@ def __repr__(self):
     def get_clan(self):
         return self.client.get_clan(self.clan_tag)
 
+class Clans(Base):
+    def from_data(self, data):
+        self.clans = [Clan(self.client, c) for c in data]
+
+class Profiles(Base):
+    def from_data(self, data):
+        self.clans = [Profile(self.client, c) for c in data]
+
 class Constants(Base):
     '''Represents the constants from cr-api'''
     def from_data(self, data):