PySloth/event_maker.py at master · muhammetenes/PySloth · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
import sys
import time

import kthread
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver import ActionChains
from selenium.webdriver.common.keys import Keys

from logger import Logger


class EventMaker:

    driver = None

    def __init__(self, driver, selenium_helper=None):
        self.driver = driver
        self.selenium_helper = selenium_helper
        pass

    def push_event(self, driver, event):
        if driver is not None:

            try:
                if 'selector' in event:
                    element = driver.find_element_by_xpath(event['selector'])
                else:
                    element = driver

                if element is not None:
                    if 'delay' in event:
                        time.sleep(event['delay'])

                    for action in event['actions']:
                        if 'delay' in action:
                            time.sleep(action['delay'])

                        switcher = {
                            "click": self.set_click,
                            "scroll": self.set_scroll,
                            "style": self.set_style,
                            "excute_script": self.set_excute_script,
                            "nothing": lambda: self.nothing,
                        }

                        func = switcher.get(action['type'], lambda: "nothing")
                        func(element, action)

                    if 'sleep' in event:
                        time.sleep(event['sleep'])

            except NoSuchElementException as e:
                print("-- NoSuchElementException: " + str(e))

            except Exception as e:
                print("Push Event ->: " + str(e))

        else:
            return None

    def push_event_to_element(self, element, events):
        try:
            for event in events:
                if 'delay' in event:
                    time.sleep(event['delay'])
                if 'selector' in event:
                    element = element.find_element_by_xpath(event['selector'])

                for action in event['actions']:
                    if 'delay' in action:
                        time.sleep(action['delay'])

                    type = action['type']
                    if type == 'click':
                        self.set_click(element, action)
                    elif type == 'click_action_element':
                        self.set_click_action_element(self.driver, action)
                    elif type == 'click_with_command':
                        self.set_click_with_command(element, action)
                    elif type == 'scroll':
                        self.set_scroll(element, action)
                    elif type == 'style':
                        self.set_style(element, action)
                    elif type == 'set_attr':
                        self.set_attr(element, action)
                    elif type == 'excute_script':
                        self.set_excute_script(element, action)
                    elif type == 'download':
                        thread = kthread.KThread(target=self.selenium_helper.download_loop,
                                                 args=(element, action))
                        thread.start()

                if 'sleep' in event:
                    time.sleep(event['sleep'])

        except NoSuchElementException as e:
            print("-- NoSuchElementException: " + str(e))

        except Exception as e:
            type, value, traceback = sys.exc_info()
            if hasattr(value, 'filename'):
                print('Error %s: %s' % (value.filename, value.strerror))
                Logger().set_error_log('Error %s: %s' % (value.filename, value.strerror))

            print("Push Event To Element () ->: " + str(e))

    def set_click_with_command(self, element, action):
        action = ActionChains(self.driver).key_down(Keys.CONTROL)
        action.move_to_element(element)
        action.click()
        action.perform()

    def set_click_action_element(self, element, action):
        if 'selector' in action:
            element = element.find_element_by_xpath(action['selector'])
            element.click()

    def set_click(self, element, action):
        element.click()

    def set_scroll(self):
        return ""

    def set_style(self, element, action):
        scriptSetAttrValue = "arguments[0].setAttribute(arguments[1],arguments[2])"
        self.driver.execute_script(scriptSetAttrValue, element, 'style', action['style'])

    def set_attr(self, element, action):
        scriptSetAttrValue = "arguments[0].setAttribute(arguments[1],arguments[2])"
        self.driver.execute_script(scriptSetAttrValue, element, action['attr'], action['attr_value'])

    def set_excute_script(self, element, action):
        self.driver.execute_script(action['script'])

    def download(self):
        pass

    def nothing(self, element, action):
        return ""