Orders parser (funpayhub#21)

Author: qvvonk

funpayparsers/parsers/__init__.py

@@ -2,3 +2,4 @@
 from .chat_preview_parser import *
 from .messages_parser import *
 from .badge_parser import *
+from .order_previews_parser import *

funpayparsers/parsers/order_previews_parser.py

@@ -0,0 +1,77 @@
+__all__ = ('OrderPreviewsParserOptions', 'OrderPreviewsParser', )
+
+from dataclasses import dataclass
+
+from funpayparsers.parsers.base import FunPayObjectParserOptions, FunPayObjectParser
+from funpayparsers.parsers.utils import extract_css_url
+from funpayparsers.types.orders import OrderPreview, OrderCounterpartyInfo
+from funpayparsers.types.enums import OrderStatus
+from funpayparsers.types.common import MoneyValue
+
+from lxml import html
+
+
+@dataclass(frozen=True)
+class OrderPreviewsParserOptions(FunPayObjectParserOptions):
+    ...
+
+
+class OrderPreviewsParser(FunPayObjectParser[
+                              list[OrderPreview],
+                              OrderPreviewsParserOptions
+                          ]):
+    options_cls: OrderPreviewsParserOptions = OrderPreviewsParserOptions
+
+    def _parse(self):
+        result = []
+
+        for o in self.tree.xpath('//a[contains(@class, "tc-item")]'):
+            order_id: str = o.get('href').split('/')[-2]
+            date: str = o.xpath('string(.//div[@class="tc-date-time"][1])')
+
+            desc = o.xpath('string(.//div[@class="order-desc"][1]/div[1])')
+            category_desc = o.xpath('string(.//div[@class="text-muted"][1])')
+
+            status_class = o.xpath('string(.//div[contains(@class, "tc-status")][1]/@class)')
+            status = OrderStatus.get_by_css_class(status_class)
+
+            val_str = o.xpath('string(.//div[contains(@class, "tc-price")])').strip()
+            val, char = val_str.split()
+            value = MoneyValue(raw_source=val_str, value=float(val), character=char)
+
+            user_tag = o.xpath('.//div[contains(@class, "media-user")][1]')[0]
+            photo_style = o.xpath(
+                'string(.//div[contains(@class, "avatar-photo")]/@style)')
+            nickname_tag = user_tag.xpath('.//div[@class="media-user-name"]/span[1]')[0]
+            user_nickname = nickname_tag.text
+            user_id = int(nickname_tag.get('data-href').split('/')[-2])
+            online = 'online' in user_tag.get('class')
+            banned = 'banned' in user_tag.get('class')
+            user_status_text: str = user_tag.xpath(
+                'string(.//div[contains(@class, "media-user-status")][1])'
+            )
+            photo_url = extract_css_url(photo_style)
+
+            counterparty = OrderCounterpartyInfo(
+                raw_source=html.tostring(user_tag, encoding='unicode'),
+                id=user_id,
+                username=user_nickname,
+                online=online,
+                avatar_url=photo_url,
+                banned=banned,
+                status_text=user_status_text
+            )
+
+            order_obj = OrderPreview(
+                raw_source=html.tostring(o, encoding='unicode'),
+                id=order_id,
+                date_text=date,
+                desc=desc,
+                category_text=category_desc,
+                status=status,
+                amount=value,
+                counterparty=counterparty
+            )
+            result.append(order_obj)
+
+        return result

funpayparsers/types/common.py

@@ -21,12 +21,16 @@ class MoneyValue(FunPayObject):
     value: int | float
     """The numeric amount of the monetary value."""
 
-    currency: Currency
-    """The currency of the value."""
-
     character: str
     """The currency character, e.g., $, €, ₽, ¤, etc."""
 
+    @property
+    def currency(self) -> Currency:
+        return Currency.get_by_character(self.character)
+
+    def __str__(self):
+        return f'<MoneyValue: {self.value} {self.currency.name}>'
+
 
 @dataclass
 class UserBadge(FunPayObject):

funpayparsers/types/orders.py

@@ -26,15 +26,18 @@ class OrderCounterpartyInfo(FunPayObject):
     online: bool
     """True, if counterparty is online."""
 
-    blocked: bool
-    """True, if counterparty is blocked."""
+    banned: bool
+    """True, if counterparty is banned."""
 
-    last_online_text: str | None
-    """Last online status text (if available)."""
+    status_text: str
+    """Status text (online / banned / last seen online)."""
 
     avatar_url: str
     """Counterpart avatar URL."""
 
+    def __str__(self):
+        return f'{self.username} (id: {self.id}), status: {self.status_text}.'
+
 
 @dataclass
 class OrderPreview(FunPayObject):
@@ -49,8 +52,8 @@ class OrderPreview(FunPayObject):
     date_text: str
     """Order date (as human-readable text)."""
 
-    desc: str | None
-    """Order description (if available)."""
+    desc: str
+    """Order description."""
 
     category_text: str
     """Order category and subcategory text."""
@@ -63,3 +66,7 @@ class OrderPreview(FunPayObject):
 
     counterparty: OrderCounterpartyInfo
     """Associated counterparty info."""
+
+    def __str__(self):
+        return (f'<{self.amount} {self.status.name} order {self.id} '
+                f'dated {self.date_text}: {self.desc}. ({self.category_text})>')

tests/orders_preview_parser_text.py

@@ -0,0 +1,66 @@
+from funpayparsers.parsers.order_previews_parser import (OrderPreviewsParser,
+                                                         OrderPreviewsParserOptions)
+from funpayparsers.types.orders import OrderPreview, OrderCounterpartyInfo
+from funpayparsers.types.enums import OrderStatus
+from funpayparsers.types.common import MoneyValue
+
+
+OPTIONS = OrderPreviewsParserOptions(empty_raw_source=True)
+
+
+refunded_order_html = """
+<a href="https://funpay.com/orders/ABCDEFGH/" class="tc-item warning">
+    <div class="tc-date">
+        <div class="tc-date-time">вчера, 13:33</div>
+        <div class="tc-date-left">22 часа назад</div>
+    </div>
+    <div class="tc-order">#ABCDEFGH</div>
+    <div class="order-desc">
+        <div>Order Description</div>
+        <div class="text-muted">Category, Subcategory</div>
+    </div>
+    <div class="tc-user">
+        <div class="media media-user offline">
+            <div class="media-left">
+                <div class="avatar-photo pseudo-a" tabindex="0" data-href="https://funpay.com/users/123456/" style="background-image: url(path/to/avatar);"></div>
+            </div>
+            <div class="media-body">
+                <div class="media-user-name">
+                    <span class="pseudo-a" tabindex="0" data-href="https://funpay.com/users/123456/">Counterparty username</span>
+                </div>
+                <div class="media-user-status">Counterparty Status</div>
+            </div>
+        </div>
+    </div>
+    <div class="tc-status text-warning">Возврат</div>
+    <div class="tc-price text-nowrap tc-buyer-sum">25.12 <span class="unit">₽</span></div>
+</a>
+"""
+
+refunded_order_obj = OrderPreview(
+    raw_source='',
+    id='ABCDEFGH',
+    date_text='вчера, 13:33',
+    desc='Order Description',
+    category_text='Category, Subcategory',
+    status=OrderStatus.REFUNDED,
+    amount=MoneyValue(
+        raw_source='',
+        value=25.12,
+        character='₽'
+    ),
+    counterparty=OrderCounterpartyInfo(
+        raw_source='',
+        id=123456,
+        username='Counterparty username',
+        online=False,
+        banned=False,
+        status_text='Counterparty Status',
+        avatar_url='path/to/avatar'
+    )
+)
+
+
+def test_refunded_order_parsing():
+    parser = OrderPreviewsParser(refunded_order_html, options=OPTIONS)
+    assert parser.parse() == [refunded_order_obj]