main.py

# coding=utf-8
from pytg.sender import Sender
from pytg.receiver import Receiver
from pytg.utils import coroutine
from collections import deque
from time import time, sleep
from getopt import getopt
import sys
import datetime
import re
import _thread
import random

# username игрового бота
bot_username = 'ChatWarsBot'
# ваш username или username человека, который может отправлять запросы этому скрипту
admin_username = 'AlexanderSvetly'
# username бота и/или человека, которые будут отправлять приказы
order_usernames = 'RedStatBot'
# username бота и/или человека которому отправлять репорты о битвах
report_username = 'AlexanderSvetly'
# имя замка
castle_name = 'red'
# бот только принимает приказы от admin_username
sleeping_hours = [0, 8]

# путь к сокет файлу
socket_path = ''
# хост чтоб слушать telegram-cli
host = 'localhost'
# порт по которому сшулать
port = 1338

# opts, args = getopt(sys.argv[1:], 'a:o:c:s:h:p', ['admin=', 'order=', 'castle=', 'socket=', 'host=', 'port='])

# for opt, arg in opts:
#     if opt in ('-a', '--admin'):
#         admin_username = arg
#     elif opt in ('-o', '--order'):
#         order_usernames = arg.split(',')
#     elif opt in ('-c', '--castle'):
#         castle_name = arg
#     elif opt in ('-s', '--socket'):
#         socket_path = arg
#     elif opt in ('-h', '--host'):
#         host = arg
#     elif opt in ('-p', '--port'):
#         port = int(arg)

orders = {
    'red':    '🇮🇲',
    'black':  '🇬🇵',
    'white':  '🇨🇾',
    'yellow': '🇻🇦',
    'blue':   '🇪🇺',

    'forest_fort':   '🌲Лесной форт',
    'mountain_fort': '⛰Горный форт',

    'forest':  '🌲Лес',
    'cave':    '🕸Пещера',
    'korovan': '🐫Грабить Корованы',

    'def_korovan': '/go',
    'defence': '🛡 Защита',
    'attack':  '⚔ Атака',
    'hero': '🏅Герой',
    'searching_opponent': '🔎Поиск соперника'
}

symbols = {
    'mountain': '⛰',
    'forest':   '🌲',
    'defence':  '🛡',
    'attack':   '⚔'
}

arena_defence  = ['🛡головы', '🛡корпуса', '🛡ног']
arena_attack = ['🗡в голову', '🗡по корпусу', '🗡по ногам']

# castle = orders[castle_name]
# текущий приказ на атаку/защиту, по умолчанию всегда защита, трогать не нужно
current_order = {'time': 0, 'order': castle_name}

sender = Sender(sock=socket_path) if socket_path else Sender(host=host, port=port)
action_list = deque([])
log_list = deque([], maxlen=30)
last_arena_visit = 0
get_info_diff = 5000
hero_message_id = ''

bot_enabled         = True
arena_enabled       = True
forest_enabled      = False
cave_enabled        = True
korovan_enabled     = False
korovan_def_enabled = True
order_enabled       = True
auto_def_enabled    = True
auto_level_up       = True

auto_buy_enabled    = True
auto_by_gold_limit  = 103
auto_by_item        = '/buy_dagger2'

need_order          = False
time_for_order      = 0


@coroutine
def work_with_message(receiver):
    while True:
        msg = (yield)
        try:
            if msg['event'] == 'message' and 'text' in msg and msg['peer'] is not None:
                parse_text(msg['text'], msg['sender']['username'], msg['id'])
        except Exception as err:
            log('Ошибка coroutine: {0}'.format(err))


