-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathmain.py
204 lines (166 loc) · 9.5 KB
/
main.py
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
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
import logging
import pickle
import pprint
from telegram.ext import (CommandHandler, ConversationHandler, InlineQueryHandler, MessageHandler, Filters,
PicklePersistence, Updater, CallbackQueryHandler, PollAnswerHandler)
# from telegram import ParseMode
# from telegram.utils.helpers import mention_html
import inline
from bot_funcs import media_reactor, morning_wisher, bday_wisher, conversation, delete_pin, welcome
from bot_funcs.commands import BotCommands as bc
from constants import shanibot, group_ids
from convos import bday, magic, nick, settings_gui, start
from bot_funcs.quiz import send_quiz, receive_answer, timedout
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(lineno)d - %(message)s', level=logging.INFO)
with open("files/token.txt", 'r') as file:
shanisir_token, test_token = file.read().split(',')
pp = PicklePersistence(filename='files/user_data')
updater = Updater(token=f'{shanisir_token}', use_context=True, persistence=pp)
dp = updater.dispatcher
def data_view() -> None:
with open('files/user_data', 'rb') as f:
pprint.PrettyPrinter(indent=2).pprint(pickle.load(f))
# def update_data(context):
# Samir update-
# context.bot_data['quizizz'][764886971]['questions_answered'] += 5
# context.bot_data['quizizz'][764886971]['answers_wrong'] = 1
# context.bot_data['quizizz'][764886971]['answers_right'] += 1
# # me-
# context.bot_data['quizizz'][476269395]['answers_right'] += 1
#
# # Jaden update-
# context.bot_data['quizizz'][847874359]['questions_answered'] += 5
# context.bot_data['quizizz'][847874359]['answers_wrong'] = 2
# context.bot_data['quizizz'][847874359]['answers_right'] += 3
#
# Samrin update-
# context.bot_data['quizizz'][1009248402]['questions_answered'] += 5
# context.bot_data['quizizz'][1009248402]['answers_wrong'] = 1
# context.bot_data['quizizz'][1009248402]['answers_right'] += 1
#
# Abdus update-
# context.bot_data['quizizz'][925784909]['questions_answered'] += 5
# context.bot_data['quizizz'][925784909]['answers_wrong'] = 2
# context.bot_data['quizizz'][925784909]['answers_right'] += 1
#
# # Rakin update-
# context.bot_data['quizizz'][831658863] = {}
# context.bot_data['quizizz'][831658863]['questions_answered'] = 5
# context.bot_data['quizizz'][831658863]['answers_wrong'] = 0
# context.bot_data['quizizz'][831658863]['answers_right'] = 5
# context.bot_data['quizizz'][831658863]['name'] = 'Rakin'
# context.bot_data['quizizz'][831658863]['profile_pic'] = 'profile_pics/Rakin.jpg'
#
# # Ronit update-
# context.bot_data['quizizz'][869309961] = {}
# context.bot_data['quizizz'][869309961]['questions_answered'] = 5
# context.bot_data['quizizz'][869309961]['answers_wrong'] = 1
# context.bot_data['quizizz'][869309961]['answers_right'] = 4
# context.bot_data['quizizz'][869309961]['name'] = 'Ronit'
# context.bot_data['quizizz'][869309961]['profile_pic'] = 'profile_pics/Ronit.jpg'
#
# # Jai update-
# context.bot_data['quizizz'][822149388]['questions_answered'] += 5
# context.bot_data['quizizz'][822149388]['answers_wrong'] = 2
# context.bot_data['quizizz'][822149388]['answers_right'] += 3
#
# # Adeep update-
# context.bot_data['quizizz'][1020219808]['questions_answered'] += 5
# context.bot_data['quizizz'][1020219808]['answers_wrong'] = 0
# context.bot_data['quizizz'][1020219808]['answers_right'] += 5
#
# print("UPDATED ALL")
# # with open('files/user_data', 'rb+') as f1:
# # dic = pickle.load(f1)
# # dic['user_data'][894016631]['nickname'].remove('Nigger')
# # dic['user_data'][894016631]['nickname'].remove('Nigga')
# # print('removed')
# # print(dic['user_data'][894016631]['nickname'])
# #
# # with open('files/user_data', 'wb+') as f2:
# # pickle.dump(dic, f2)
# # print('updated')
# context.dispatcher.persistence.flush()
# print('saved')
#
# def user(context):
#
# context.bot.send_photo(chat_id=group_ids['grade12'], photo=open('leaderboard.png', 'rb'),
# caption="Current standings.")
# mention = ''
# for _id, name in [(822149388, 'Jai ')]:
# mention += mention_html(user_id=_id, name=name) # Get their mention in html
#
# context.bot.send_message(chat_id=group_ids['grade12'],
# text=mention + "Are you fine? Physics is easy what's the problem like",
# parse_mode=ParseMode.HTML)
dp.add_handler(InlineQueryHandler(inline.inline_clips))
dp.add_handler(CommandHandler(command='help', callback=bc.helper))
dp.add_handler(CommandHandler(command='secret', callback=bc.secret))
dp.add_handler(CommandHandler(command='start', callback=bc.start))
dp.add_handler(CommandHandler(command='swear', callback=bc.swear))
dp.add_handler(CommandHandler(command='snake', callback=bc.snake))
dp.add_handler(CommandHandler(command='facts', callback=bc.facts))
dp.add_handler(CommandHandler(command='quizizz', callback=bc.quizizz))
dp.add_handler(PollAnswerHandler(callback=receive_answer))
# /8ball conversation-
magicball_handler = ConversationHandler(
entry_points=[CommandHandler(command="8ball", callback=magic.magic8ball, filters=~Filters.reply),
MessageHandler(filters=Filters.command(False) & Filters.regex("8ball") & Filters.reply,
callback=magic.thinking)],
states={
magic.PROCESSING: [MessageHandler(filters=Filters.reply & Filters.text, callback=magic.thinking)],
ConversationHandler.TIMEOUT: [MessageHandler(filters=Filters.all, callback=magic.timedout)]},
fallbacks=[CommandHandler(command='cancel', callback=magic.cancel)], conversation_timeout=40)
dp.add_handler(magicball_handler)
# /tell conversation-
tell_handler = ConversationHandler(
entry_points=[CommandHandler('tell', start.initiate)],
states={
start.CHOICE: [MessageHandler(filters=Filters.regex("^Birthday$"), callback=bday.bday),
MessageHandler(filters=Filters.regex("^Nickname$"), callback=nick.nick),
MessageHandler(filters=Filters.regex("^Nothing$"), callback=start.leave)],
bday.INPUT: [MessageHandler(filters=Filters.regex("^([1-9][0-9]{3}-[0-9]{2}-[0-9]{2})$"), # Valid date check
callback=bday.bday_add_or_update),
MessageHandler(filters=Filters.text, callback=bday.wrong)], # If it is not a date
bday.MODIFY: [MessageHandler(filters=Filters.regex("^Forget my birthday sir$"), callback=bday.bday_del),
MessageHandler(filters=Filters.regex("^Update my birthday sir$"), callback=bday.bday_mod)],
nick.SET_NICK: [MessageHandler(filters=Filters.text & Filters.reply, callback=nick.add_edit_nick)],
nick.MODIFY_NICK: [MessageHandler(filters=Filters.regex("^Change nickname$"), callback=nick.edit_nick),
MessageHandler(filters=Filters.regex("^Remove nickname$"), callback=nick.del_nick),
MessageHandler(filters=Filters.regex("^Back$"), callback=nick.back)],
ConversationHandler.TIMEOUT: [MessageHandler(filters=Filters.all, callback=start.timedout)]},
fallbacks=[MessageHandler(Filters.regex("^No, thank you sir$"), callback=bday.reject),
CommandHandler("cancel", start.leave)],
name="/tell convo", persistent=True, allow_reentry=True, conversation_timeout=40)
dp.add_handler(tell_handler)
settings_gui_handler = ConversationHandler(
entry_points=[CommandHandler('settings', settings_gui.start)],
states={
settings_gui.UPDATED: [CallbackQueryHandler(settings_gui.change_prob, pattern="MEDIA_PROB|PROFANE_PROB"),
CallbackQueryHandler(settings_gui.morn_swap, pattern="Morning"),
CallbackQueryHandler(settings_gui.save, pattern="SAVE")],
settings_gui.PROBABILITY: [
CallbackQueryHandler(settings_gui.prob_updater, pattern="0.0|-0.1|-0.05|0.05|0.1|1.0"),
CallbackQueryHandler(settings_gui.go_back, pattern="Back")]},
fallbacks=[CommandHandler('cancel', settings_gui.save)])
dp.add_handler(settings_gui_handler)
media_filters = (Filters.document | Filters.photo | Filters.video | Filters.voice | Filters.audio)
edit_filter = Filters.update.edited_message
pin_filter = Filters.status_update.pinned_message
new_mem = Filters.status_update.new_chat_members
dp.add_handler(MessageHandler(new_mem & Filters.chat(chat_id=int(group_ids['grade12'])), callback=welcome.welcome))
dp.add_handler(MessageHandler(media_filters, media_reactor.media))
dp.add_handler(MessageHandler(pin_filter & Filters.user(username=shanibot), delete_pin.de_pin))
dp.add_handler(MessageHandler(Filters.reply & Filters.group & ~ edit_filter, conversation.reply))
dp.add_handler(MessageHandler(Filters.group & Filters.text & ~ edit_filter, conversation.group))
dp.add_handler(MessageHandler(Filters.private & Filters.text & ~ edit_filter, conversation.shanifier))
dp.add_handler(MessageHandler(Filters.command, bc.unknown))
updater.job_queue.run_repeating(bday_wisher.wish, 86400, first=1) # Runs every time script is started, and once a day.
updater.job_queue.run_repeating(morning_wisher.morning_goodness, 3600, first=1)
updater.job_queue.run_repeating(inline.get_clips, 60, first=1) # Have to re-fetch clips since links expire
updater.job_queue.run_repeating(send_quiz, 604800, first=1) # Send quiz to TMS'20 weekly
updater.job_queue.run_repeating(timedout, 86400, first=20)
data_view()
updater.start_polling()
updater.idle()