devops_stream2/echo_bot.py at master · ciuses/devops_stream2 · 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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
import os
import logging
from dotenv import load_dotenv
from telegram.ext import (Updater,
                          CommandHandler,
                          MessageHandler,
                          Filters,
                          ConversationHandler,
                          CallbackQueryHandler)

from db_model_data import create_tables
from linux_funcs import (linux_release,
                         linux_df,
                         linux_free,
                         linux_uptime,
                         linux_uname,
                         linux_auths,
                         linux_w,
                         linux_mpstat,
                         linux_apt_list,
                         linux_apt_list_one,
                         linux_apt_list_many,
                         linux_apt_list_one_get,
                         linux_packages_services,
                         all_install_packages,
                         all_up_services,
                         single_package_get,
                         single_service_get,
                         single_package_post,
                         single_service_post,
                         linux_critical,
                         linux_ps,
                         linux_ss,
                         linux_replica_log2)

from main_funcs import (echo,
                        start,
                        my_exit,
                        my_help,
                        find_tel_numbers,
                        find_tel_numbers_command,
                        find_emails,
                        find_emails_command,
                        check_pas,
                        check_pas_command,
                        get_from_the_database_emails,
                        get_from_the_database_telephons,
                        write_tel_numbers,
                        write_emails)

load_dotenv()
TG_TOKEN = os.getenv('ciuse_bot')
logging.basicConfig(filename='my_log.log',
                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                    level=logging.INFO)
logger = logging.getLogger(__name__)


def run() -> None:
    updater = Updater(TG_TOKEN, use_context=True)
    my_disp = updater.dispatcher

    '''Перехват сообщений'''
    echo_handler = MessageHandler(Filters.text & ~Filters.command, echo)
    '''Перехват команд'''
    start_handler = CommandHandler('start', start)
    help_handler = CommandHandler('help', my_help)
    exit_handler = CommandHandler('exit', my_exit)
    linux_release_handler = CommandHandler('get_release', linux_release)
    linux_uname_handler = CommandHandler('get_uname', linux_uname)
    linux_uptime_handler = CommandHandler('get_uptime', linux_uptime)
    linux_df_handler = CommandHandler('get_df', linux_df)
    linux_free_handler = CommandHandler('get_free', linux_free)
    linux_mpstat_handler = CommandHandler('get_mpstat', linux_mpstat)
    linux_w_handler = CommandHandler('get_w', linux_w)
    linux_auths_handler = CommandHandler('get_auths', linux_auths)
    linux_critical_handler = CommandHandler('get_critical', linux_critical)
    linux_ps_handler = CommandHandler('get_ps', linux_ps)
    linux_ss_handler = CommandHandler('get_ss', linux_ss)
    '''Задание про базы '''
    linux_replica_log_handler = CommandHandler('get_repl_logs', linux_replica_log2)
    get_emails_handler = CommandHandler('get_emails', get_from_the_database_emails)
    get_telephons_handler = CommandHandler('get_phone_numbers', get_from_the_database_telephons)

    '''Перехват диалога тела'''
    find_tel_numbers_handler = ConversationHandler(
        entry_points=[CommandHandler('find_tel_numbers', find_tel_numbers_command)],
        states={'find_tel_numbers': [MessageHandler(Filters.text & ~Filters.command, find_tel_numbers)],
                'telephone_step': [MessageHandler(Filters.regex('Да|да'), write_tel_numbers)], },
        fallbacks=[CommandHandler('exit', my_exit)])

    '''Перехват диалога мыла'''
    find_emails_handler = ConversationHandler(
        entry_points=[CommandHandler('find_emails', find_emails_command)],
        states={'find_emails': [MessageHandler(Filters.text & ~Filters.command, find_emails)],
                'email_step': [MessageHandler(Filters.regex('Да|да'), write_emails)], },
        fallbacks=[CommandHandler('exit', my_exit)])

    '''Перехват диалога пароля'''
    check_pas_handler = ConversationHandler(
        entry_points=[CommandHandler('verify_password', check_pas_command)],
        states={'check_pas': [MessageHandler(Filters.text & ~Filters.command, check_pas)], },
        fallbacks=[CommandHandler('exit', my_exit)])

    '''Перехват диалога apt list'''
    apt_list_handler = ConversationHandler(  # TODO выпилить этот блок, вместе с функциями, оно эксперементально
        entry_points=[CommandHandler('get_apt_list', linux_apt_list)],
        states={'linux_apt_list_one': [MessageHandler(Filters.regex('^(Один)$'), linux_apt_list_one),
                                       MessageHandler(Filters.regex('^(Много)$'), linux_apt_list_many),
                                       MessageHandler(Filters.text, linux_apt_list_one_get)], },
        fallbacks=[CommandHandler('exit', my_exit)])

    '''Перехват деалога запроса пакетов и сервисов'''
    decision_tree = ConversationHandler(
        entry_points=[CommandHandler('packages_services', linux_packages_services)],
        states={'first_level': [CallbackQueryHandler(all_install_packages, pattern='^all_packages$'),
                                CallbackQueryHandler(all_up_services, pattern='^all_services$'),
                                CallbackQueryHandler(single_package_get, pattern='^single_package$'),
                                CallbackQueryHandler(single_service_get, pattern='^single_service$'), ],

                'second_level': [MessageHandler(Filters.text & ~Filters.command, single_package_post), ],
                'third_level': [MessageHandler(Filters.text & ~Filters.command, single_service_post), ]
                },
        fallbacks=[CommandHandler('packages_services', linux_packages_services),
                   CommandHandler('exit', my_exit)])

    '''Диспетчеры'''
    my_disp.add_handler(find_tel_numbers_handler)
    my_disp.add_handler(find_emails_handler)
    my_disp.add_handler(check_pas_handler)

    my_disp.add_handler(start_handler)
    my_disp.add_handler(help_handler)
    my_disp.add_handler(exit_handler)

    '''Linux'''
    my_disp.add_handler(linux_release_handler)
    my_disp.add_handler(linux_uname_handler)
    my_disp.add_handler(linux_uptime_handler)
    my_disp.add_handler(linux_df_handler)
    my_disp.add_handler(linux_free_handler)
    my_disp.add_handler(linux_auths_handler)
    my_disp.add_handler(linux_w_handler)
    my_disp.add_handler(linux_mpstat_handler)
    my_disp.add_handler(linux_critical_handler)
    my_disp.add_handler(linux_ps_handler)
    my_disp.add_handler(linux_ss_handler)
    '''Про базки'''
    my_disp.add_handler(linux_replica_log_handler)
    my_disp.add_handler(get_emails_handler)
    my_disp.add_handler(get_telephons_handler)

    my_disp.add_handler(apt_list_handler)
    my_disp.add_handler(decision_tree)

    my_disp.add_handler(echo_handler)

    updater.start_polling()
    updater.idle()


if __name__ == '__main__':
    create_tables()
    run()
    # TODO раскидать логирование