oli/PoeTrade
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fixed an issue with nrItems

Browse Source
This commit is contained in:
Oliver Hartmann 2023-01-03 21:57:16 +01:00
parent d24d53116f
commit bc520acae1
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/data.py
View File

@ -50,7 +50,7 @@ channel_mapping = {'#': Channel.GLOBAL,
class Trade(): class Trade():
def __init__(self, def __init__(self,
nrItems: str, nrItems: int,
item: str, item: str,
amount: float, amount: float,
currency: str, currency: str,
@ -69,7 +69,12 @@ class Trade():
@classmethod @classmethod
def by_regex_result(cls, res): def by_regex_result(cls, res):
return cls(nrItems=res['nrItems'], if 'nrItems' in res.groupdict():
nrItems = int(res['nrItems'])
else:
# If no number is specified we only search for one item
nrItems = 1
return cls(nrItems=nrItems,
item=res['item'], item=res['item'],
amount=int(res['amount']), amount=int(res['amount']),
currency=res['currency'], currency=res['currency'],
@ -130,11 +135,4 @@ class Message():
def parse_trade(self) -> None: def parse_trade(self) -> None:
res = re_trade.search(self.message) res = re_trade.search(self.message)
if res: if res:
self.trade = Trade(nrItems=res['nrItems'], self.trade = Trade.by_regex_result(res)
item=res['item'],
amount=float(res['amount']),
currency=res['currency'],
tab=res['tab'],
row=res['row'],
col=res['col'],
league=res['league'])