tradezero_api.notification

 1from __future__ import annotations
 2
 3from selenium.webdriver.common.by import By
 4
 5from .time_helpers import Time
 6
 7
 8class Notification(Time):
 9    """A class for retrieving notifications from the web-app"""
10
11    def __init__(self, driver):
12        self.driver = driver
13
14    def get_last_notification_message(self):
15        """
16        return last notification message
17
18        :return: str
19        """
20        return self.driver.find_element(By.CSS_SELECTOR, 'span.message').text
21
22    def get_notifications(self, notif_amount: int = 1):
23        """
24        return a nested list with each sublist containing [time, title, message],
25        note that u can only view the amount of notifications that are visible in the box/widget
26        without scrolling down (which usually is around 6-9 depending on each message length)
27        example of nested list: (see the docs for a better look):
28        [['11:34:49', 'Order canceled', 'Your Limit Buy order of 1 AMD was canceled.'],
29        ['11:23:34', 'Level 2', 'You are not authorized for symbol: AMD'],
30        ['11:23:34', 'Error', 'You are not authorized for symbol: AMD']].
31
32        :param notif_amount: int amount of notifications to retrieve sorted by most recent
33        :return: nested list
34        """
35        notif_lst = self.driver.find_elements(By.XPATH,
36                                              '//*[@id="notifications-list-1"]/li')
37        notif_lst_text = [x.text.split('\n') for x in notif_lst[0:notif_amount] if x.text != '']
38
39        notifications = []
40        for (notification, i) in zip(notif_lst_text, range(notif_amount)):
41            if len(notification) == 2:
42                notification.insert(0, str(self.time))
43
44            elif notification[0] == '' or notification[0] == '-':
45                notification[0] = str(self.time)
46
47            notifications.append(notification)
48        return notifications
49    
50    def notifications_generator(self):
51        """
52        A notification generator, similarly to get_notifications(), this yields one notification at a time,
53        on each next() it will yield a list like so: 
54        ['11:34:49', 'Order canceled', 'Your Limit Buy order of 1 AMD was canceled.']
55        You can think of it as a more dynamic version of get_notifications(), on each next(); it gets 
56        one notification message.
57
58        :return: list
59        """
60        notif_lst = self.driver.find_elements(By.XPATH, '//*[@id="notifications-list-1"]/li')
61        for item in notif_lst:
62            if item.text == '':
63                continue
64                
65            notification = item.text.split('\n')
66            if len(notification) == 2:
67                notification.insert(0, str(self.time))
68
69            elif notification[0] == '' or notification[0] == '-':
70                notification[0] = str(self.time)
71
72            yield notification