add support for multiple clan requests

Author: fourjr

crasync/core.py

@@ -1,6 +1,6 @@
 import aiohttp
 import asyncio
-from .models import Profile, Clan, Constants
+from .models import Profile, Clan, Clans, Constants
 
 
 class Client:
@@ -33,20 +33,36 @@ async def get_profile(self, tag):
 
         return Profile(self, data)
 
-
-    async def get_clan(self, tag):
-        '''Get a clan object using a tag'''
-
-        url = f'{self.BASE}/clan/{tag}'
-
+    async def get_clans(self, *, tags):
+        '''Get a clan object using tag(s)'''
+        tags = tags.split(',')
+
+        if len(tags) == 1:
+            url = f'{self.BASE}/clan/{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 Clan(self, data)
+
+        taglist = ''
+        i = 0
+        for tag in tags:
+            i += 1
+            taglist += tag[1:]
+            if i != len(tags):
+                taglist += ','
+
+        url = f'{self.BASE}/clan/{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 Clan(self, data)
+        return Clans(self, data)
 
     async def get_constants(self):
         '''Get a profile object using a tag.'''

crasync/models.py

@@ -9,23 +9,23 @@ def __init__(self, client, data):
         self.raw_data = data
         self.name = data.get('name')
         self.tag = data.get('tag')
-        #self.from_data(data)
+        self.from_data(data)
         endpoint = type(self).__name__.lower()
         self.url = f'{client.BASE}/{endpoint}/{self.tag}'
 
     def __str__(self):
         return f'{self.name} (#{self.tag})'
 
-    #async def from_data(self):
-        #raise NotImplementedError
+    async def from_data(self):
+        raise NotImplemented
 
     async def update(self):
 
         async with self.client.session.get(self.url) as resp:
             data = await resp.json()
 
         self.raw_data = data
-        #self.from_data(data)
+        self.from_data(data)
 
         return self
 
@@ -173,6 +173,10 @@ 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.
     Includes a clan maybe? (requires a seperate request tho)