Skip to content

Commit a376125

Browse files
committed
fix sync requests, fix get_club() and get_member() for official api
1 parent cb47271 commit a376125

6 files changed

Lines changed: 14 additions & 14 deletions

File tree

brawlstats/brawlapi/core.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def _get_model(self, url, model, key=None):
204204
# Calls the async function
205205
return self._aget_model(url, model=model, key=key)
206206

207-
data, resp = self._request(url)
207+
data = self._request(url)
208208
if self.prevent_ratelimit:
209209
time.sleep(1 / self.ratelimit[0])
210210

@@ -214,9 +214,7 @@ def _get_model(self, url, model, key=None):
214214

215215
if model == Constants:
216216
if key:
217-
print(key)
218217
if data.get(key):
219-
print(f'valid key {key}')
220218
return model(self, data.get(key))
221219
else:
222220
raise KeyError('No such Constants key "{}"'.format(key))

brawlstats/brawlapi/models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from box import Box, BoxList
2+
from .utils import bstag
23

34

45
class BaseBox:
@@ -59,7 +60,7 @@ def get_club(self, full=True):
5960
if not self.club:
6061
return None
6162
if full:
62-
url = '{}?tag={}'.format(self.client.api.CLUB, self.club.tag)
63+
url = '{}?tag={}'.format(self.client.api.CLUB, bstag(self.club.tag))
6364
club = self.client._get_model(url, model=Club)
6465
else:
6566
club = PartialClub(self.client, self.club)

brawlstats/officialapi/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ def _get_model(self, url, model, key=None):
196196
# Calls the async function
197197
return self._aget_model(url, model=model, key=key)
198198

199-
data, resp = self._request(url)
199+
data = self._request(url)
200200
if self.prevent_ratelimit:
201201
time.sleep(1 / self.ratelimit[0])
202202

brawlstats/officialapi/models.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from box import Box, BoxList
2+
from .utils import bstag
23

34

45
class BaseBox:
@@ -53,7 +54,7 @@ def get_club(self):
5354
"""
5455
if not self.club:
5556
return None
56-
url = '{}?tag={}'.format(self.client.api.CLUB, self.club.tag)
57+
url = '{}/{}'.format(self.client.api.CLUB, bstag(self.club.tag))
5758
return self.client._get_model(url, model=Club)
5859

5960

@@ -74,8 +75,8 @@ def get_members(self):
7475
7576
Returns Members
7677
"""
77-
url = '{}/{}/members'.format(self.client.api.CLUB, self.tag)
78-
return self._get_model(url, model=Members)
78+
url = '{}/{}/members'.format(self.client.api.CLUB, bstag(self.tag))
79+
return self.client._get_model(url, model=Members)
7980

8081

8182
class Members(BaseBox):

tests/brawlapi/test_async.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -103,13 +103,13 @@ async def request():
103103
await self.client.get_leaderboard(invalid_type, invalid_limit)
104104
invalid_type = 'test'
105105
invalid_limit = 200
106-
self.assertRaises(ValueError, request)
106+
self.assertAsyncRaises(ValueError, request)
107107
invalid_type = 'players'
108108
invalid_limit = 201
109-
self.assertRaises(ValueError, request)
109+
self.assertAsyncRaises(ValueError, request)
110110
invalid_type = 'players'
111111
invalid_limit = -5
112-
self.assertRaises(ValueError, request)
112+
self.assertAsyncRaises(ValueError, request)
113113

114114

115115
if __name__ == '__main__':

tests/brawlapi/test_blocking.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,13 +98,13 @@ def test_invalid_lb(self):
9898
get_lb = self.client.get_leaderboard
9999
invalid_type = 'test'
100100
invalid_limit = 200
101-
self.assertRaises(ValueError, get_lb, invalid_type, invalid_limit)
101+
self.assertAsyncRaises(ValueError, get_lb, invalid_type, invalid_limit)
102102
invalid_type = 'players'
103103
invalid_limit = 201
104-
self.assertRaises(ValueError, get_lb, invalid_type, invalid_limit)
104+
self.assertAsyncRaises(ValueError, get_lb, invalid_type, invalid_limit)
105105
invalid_type = 'players'
106106
invalid_limit = -5
107-
self.assertRaises(ValueError, get_lb, invalid_type, invalid_limit)
107+
self.assertAsyncRaises(ValueError, get_lb, invalid_type, invalid_limit)
108108

109109

110110
if __name__ == '__main__':

0 commit comments

Comments
 (0)