moswar_bot/restore_hp.py at master · gil9red/moswar_bot · 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
#!/usr/bin/env python
# -*- coding: utf-8 -*-

__author__ = 'ipetrash'


from PySide.QtCore import QObject, Signal, QTimer, QEventLoop
from common import get_logger


logger = get_logger('restore_hp')


class RestoreHP(QObject):
    def __init__(self, mw):
        super().__init__()

        self.mw = mw
        self._restore_hp_window = None

        # Таймер для поиска диалога восстановления жизней
        self._timer = QTimer()
        self._timer.setInterval(333)
        self._timer.timeout.connect(self._find_restore_hp_window)

        # Таймер ожидания восстановления жизней
        self._timer_check_max_hp = QTimer()
        self._timer_check_max_hp.setInterval(333)
        self._timer_check_max_hp.timeout.connect(self._tick_check_max_hp)

    _find_restore_hp_window_finded = Signal()
    _hp_maximum = Signal()

    def _find_restore_hp_window(self):
        """Функция для поиска диалогового окна восстановления жизней."""

        for el in self.mw.doc.findAll('.alert'):
            title = el.findFirst('#alert-title')
            if not title.isNull() and title.toPlainText() == 'Восстановить здоровье':
                logger.debug('Найдено окно восстановления жизней.')

                self._restore_hp_window = el
                self._find_restore_hp_window_finded.emit()
                self._timer.stop()
                return

    def _tick_check_max_hp(self):
        """Функция для проверки максимума жизней. Если жизни максимальны, отправляется сигнал _hp_maximum."""

        if self.mw.current_hp() >= self.mw.max_hp():
            self._timer_check_max_hp.stop()
            self._hp_maximum.emit()

    def run(self):
        """Функция восставливления жизней."""

        logger.debug('Выполняю восстановление жизней.')

        if self.mw.current_hp() == self.mw.max_hp():
            logger.info('Жизни полные!')
            return

        self.mw.click_tag(".life .plus-icon")

        self._timer.start()

        loop = QEventLoop()
        self._find_restore_hp_window_finded.connect(loop.quit)
        loop.exec_()

        logger.debug('Нажимаю на кнопку "Вылечиться".')

        button = self._restore_hp_window.findFirst('button')
        button.evaluateJavaScript('this.click()')

        # Ждем пока жизни полностью восстановятся -- иначе может случится так, что
        # бот нападет в тот момент, когда жизни еще регенятся, и тогда не получится напасть и
        # бот дальше фазы выбора противника не продвинется
        if self.mw.current_hp() < self.mw.max_hp():
            logger.info('Жизни (%s/%s) не полные -- жду восстановления.', self.mw.current_hp(), self.mw.max_hp())

            self._timer_check_max_hp.start()
            self._hp_maximum.connect(loop.quit)
            loop.exec_()

            logger.info('Жизни восстановлены.')