def reg_message(message):
conn = psycopg2.connect(dbname='', user='', password='', host='')
cursor = conn.cursor()
cursor.execute(
'CREATE TABLE IF NOT EXISTS groups (group_name VARCHAR(255) NOT NULL, group_id VARCHAR(255))'
)
cursor.execute('SELECT group_id FROM groups')
res = cursor.fetchall()
for elem in res:
if str(message.chat.id) in elem:
bot.send_message(message.chat.id,
"Похоже данная группа уже зарегистрированна")
cursor.close()
conn.close()
return
cursor.execute("INSERT INTO groups VALUES (%s, %s)",
(str(message.chat.title), str(message.chat.id)))
string = "CREATE TABLE IF NOT EXISTS " + str(
func.idtoname(message.chat.id)
) + " (person VARCHAR(255), username VARCHAR(255), id VARCHAR(255))"
cursor.execute(string)
bot.send_message(
message.chat.id,
"Регистрация группы " + str(message.chat.title) + " успешна")
conn.commit()
cursor.close()
conn.close()
def set_task_message(message):
if func.check_set_message(message.text) != 0:
data = func.check_set_message(message.text)
string = data[1].split(':')
if len(string) == 5:
if 0 <= int(string[0]) <= 23:
tz = pytz.timezone('Europe/Moscow')
deadline = func.fetch_deadline(data[1])
if deadline > datetime.now(tz):
if func.check_group(message):
if func.check_person(message, data[0]):
conn = psycopg2.connect(dbname='',
user='',
password='',
host='')
cursor = conn.cursor()
string = 'INSERT INTO ' + (
data[0])[1:] + ' VALUES (%s, %s, %s, %s)'
cursor.execute(
string, (str(data[2]), str(message.chat.title),
str(func.idtoname(
message.chat.id)), str(data[1])))
conn.commit()
cursor.close()
conn.close()
bot.send_message(
message.chat.id, 'Задание для ' + "@" +
(data[0])[1:] + ' успешно созданно')
answer = '@' + str(
message.from_user.username
) + ' установил вам задачу: "' + str(
data[2]) + '" до ' + str(data[1])
bot.send_message(func.get_id(message, data[0]),
answer)
else:
bot.send_message(
message.chat.id, "Похоже " + data[0] +
' еще не присоединилась к данной группе')
else:
bot.send_message(
message.chat.id,
"Похоже данная группа еще не зарегистрированна, для регистрации введите /reg"
)
else:
bot.send_message(
message.chat.id,
"К сожалению, у меня нет машины времени =(")
else:
bot.send_message(message.chat.id, "Ошибка, неверный формат")
else:
bot.send_message(message.chat.id, "Ошибка, неверный формат")
else:
bot.send_message(
message.chat.id,
"Похоже вы ввели команду в неправильном формате, для подробной информации введите /help"
)
def join_message(message):
conn = psycopg2.connect(dbname='', user='', password='', host='')
cursor = conn.cursor()
cursor.execute(
'CREATE TABLE IF NOT EXISTS groups (group_name VARCHAR(255) NOT NULL, group_id VARCHAR(255))'
)
cursor.execute('SELECT group_id FROM groups')
res = cursor.fetchall()
for elem in res:
if str(message.chat.id) in elem:
cursor.execute('SELECT id FROM ' +
str(func.idtoname(message.chat.id)))
tmp = cursor.fetchall()
for sup in tmp:
if str(message.from_user.id) in sup:
bot.send_message(
message.chat.id,
"Похоже вы уже присоединились к данной группе")
return
string = "INSERT INTO " + str(func.idtoname(
message.chat.id)) + " VALUES (%s, %s, %s)"
cursor.execute(
string,
(str(message.from_user.first_name),
str(message.from_user.username), str(message.from_user.id)))
string = 'CREATE TABLE IF NOT EXISTS ' + str(message.from_user.username) +\
' (task VARCHAR(255) NOT NULL, group_name VARCHAR(255)' + \
', group_id VARCHAR(255), deadline VARCHAR(255))'
cursor.execute(string)
conn.commit()
cursor.close()
conn.close()
bot.send_message(message.chat.id,
"Вы успешно присоединились к группе")
return
bot.send_message(
message.chat.id,
"Похоже данная группа еще не зарегистрированна, для регистрации введите /reg"
)
conn.commit()
cursor.close()
conn.close()
def drop_message(message):
conn = psycopg2.connect(dbname='', user='', password='', host='')
cursor = conn.cursor()
cursor.execute('SELECT group_id FROM groups')
res = cursor.fetchall()
for elem in res:
string = 'DROP TABLE ' + str(func.idtoname(elem[0]))
cursor.execute(string)
cursor.execute('DROP TABLE groups')
conn.commit()
cursor.close()
conn.close()
def remember(message):
bot.send_message(
message.from_user.id,
"Когда у вас дедлайн? (формат: часы:минуты:день:месяц:год)")
conn = psycopg2.connect(dbname='', user='', password='', host='')
cursor = conn.cursor()
string = 'CREATE TABLE IF NOT EXISTS ' + str(message.from_user.username) + \
' (task VARCHAR(255) NOT NULL, group_name VARCHAR(255)' + \
', group_id VARCHAR(255), deadline VARCHAR(255))'
cursor.execute(string)
string = 'INSERT INTO ' + str(
message.from_user.username) + ' VALUES (%s, %s, %s, %s)'
cursor.execute(string, (message.text, str(
message.chat.title), str(func.idtoname(message.chat.id)), 'inserting'))
conn.commit()
cursor.close()
conn.close()
bot.register_next_step_handler(message, date)