def queue_worker():
    global get_info_diff
    lt_info = 0
    print(lt_info)
    while True:
        try:
            #if time() - last_command_time > time_between_commands:
            #last_command_time = time()
            if time() - lt_info > get_info_diff and not is_slpeeping_time():
                lt_info = time()
                print('flag3')
                print(action_list)
                get_info_diff = random.randint(3600, 5000)
                send_msg(bot_username, orders['hero'])
                continue

            if len(action_list):
                log('Отправляем ' + action_list[0])
                send_msg(bot_username, action_list.popleft())
            # не ставить большое, иначе арена не будет работать
            sleep_time = random.randint(2, 8)
            sleep(sleep_time)
        except Exception as err:
            log('Ошибка очереди: {0}'.format(err))

def is_slpeeping_time() -> bool:
    global sleeping_hours
    cur_hour = datetime.datetime.now().hour
    return sleeping_hours[0] <= cur_hour and cur_hour <= sleeping_hours[1]

def parse_text(text, username, message_id):
    global last_arena_visit
    global hero_message_id
    global bot_enabled
    global order_enabled

    global forest_enabled
    global cave_enabled
    global korovan_enabled
    
    global auto_level_up
    global korovan_def_enabled
    global arena_enabled
    global auto_def_enabled

    global auto_buy_enabled
    global auto_by_gold_limit
    global auto_by_item

    global need_order
    global time_for_order
    global report_username

    if not bot_enabled:
        return

    if need_order and time() > time_for_order:
        action_list.append('/report')
        need_order = False

    if username == bot_username and not is_slpeeping_time():
        log('Получили сообщение от бота. Проверяем условия')

        if text.find('Твои результаты в бою') != -1:
            fwd(report_username, message_id)

        # TO DO: level up
        if auto_level_up and text.find('/level_up') != -1:
            action_list.append('+1 ⚔Атака')

        # защита корована
        if korovan_def_enabled and text.find('/go') != -1:
            action_list.append(orders['def_korovan'])

        if (orders['def_korovan'] in action_list) and (time() - current_order['time'] < 3600):
            update_order(current_order['order'])
        elif text.find('Битва пяти замков через') != -1:
            hero_message_id = message_id
            m = re.search('Битва пяти замков через(?: ([0-9]+)ч){0,1}(?: ([0-9]+)){0,1}', text)
            if not m.group(1):
                if m.group(2) and int(m.group(2)) <= 59:
                    # send_msg(admin_username, 'До битвы ' + m.group(2) + ' минут(ы)!')
                    # прекращаем все действия
                    state = re.search('Состояние:\\n(.*)$', text)
                    if auto_def_enabled and time() - current_order['time'] > 3600:
                        update_order(orders[castle_name])
                    return
            log('Времени достаточно')
            # теперь узнаем, сколько у нас выносливости и золота
            # m = re.search('Золото: (-*[0-9]+)\\n.*Выносливость: ([0-9]+) из', text)
            gold = int(re.search('💰([0-9]+)', text).group(1))
            stamina = int(re.search('Выносливость: ([0-9]+)', text).group(1))
            log('Золото: {0}, выносливость: {1}'.format(gold, stamina))
            if auto_buy_enabled and gold >= auto_by_gold_limit:
                action_list.append(auto_by_item)
            if forest_enabled and stamina > 0 and orders['forest'] not in action_list:
                action_list.append(orders['forest'])
            elif cave_enabled and stamina > 1 and orders['cave'] not in action_list:
                action_list.append(orders['cave'])
            elif korovan_enabled and stamina > 1 and orders['korovan'] not in action_list:
                action_list.append(orders['korovan'])
            elif (arena_enabled and gold >= 5) and (orders['searching_opponent'] not in action_list) and (time() - last_arena_visit > 3600):
                print('flag1. action_list.append(orders[searching_opponent])')
                action_list.append(orders['searching_opponent'])

        elif arena_enabled and text.find('выбери точку атаки и точку защиты') != -1:
            last_arena_visit = time()
            attack_chosen = arena_attack[random.randint(0, 2)]
            defence_chosen  = arena_defence[random.randint(0, 2)]
            log('Атака: {0}, Защита: {1}'.format(attack_chosen, defence_chosen))
            action_list.append(attack_chosen)
            action_list.append(defence_chosen)
            print('flag2. action_list.append(attack_chosen)')

    else:
        if order_enabled and username in order_usernames:
            if text.find(orders['red']) != -1:
                update_order(orders['red'])
                need_order = True
                time_for_order = time() + 3600
            elif text.find(orders['black']) != -1:
                update_order(orders['black'])
                need_order = True
                time_for_order = time() + 3600
            elif text.find(orders['white']) != -1:
                update_order(orders['white'])
                need_order = True
                time_for_order = time() + 3600
            elif text.find(orders['yellow']) != -1:
                update_order(orders['yellow'])
                need_order = True
                time_for_order = time() + 3600
            elif text.find(orders['blue']) != -1:
                update_order(orders['blue'])
                need_order = True
                time_for_order = time() + 3600

            elif text.find(symbols['forest']) != -1:
                update_order(orders['forest_fort'])
                need_order = True
                time_for_order = time() + 3600
            elif text.find(symbols['mountain']) != -1:
                update_order(orders['mountain_fort'])
                need_order = True
                time_for_order = time() + 3600
            elif text.find(symbols['defence']) != -1:
                update_order(orders[castle_name])
                need_order = True
                time_for_order = time() + 3600

            # need_order = true
            # time_for_order = time() + 60*15

            # send_msg(admin_username, 'Получили команду ' + current_order['order'] + ' от ' + username)

        if username == admin_username:
            if text == '#help':
                send_msg(admin_username, '\n'.join([
                    '#enable_bot - Включить бота',
                    '#disable_bot - Выключить бота',
                    '#enable_arena - Включить арену',
                    '#disable_arena - Выключить арену',
                    '#enable_forest - Включить лес',
                    '#disable_forest - Выключить лес',
                    '#enable_cave - Включить пещеру',
                    '#disable_cave - Выключить пещеру',
                    '#enable_korovan - Включить корован',
                    '#disable_korovan - Выключить корован',
                    '#enable_def_korovan - Включить защиту корована',
                    '#disable_def_korovan - Выключить защиту корована',
                    '#enable_order - Включить приказы',
                    '#disable_order - Выключить приказы',
                    '#enable_auto_def - Включить авто деф',
                    '#disable_auto_def - Выключить авто деф',
                    '#enable_auto_level_up - Включить авто level up',
                    '#disable_auto_level_up - Выключить авто level up',
                    '#status - Получить статус',
                    '#hero - Получить информацию о герое',
                    '#push_order - Добавить приказ ({0})'.format(','.join(orders)),
                    '#order - Дебаг, последняя команда защиты/атаки замка',
                    '#log - Дебаг, последние 30 сообщений из лога',
                    '#time - Дебаг, текущее время',
                    '#last_arena_visit - Дебаг, последняя битва на арене',
                    '#get_info_diff - Дебаг, последняя разница между запросами информации о герое',
                    '#ping - Дебаг, проверить жив ли бот',
                ]))

            # Вкл/выкл бота
            if text == '#enable_bot':
                bot_enabled = True
                send_msg(admin_username, 'Бот успешно включен')
            if text == '#disable_bot':
                bot_enabled = False
                send_msg(admin_username, 'Бот успешно выключен')

            # Вкл/выкл арены
            if text == '#enable_arena':
                arena_enabled = True
                send_msg(admin_username, 'Арена успешно включена')
            if text == '#disable_arena':
                arena_enabled = False
                send_msg(admin_username, 'Арена успешно выключена')

            # Вкл/выкл леса
            if text == '#enable_forest':
                forest_enabled = True
                send_msg(admin_username, 'Лес успешно включен')
            if text == '#disable_forest':
                forest_enabled = False
                send_msg(admin_username, 'Лес успешно выключен')

            # Вкл/выкл пещеры
            if text == '#enable_cave':
                cave_enabled = True
                send_msg(admin_username, 'Пещера успешно включена')
            if text == '#disable_cave':
                cave_enabled = False
                send_msg(admin_username, 'Пещера успешно выключена')

            # Вкл/выкл корована
            if text == '#enable_korovan':
                korovan_enabled = True
                send_msg(admin_username, 'Корованы успешно включены')
            if text == '#disable_korovan':
                korovan_enabled = False
                send_msg(admin_username, 'Корованы успешно выключены')

            # Вкл/выкл корована
            if text == '#enable_def_korovan':
                korovan_def_enabled = True
                send_msg(admin_username, 'Защита корованов успешно включена')
            if text == '#disable_def_korovan':
                korovan_def_enabled = False
                send_msg(admin_username, 'Защита корованов успешно выключена')

            # Вкл/выкл команд
            if text == '#enable_order':
                order_enabled = True
                send_msg(admin_username, 'Приказы успешно включены')
            if text == '#disable_order':
                order_enabled = False
                send_msg(admin_username, 'Приказы успешно выключены')

            # Вкл/выкл авто деф
            if text == '#enable_auto_def':
                auto_def_enabled = True
                send_msg(admin_username, 'Авто деф успешно включен')
            if text == '#disable_auto_def':
                auto_def_enabled = False
                send_msg(admin_username, 'Авто деф успешно выключен')

            # Вкл/выкл авто деф
            if text == '#enable_auto_level_up':
                auto_level_up = True
                send_msg(admin_username, 'Авто levelup успешно включен')
            if text == '#disable_auto_level_up':
                auto_level_up = False
                send_msg(admin_username, 'Авто levelup успешно выключен')

            # Получить статус
            if text == '#status':
                send_msg(admin_username, '\n'.join([
                    'Бот включен: {0}',
                    'Арена включена: {1}',
                    'Лес включен: {2}',
                    'Пещеры включены: {3}',
                    'Корованы включены: {4}',
                    'Защита корованов включена: {5}',
                    'Приказы включены: {6}',
                    'Авто деф включен: {7}',
                    'Авто levelup включен: {8}'
                ]).format(bot_enabled, arena_enabled, forest_enabled, cave_enabled, korovan_enabled, korovan_def_enabled, order_enabled, auto_def_enabled, auto_level_up))

            # Информация о герое
            if text == '#hero':
                fwd(admin_username, hero_message_id)

            # Получить лог
            if text == '#log':
                send_msg(admin_username, '\n'.join(log_list))
                log_list.clear()

            if text == '#last_arena_visit':
                send_msg(admin_username, str(last_arena_visit))

            if text == '#order':
                text_date = datetime.datetime.fromtimestamp(current_order['time']).strftime('%Y-%m-%d %H:%M:%S')
                send_msg(admin_username, current_order['order'] + ' ' + text_date)

            if text == '#time':
                text_date = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
                send_msg(admin_username, text_date)

            if text == '#ping':
                send_msg(admin_username, '#pong')

            if text == '#get_info_diff':
                send_msg(admin_username, str(get_info_diff))

            if text.startswith('#push_order'):
                command = text.split(' ')[1]
                if command in orders:
                    update_order(orders[command])
                    send_msg(admin_username, 'Команда ' + command + ' применена')
                else:
                    send_msg(admin_username, 'Команда ' + command + ' не распознана')


def send_msg(to, message):
    sender.send_msg('@' + to, message)


def fwd(to, message_id):
    sender.fwd('@' + to, message_id)


def update_order(order):
    current_order['order'] = order
    current_order['time'] = time()
    if order == orders[castle_name]:
        action_list.append(orders['defence'])
    else:
        action_list.append(orders['attack'])
    action_list.append(order)


def log(text):
    message = '{0:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now()) + ' ' + text
    print(message)
    log_list.append(message)


if __name__ == '__main__':
    receiver = Receiver(sock=socket_path) if socket_path else Receiver(port=port)
    receiver.start()  # start the Connector.
    _thread.start_new_thread(queue_worker, ())
    receiver.message(work_with_message(receiver))
    receiver.stop()