Skip to content

Commit 47d598f

Browse files
authored
Merge pull request #229 from Bot-detector/develop
release
2 parents 55946f6 + 2929b24 commit 47d598f

File tree

3 files changed

+17
-9
lines changed

3 files changed

+17
-9
lines changed

src/cogs/rsn_linking_commands.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ async def link(self, ctx: Context, *, name: str):
179179

180180
# register verification
181181
await config.api.post_discord_code(
182-
discord_id=ctx.author.id, player_name=player.get("name"), code=code
182+
discord_id=ctx.author.id, name=player.get("name"), code=code
183183
)
184184

185185
# send user via pm the random code

src/utils/bot_detector_api.py

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
import aiohttp
66
from src import config
7+
from src.utils.string_processing import to_jagex_name
78

89
logger = logging.getLogger(__name__)
910

@@ -198,6 +199,7 @@ async def create_player(self, name: str, debug: bool = False) -> None:
198199
None
199200
"""
200201
url = self.url + "/v1/player"
202+
name = to_jagex_name(name=name)
201203
data = await self._webrequest(
202204
url,
203205
type="post",
@@ -223,7 +225,7 @@ async def get_player(self, name: str, debug: bool = False) -> dict:
223225
"""
224226
url = self.url + "/v1/player"
225227
params = {
226-
"player_name": name,
228+
"player_name": to_jagex_name(name=name),
227229
"token": self.token,
228230
"row_count": 1,
229231
"page": 1,
@@ -234,7 +236,7 @@ async def get_player(self, name: str, debug: bool = False) -> dict:
234236
return data
235237

236238
# TODO: API design
237-
async def get_discord_player(self, runescape_name: str) -> List[dict]:
239+
async def get_discord_player(self, name: str) -> List[dict]:
238240
"""
239241
Get a player's Discord account by their RuneScape name.
240242
@@ -243,22 +245,23 @@ async def get_discord_player(self, runescape_name: str) -> List[dict]:
243245
player's Discord account information.
244246
245247
Args:
246-
runescape_name (str): The RuneScape name of the player to get.
248+
name (str): The RuneScape name of the player to get.
247249
248250
Returns:
249251
List[dict]: A list of dictionaries containing the player's Discord
250252
account information, or an empty list if the player was not found.
251253
"""
254+
name = to_jagex_name(name=name)
252255
url = (
253256
self.url
254-
+ f"/discord/verify/player_rsn_discord_account_status/{self.token}/{runescape_name}"
257+
+ f"/discord/verify/player_rsn_discord_account_status/{self.token}/{name}"
255258
)
256259
data = await self._webrequest(url, type="get")
257260
return data
258261

259262
# TODO: API design
260263
async def post_discord_code(
261-
self, discord_id: str, player_name: int, code: str
264+
self, discord_id: str, name: int, code: str
262265
) -> List[dict]:
263266
"""
264267
Post a Discord verification code for a player.
@@ -274,8 +277,9 @@ async def post_discord_code(
274277
Returns:
275278
None
276279
"""
280+
name = to_jagex_name(name=name)
277281
url = self.url + f"/discord/verify/insert_player_dpc/{self.token}"
278-
data = {"discord_id": discord_id, "player_name": player_name, "code": code}
282+
data = {"discord_id": discord_id, "player_name": name, "code": code}
279283
await self._webrequest(url, type="post", json=data)
280284

281285
# TODO: API design
@@ -374,6 +378,7 @@ async def get_prediction(self, player_name: str, breakdown: bool = True) -> dict
374378
player was not found.
375379
"""
376380
url = self.url + "/v1/prediction"
381+
player_name = to_jagex_name(name=player_name)
377382
params = {"name": player_name, "breakdown": int(breakdown)}
378383
data = await self._webrequest(url, type="get", params=params)
379384
return data
@@ -429,7 +434,7 @@ async def get_latest_sighting(self, name: str):
429434
if the player was not found.
430435
"""
431436
url = self.url + f"/discord/get_latest_sighting/{self.token}"
432-
params = {"player_name": name}
437+
params = {"player_name": to_jagex_name(name=name)}
433438
data = await self._webrequest(url, type="post", json=params)
434439
return data
435440

@@ -448,6 +453,6 @@ async def get_xp_gainz(self, name: str):
448453
if the player was not found.
449454
"""
450455
url = self.url + f"/discord/get_xp_gains/{self.token}"
451-
params = {"player_name": name}
456+
params = {"player_name": to_jagex_name(name=name)}
452457
data = await self._webrequest(url, type="post", json=params)
453458
return data

src/utils/string_processing.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,3 +20,6 @@ def plus_minus(var, compare):
2020
diff_control = "+"
2121

2222
return diff_control
23+
24+
def to_jagex_name(name: str) -> str:
25+
return name.lower().replace("_", " ").replace("-", " ").strip()

0 commit comments

Comments
 (0)