import pypyodbc
mySQLServer = "DESKTOP-3F2B3B3\SQLEXPRESS"
myDataBase = "DfLessonDb"
itstrue = ('да', 'Да', 'ДА', 'уверен', 'Уверен', 'Верны', 'верны', 'ага', 'Ага', 'Верно', 'верно')
import telebot
from telebot import types
bot = telebot.TeleBot('909191584:AAFMYcMImK84IBchpY1NiGQ3bLeUlpYRUYg')
name = ''
surname = ''
telegramid = 0
id = 0


@bot.message_handler(commands=['start'])
def send_welcome(message):
    user_id = str(message.from_user.id)
    bot.send_message(message.from_user.id, '''Привет, я бот-журнал для списка команд введи /help\nВаш telegram id:''' + user_id)


@bot.message_handler(commands=['help'])
def send_welcome(message):
    bot.send_message(message.from_user.id, "/reg - Начать регистрацию пользователя\n/infoaboutme - получить "
                                           "информацию о себе\n/leavethecours - покинуть курс")


@bot.message_handler(commands=['reg'])
def start_register(message):
    connection = pypyodbc.connect('Driver={SQL Server};'
                                  'Server=' + mySQLServer + ';'
                                                            'Database=' + myDataBase + ';')
    cursor = connection.cursor()
    telegramid = int(message.from_user.id)
    mySQLQuery = "/select * from Students where telegramid={student[0]!r}".format(student=[telegramid])
    cursor.execute(mySQLQuery[1:])
    results = cursor.fetchall()
    if not results:
        bot.send_message(message.from_user.id, 'Введите свое имя')
        bot.register_next_step_handler(message, get_name)
    else:
        bot.send_message(message.from_user.id, 'Вы уже зарегистрированны!')

@bot.message_handler(commands=['infoaboutme'])
def start_register(message):
    connection = pypyodbc.connect('Driver={SQL Server};'
                                  'Server=' + mySQLServer + ';'
                                                            'Database=' + myDataBase + ';')
    print(message)
    cursor = connection.cursor()
    telegramid = int(message.from_user.id)
    mySQLQuery = "/select * from Students where telegramid={student[0]!r}".format(student=[telegramid])
    cursor.execute(mySQLQuery[1:])
    results = str(cursor.fetchall())[2:-2].replace("'", '').replace(',', '', 1).replace(',', '\nНомер студенческого:',
                                                                                        1).replace(',', '\nТелеграм id:')
    if not results:
        bot.send_message(message.from_user.id, 'Вы не зарегистрированны в системе!')
    else:
        bot.send_message(message.from_user.id, results)
    connection.close()

@bot.message_handler(commands=['leavethecours'])
def leave_cours(message):
    bot.send_message(message.from_user.id, 'Вы уверенны что хотите покинуть курс и удалить все данные о себе?')
    bot.register_next_step_handler(message, get_answer_leave)
def get_answer_leave(message):
    print(str(message.text))
    if str(message.text) in itstrue:
        bot.send_message(message.from_user.id, 'До скорых встреч!')
        connection = pypyodbc.connect('Driver={SQL Server};'
                                      'Server=' + mySQLServer + ';'
                                                                'Database=' + myDataBase + ';')
        telegramid = int(message.from_user.id)
        cursor = connection.cursor()
        mySQLQuery = "/delete from Students where telegramid={student[0]!r}".format(student=[telegramid])
        cursor.execute(mySQLQuery[1:])
        connection.commit()
        connection.close()
    else:
        bot.send_message(message.from_user.id, 'Не играй с огнем)')

def get_name(message):
    global name
    name = message.text
    bot.send_message(message.from_user.id, 'Введите свою фамилию')
    bot.register_next_step_handler(message, get_surname)


def get_surname(message):
    global surname
    surname = message.text
    bot.send_message(message.from_user.id, 'Введите номер своего студенческого билета')
    bot.register_next_step_handler(message, get_id)


def get_id(message):
    global id
    id = message.text
    bot.send_message(message.from_user.id,
                     'Данные верны?\nИмя: ' + name + '\n' + 'Фамилия: ' + surname + '\n' + 'Номер суденческого: ' + id)
    bot.register_next_step_handler(message, get_answer_reg)


def get_answer_reg(message):
    if str(message.text) in itstrue:
        bot.send_message(message.from_user.id, 'Успех!')
        connection = pypyodbc.connect('Driver={SQL Server};'
                                      'Server=' + mySQLServer + ';'
                                                                'Database=' + myDataBase + ';')
        telegramid = int(message.from_user.id)
        cursor = connection.cursor()
        mySQLQuery = "/insert into Students (namestud, surnamestud, studid, telegramid) values ({student[0]!r}, {student[1]!r}, " \
                     "{student[2]!r}, {student[3]!r})".format(student=[name, surname, id, telegramid])
        cursor.execute(mySQLQuery[1:])
        connection.commit()
        connection.close()
    else:
        bot.send_message(message.from_user.id, 'Попробуйте снова')
bot.polling()