add deals balance parsing to TransactionsPage

Author: PythonPapochka

funpayparsers/parsers/page_parsers/transactions_page_parser.py

@@ -72,21 +72,29 @@ class TransactionsPageParser(
 
     def _parse(self) -> TransactionsPage:
         money_values = []
-        for i in self.tree.css('span.balances-value'):
+        balances = self.tree.css('span.balances-value')
+        for i in balances[:3]:
             money_values.append(
                 MoneyValueParser(
                     i.text().strip(),
                     options=self.options.money_value_parsing_options,
                     parsing_mode=MoneyValueParsingMode.FROM_STRING,
                 ).parse(),
             )
-            if len(money_values) == 3:
-                break
 
         rub_balance = [i for i in money_values if i.currency is Currency.RUB]
         usd_balance = [i for i in money_values if i.currency is Currency.USD]
         eur_balance = [i for i in money_values if i.currency is Currency.EUR]
 
+        deals_balance = None
+        if len(balances) > 3:
+            deals_text = balances[3].text().strip().split(' ', 1)[-1]
+            deals_balance = MoneyValueParser(
+                deals_text,
+                options=self.options.money_value_parsing_options,
+                parsing_mode=MoneyValueParsingMode.FROM_STRING,
+            ).parse()
+
         transactions_div = self.tree.css('div.tc-finance:not(.hidden)')
         if not transactions_div:
             transactions = None
@@ -109,5 +117,6 @@ def _parse(self) -> TransactionsPage:
             rub_balance=rub_balance[0] if rub_balance else None,
             usd_balance=usd_balance[0] if usd_balance else None,
             eur_balance=eur_balance[0] if eur_balance else None,
+            deals_balance=deals_balance,
             transactions=transactions,
         )

funpayparsers/types/pages/transactions_page.py

@@ -30,6 +30,9 @@ class TransactionsPage(FunPayPage):
     eur_balance: MoneyValue | None
     """EUR balance."""
 
+    deals_balance: MoneyValue | None
+    """Deals balance."""
+
     transactions: TransactionPreviewsBatch | None
     """Transaction previews."""
 

tests/parsers_test/money_value_parser_test.py

@@ -50,6 +50,20 @@
     character='₽'
 )
 
+deals_money_value_str_ru = """Сделки 1 282.95 ₽"""
+deals_money_value_obj_ru = MoneyValue(
+    raw_source='',
+    value=1282.95,
+    character='₽'
+)
+
+deals_money_value_str_en = """Deals 50.00 ₽"""
+deals_money_value_obj_en = MoneyValue(
+    raw_source='',
+    value=50.0,
+    character='₽'
+)
+
 
 def test_transaction_preview_money_value_parsing():
     options = MoneyValueParsingOptions(parsing_mode=MoneyValueParsingMode.FROM_TRANSACTION_PREVIEW)
@@ -80,3 +94,17 @@ def test_string_money_value_parsing():
     options = MoneyValueParsingOptions(parsing_mode=MoneyValueParsingMode.FROM_STRING)
     parser = MoneyValueParser(string_money_value_str, options=options & OPTIONS)
     assert parser.parse() == string_money_value_obj
+
+
+def test_deals_money_value_ru_parsing():
+    options = MoneyValueParsingOptions(parsing_mode=MoneyValueParsingMode.FROM_STRING)
+    raw = deals_money_value_str_ru.strip().split(' ', 1)[-1]
+    parser = MoneyValueParser(raw, options=options & OPTIONS)
+    assert parser.parse() == deals_money_value_obj_ru
+
+
+def test_deals_money_value_en_parsing():
+    options = MoneyValueParsingOptions(parsing_mode=MoneyValueParsingMode.FROM_STRING)
+    raw = deals_money_value_str_en.strip().split(' ', 1)[-1]
+    parser = MoneyValueParser(raw, options=options & OPTIONS)
+    assert parser.parse() == deals_money_value_obj_en