-
Notifications
You must be signed in to change notification settings - Fork 0
/
telegbot_events.py
197 lines (170 loc) · 8.14 KB
/
telegbot_events.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
#!/usr/bin/python3
import telebot
from telebot import types
import datetime
from dateutil.parser import parse
#from telegramcalendar import create_calendar
import pandas as pd
import newWebCalendar
import time
import threading
#here I store token from Telegram bot in separate file
import mainInfo
bot = telebot.TeleBot(mainInfo.botToken)
TO_CHAT_ID = mainInfo.whomReplay
mess = {}
mess_time = datetime.date.today()
current_shown_dates = {}
eventsFile = pd.DataFrame([])
eventsFile.to_csv('fitosEvents.csv', header=None, index_label=None, mode='a')
dateBegin = 0
timeEvent = 0
dateEnd = 0
updateTimer = 0
lastImportantMessage = 'Здесь выводятся важные сообщения'
def is_date(string, fuzzy=False):
"""
Return whether the string can be interpreted as a date.
:param string: str, string to check for date
:param fuzzy: bool, ignore unknown tokens in string if True
"""
try:
parse(string, fuzzy=fuzzy)
return True
except ValueError:
return False
def csvEmpty(str):
"""
Return whether csv-file has any string or not
:param string: str, name of the csv-file, e.g. 'myEvents.csv'
:param fuzzy: bool, ignore unknown tokens in string if True
"""
try:
df = pd.read_csv(str,header=None, sep=',')
return False
except ValueError:
return True
@bot.message_handler(commands=['start'])
def start(message):
start = telebot.types.ReplyKeyboardMarkup(True, False)
#messageStart = bot.send_message(message.chat.id, 'Пришлите дату мероприятия в формате DD.MM.YYYY')
start.row('Создать мероприятие')
start.row('Показать список мероприятий')
start.row('Срочное сообщение')
if message.chat.id==TO_CHAT_ID:
start.row('Удалить мероприятие')
messageStart = bot.send_message(message.from_user.id, 'Выберите действие', reply_markup=start)
bot.register_next_step_handler(messageStart, getCommand)
"""
Draft for future releases:
# processing continuos events
def second_date(message):
# administration processing
def admin(message):
"""
# important messages (below the calendar)
# add ability to set what time it can be alive (up to a day)
def breakingMessage(message):
newWebCalendar.createHTMLFile('fitosEvents.csv', message.text)
global lastImportantMessage
lastImportantMessage = str(message.text)
messageErr = bot.send_message(message.chat.id, 'Сообщение опубликовано')
bot.register_next_step_handler(messageErr, getCommand)
# autoupdate for html-file need to run in parallel
def updateCalendar():
global updateTimer
if updateTimer == 5:
updateTimer = 0
newWebCalendar.createHTMLFile('fitosEvents.csv', lastImportantMessage)
# print(datetime.datetime.today())
# print('Calendar updated')
else:
time.sleep(4)
updateTimer = updateTimer + 1
thread1 = threading.Thread(target=updateCalendar)
thread1.start()
def getCommand(message):
if message.text == 'Создать мероприятие':
messageStart = bot.send_message(message.chat.id, 'Пришлите дату мероприятия в формате DD.MM.YYYY')
bot.register_next_step_handler(messageStart, eventDate)
elif message.text == 'Показать список мероприятий':
if (csvEmpty('fitosEvents.csv')):
messageStart = bot.send_message(message.chat.id, 'Все мероприятия прошли, а новых еще не создали')
bot.register_next_step_handler(messageStart, getCommand)
else:
df = pd.read_csv('fitosEvents.csv',header=None, sep=',')
listDates = ''
for i in range(len(df)):
listDates =listDates+str(i) + ') ' +df[0][i] + ' '+df[2][i]+'\n'
message1 = bot.send_message(message.chat.id, listDates)
bot.register_next_step_handler(message1, getCommand)
elif (message.text == 'Удалить мероприятие'):
if message.chat.id == mainInfo.whomReplay:
if (csvEmpty('fitosEvents.csv')):
messageStart = bot.send_message(message.chat.id, 'Все мероприятия прошли, а новых еще не создали')
bot.register_next_step_handler(messageStart, getCommand)
else:
messageErr = bot.send_message(message.chat.id, 'Введите номер записи для удаления:')
bot.register_next_step_handler(messageErr, deleteEvent)
else:
messageErr = bot.send_message(message.chat.id, 'Данная команда Вам недоступна, выберите другую')
bot.register_next_step_handler(messageErr, getCommand)
elif(message.text == 'Срочное сообщение'):
message1 = bot.send_message(message.chat.id, 'Введите сообщение')
bot.register_next_step_handler(message1, breakingMessage)
else:
bot.send_message(message.chat.id, 'Неверная команда! Выберите из меню')
def deleteEvent(message):
# csv-file editing (admin privelege)
if message.text.isdigit()==False:
messageErr = bot.send_message(message.chat.id, 'Введены некорректные данные. Начните сначала')
bot.register_next_step_handler(messageErr, getCommand)
else:
df = pd.read_csv('fitosEvents.csv', header=None)
# Backup deleted events
f1 = open('fitosDeletedEvents.csv', 'a')
f1.write(df[0][int(message.text)]+' '+df[1][int(message.text)]+' '+df[2][int(message.text)]+' '+df[3][int(message.text)]+'\n')
f1.close()
df = df.drop([int(message.text)])
df.to_csv('fitosEvents.csv', sep=',',header=None, index=None)
messageErr = bot.send_message(message.chat.id, 'Мероприятие удалено')
bot.register_next_step_handler(messageErr, getCommand)
newWebCalendar.createHTMLFile('fitosEvents.csv', lastImportantMessage)
def eventDate(message):
global dateEnd
dateEnd = message.text
if (is_date(dateEnd)):
checkDelta = datetime.datetime.strptime(dateEnd, '%d.%m.%Y') - datetime.datetime.today()
if checkDelta.days < 0:
messageEventName = bot.send_message(message.chat.id, 'Мероприятие уже прошло! введите другую дату')
bot.register_next_step_handler(messageEventName, eventDate)
else:
messageEventName = bot.send_message(message.chat.id, 'В какое время начинается мероприятие?')
bot.register_next_step_handler(messageEventName, eventTime)
else:
messageFinish = bot.send_message(message.chat.id, 'Дата неверна! попробуйте еще раз')
bot.register_next_step_handler(messageFinish, eventDate)
print(dateEnd)
def eventTime(message):
global timeEvent
timeEvent = message.text
if (is_date(timeEvent)):
messageEventName = bot.send_message(message.chat.id, 'Как называется мероприятие?')
bot.register_next_step_handler(messageEventName, registerEvent)
else:
messageFinish = bot.send_message(message.chat.id, 'Время в неверном формате! Попробуйте еще раз')
bot.register_next_step_handler(messageFinish, eventTime)
print(timeEvent)
def registerEvent(message):
eventsFile = pd.DataFrame([[dateEnd,dateEnd,message.text,timeEvent]])
eventsFile.to_csv('./fitosEvents.csv', header=None, index=None, mode='a')
print(eventsFile)
messageEventCreated = bot.send_message(message.chat.id, 'Спасибо! внесено в список.')
bot.register_next_step_handler(messageEventCreated, getCommand)
bot.forward_message(TO_CHAT_ID, message.chat.id, message.message_id)
newWebCalendar.createHTMLFile('fitosEvents.csv',lastImportantMessage)
# updateCalendar()
thread1 = threading.Thread(target=updateCalendar)
thread1.start()
if __name__ == '__main__':
bot.infinity_polling()