main.py

import config
import middleware
import telebot
import models
import time
from datetime import datetime
from datetime import timedelta
from middleware import keyboard
from app import fsm, bot
from app import main_base as base
from tool import language_check, create_inlineKeyboard

middleware.start_draw_timer()
middleware.end_draw_timer()

# -------------------------------------- # START # -------------------------------------- #
@bot.message_handler(commands=['start'])
def start(message):
	print(base.select_all(models.DrawNot))
	base.delete(models.State, user_id=message.chat.id)
	if message.chat.type == 'private':
		text = language_check(message.chat.id)
		if text[0] == True:
			bot.send_message(message.chat.id, language_check(message.chat.id)[1]['menu']['welcome_text'], reply_markup=keyboard.get_menu_keyboard(message.chat.id))
		else:
			bot.send_message(message.chat.id, language_check(message.chat.id)[1]['menu']['welcome_text'], reply_markup=keyboard.get_menu_keyboard(message.chat.id))
			base.new(models.User, str(message.chat.id), str(message.chat.username), "RU")










@bot.callback_query_handler(func=lambda call: True and call.data.split('_')[0] == 'geton')
def get_on_draw(call):
	try:
		text = language_check(call.message.chat.id)[1]['draw']
		tmp = middleware.new_player(call)
		if tmp[1] == 'not_subscribe':
			bot.answer_callback_query(callback_query_id=call.id, show_alert=True,  text=text['not_subscribe'])
		if tmp[0] == False:
			bot.answer_callback_query(callback_query_id=call.id, show_alert=True,  text=text['already_in'])
		else:
			bot.answer_callback_query(callback_query_id=call.id, show_alert=True,  text=text['got_on'])
			bot.edit_message_reply_markup(chat_id=call.message.chat.id, message_id=call.message.message_id, inline_message_id=call.inline_message_id, reply_markup=create_inlineKeyboard({f"({tmp[1]}) {tmp[2]}":call.data}))
	except:
		pass


############################################# laguage checkers ######################################################################################
# -------------------------------------- # change language # -------------------------------------- #
@bot.message_handler(func=lambda message: True and message.text == language_check(message.chat.id)[1]['menu']['menu_buttons'][2])
def change_language(message):
	user = base.get_one(models.User, user_id=str(message.chat.id))
	if user.language == 'RU':
		base.update(models.User, {'language': "ENG"}, user_id=str(message.chat.id))
		bot.send_message(message.chat.id, language_check(message.chat.id)[1]['menu']['welcome_text'], reply_markup=keyboard.get_menu_keyboard(message.chat.id))
	else:
		base.update(models.User, {'language': "RU"}, user_id=str(message.chat.id))
		bot.send_message(message.chat.id, language_check(message.chat.id)[1]['menu']['welcome_text'], reply_markup=keyboard.get_menu_keyboard(message.chat.id))


# -------------------------------------- # invite # -------------------------------------- #
@bot.message_handler(func=lambda message: True and message.text == language_check(message.chat.id)[1]['menu']['menu_buttons'][3])
def invite(message):
	text = language_check(str(message.chat.id))
	bot.send_message(message.chat.id, text[1]['menu']['invite'])


# -------------------------------------- # support # -------------------------------------- #
@bot.message_handler(func=lambda message: True and message.text == language_check(message.chat.id)[1]['menu']['menu_buttons'][4])
def support(message):
	text = language_check(str(message.chat.id))
	bot.send_message(message.chat.id, text[1]['menu']['support'])


# -------------------------------------- # back in main menu # -------------------------------------- #
@bot.message_handler(func=lambda message: True and message.text == language_check(message.chat.id)[1]['draw']['back_in_menu'])
def back_in_menu(message):
	base.delete(models.State, user_id=str(message.chat.id))
	base.delete(models.DrawProgress, user_id=(str(message.chat.id)))
	base.delete(models.SubscribeChannel, user_id=(str(message.chat.id)))
	bot.send_message(message.chat.id, language_check(message.chat.id)[1]['menu']['welcome_text'], reply_markup=keyboard.get_menu_keyboard(message.chat.id))


# -------------------------------------- # back in draw menu # -------------------------------------- #
@bot.message_handler(func=lambda message: True and message.text == language_check(message.chat.id)[1]['draw']['back'] and middleware.check_post(message.chat.id) != None)
def back_in_draw_menu(message):
	base.delete(models.State, user_id=str(message.chat.id))
	middleware.send_draw_info(message.chat.id)


# -------------------------------------- # back in draw menu # -------------------------------------- #
@bot.message_handler(func=lambda message: True and message.text == language_check(message.chat.id)[1]['menu']['menu_buttons'][1])
def my_draws(message):
	middleware.my_draw_info(message.chat.id)
	fsm.set_state(message.chat.id, 'my_draws', number=0)



@bot.callback_query_handler(func=lambda call: True and call.data == 'next')
def next(call):
	try:
		text = language_check(call.message.chat.id)[1]['my_draw']
		number = int(fsm.get_state(call.message.chat.id)[1]['number']) + 1
		tmp = middleware.my_draw_info(call.message.chat.id, row=number)
		if tmp == 'last':
			bot.answer_callback_query(callback_query_id=call.id, show_alert=False,  text=text['last'])
			return 
		bot.delete_message(call.message.chat.id, call.message.message_id)
		fsm.set_state(call.message.chat.id, 'my_draws', number=number)
	except:
		fsm.remove_state(call.message.chat.id)
		bot.delete_message(call.message.chat.id, call.message.message_id)



@bot.callback_query_handler(func=lambda call: True and call.data == 'back')
def back(call):
	try:
		text = language_check(call.message.chat.id)[1]['my_draw']
		number = int(fsm.get_state(call.message.chat.id)[1]['number']) - 1
		tmp = middleware.my_draw_info(call.message.chat.id, row=number)
		if tmp == 'first':
			bot.answer_callback_query(callback_query_id=call.id, show_alert=False,  text=text['first'])
			return 

		bot.delete_message(call.message.chat.id, call.message.message_id)
		fsm.set_state(call.message.chat.id, 'my_draws', number=number)
	except:
		fsm.remove_state(call.message.chat.id)
		bot.delete_message(call.message.chat.id, call.message.message_id)




	
############################################ draw func #################################################
# -------------------------------------- # submit # -------------------------------------- #
@bot.message_handler(func=lambda message: True and middleware.check_post(message.chat.id) != None and message.text == language_check(message.chat.id)[1]['draw']['draw_buttons'][-2])
def submit(message):
	text = language_check(str(message.chat.id))
	bot.send_message(message.chat.id, text[1]['draw']['submit_text'], reply_markup=keyboard.get_menu_keyboard(message.chat.id))
	tmp = base.get_one(models.DrawProgress, user_id=str(message.chat.id))
	base.new(models.DrawNot, tmp.id, tmp.user_id, tmp.chanel_id, tmp.chanel_name, tmp.text, tmp.file_type, tmp.file_id, tmp.winers_count, tmp.post_time, tmp.end_time)
	base.delete(models.DrawProgress, user_id=(str(message.chat.id)))
	base.delete(models.State, user_id=(str(message.chat.id)))


# -------------------------------------- # enter_id # -------------------------------------- #
@bot.message_handler(func=lambda message: True and message.text == language_check(message.chat.id)[1]['menu']['menu_buttons'][0])
def enter_id(message):
	base.delete(models.DrawProgress, user_id=(str(message.chat.id)))
	base.delete(models.SubscribeChannel, user_id=(str(message.chat.id)))
	text = language_check(str(message.chat.id))[1]['draw']
	back_button = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True)
	back_button.row(text['back_in_menu'])
	fsm.set_state(message.chat.id, "writting_channel_id")
	bot.send_message(message.chat.id, text['chanel_id'], reply_markup=back_button)


# -------------------------------------- # enter_text # -------------------------------------- #
@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'writting_channel_id')
def enter_text(message):
	status = ['creator', 'administrator']
	text = language_check(str(message.chat.id))[1]['draw']
	back_button = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True)
	back_button.row(text['back_in_menu'])
	
	try:
		if str(bot.get_chat_member(chat_id=message.text, user_id=message.from_user.id).status) not in status:
			bot.send_message(text['not_admin'], reply_markup=back_button)
			return ''
		tmp = bot.send_message(message.text, 'test')
		bot.delete_message(tmp.chat.id, tmp.message_id)
	except:
		bot.send_message(message.chat.id, text['not_in_chanel'], reply_markup=back_button)
		return ''
	fsm.set_state(message.chat.id, "writting_text", chanel_id=message.text, chanel_name=tmp.chat.title)
	bot.send_message(message.chat.id, text['draw_text'], reply_markup=back_button)


# -------------------------------------- # writting_text # -------------------------------------- #
@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'writting_text')
def enter_photo(message):
	text = language_check(str(message.chat.id))[1]['draw']
	back_button = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True)
	back_button.row(text['back_in_menu'])
	tmp = fsm.get_state(message.chat.id)[1]
	fsm.set_state(message.chat.id, "enter_photo", chanel_id=tmp['chanel_id'], chanel_name=tmp['chanel_name'], draw_text=message.text)
	bot.send_message(message.chat.id, text['file'], reply_markup=back_button)


# -------------------------------------- # enter_photo # -------------------------------------- #
@bot.message_handler(content_types=['text', 'photo', 'document'], func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'enter_photo')
def enter_photo(message):
	file_id = ''
	file_type = ''
	text = language_check(str(message.chat.id))[1]['draw']
	back_button = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True)
	back_button.row(text['back_in_menu'])
	tmp = fsm.get_state(message.chat.id)[1]
	if message.content_type == 'photo':
		file_id = message.photo[0].file_id
		file_type = 'photo'
	elif message.content_type == 'document':
		file_id = message.document.file_id
		file_type = 'document'
	else:
		file_id = ''
		file_type = 'text'
	
	fsm.set_state(message.chat.id, "enter_winers_count", chanel_id=tmp['chanel_id'], chanel_name=tmp['chanel_name'], draw_text=tmp['draw_text'], file_type=file_type, file_id=file_id)
	bot.send_message(message.chat.id, text['winers_count'], reply_markup=back_button)


# -------------------------------------- # enter_winers_count # -------------------------------------- #
@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'enter_winers_count')
def enter_winers_count(message):
	text = language_check(str(message.chat.id))[1]['draw']
	try:
		print(int(message.text))
	except:
		bot.send_message(message.chat.id, text['not_int'])
		return 'gg'
	
	back_button = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True)
	back_button.row(text['back_in_menu'])
	tmp = fsm.get_state(message.chat.id)[1]
	fsm.set_state(message.chat.id, "enter_start_time", chanel_id=tmp['chanel_id'], chanel_name=tmp['chanel_name'], draw_text=tmp['draw_text'], 
				  file_type=tmp['file_type'], file_id=tmp['file_id'], winers_count=message.text)
	
	bot.send_message(message.chat.id, text['post_time'], reply_markup=back_button)



# -------------------------------------- # enter_start_time # -------------------------------------- #
@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'enter_start_time')
def enter_start_time(message):
	text = language_check(str(message.chat.id))[1]['draw']
	try:																	# Проверяет правильно ли ввёл время юзер
		print(time.strptime(message.text, '%Y-%m-%d %H:%M'))
	except:
		bot.send_message(message.chat.id, text['invalid_format_time'])
		return 'gg'

	if time.strptime(datetime.now().strftime('%Y-%m-%d %H:%M'), '%Y-%m-%d %H:%M') >= time.strptime(message.text, '%Y-%m-%d %H:%M'):
		bot.send_message(message.chat.id, text['over_time'])
		return 'gg'
	

	back_button = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True)
	back_button.row(text['back_in_menu'])
	
	tmp = fsm.get_state(message.chat.id)[1]
	fsm.set_state(message.chat.id, "enter_end_time", chanel_id=tmp['chanel_id'], chanel_name=tmp['chanel_name'], draw_text=tmp['draw_text'], 
				  file_type=tmp['file_type'], file_id=tmp['file_id'], winers_count=tmp['winers_count'], start_time=message.text)
	
	bot.send_message(message.chat.id, text['end_time'], reply_markup=back_button)


# -------------------------------------- # enter_end_time # -------------------------------------- #
@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'enter_end_time')
def enter_end_time(message):
	text = language_check(str(message.chat.id))[1]['draw']
	try:																	
		print(time.strptime(message.text, '%Y-%m-%d %H:%M'))
	except:
		bot.send_message(message.chat.id, text['invalid_format_time'])
		return 'gg'
	
	tmp = fsm.get_state(message.chat.id)[1]
	if time.strptime(tmp['start_time'], '%Y-%m-%d %H:%M') >= time.strptime(message.text, '%Y-%m-%d %H:%M'):
		bot.send_message(message.chat.id, text['post_biger'])
		return 'gg'

	if time.strptime(datetime.now().strftime('%Y-%m-%d %H:%M'), '%Y-%m-%d %H:%M') >= time.strptime(message.text, '%Y-%m-%d %H:%M'):
		bot.send_message(message.chat.id, text['over_time'])
		return 'gg'

	back_button = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True)
	back_button.row(text['back_in_menu'])
	fsm.set_state(message.chat.id, "enter_end_time", chanel_id=tmp['chanel_id'], chanel_name=tmp['chanel_name'], draw_text=tmp['draw_text'], file_type=tmp['file_type'], 
				  file_id=tmp['file_id'], winers_count=tmp['winers_count'], start_time=tmp['start_time'], end_time=message.text)
	tmp = fsm.get_state(message.chat.id)[1]
	if tmp['file_type'] == 'photo':
		bot.send_photo(message.chat.id, tmp['file_id'], middleware.create_draw_progress(message.chat.id, tmp), reply_markup=keyboard.get_draw_keyboard(message.chat.id))
	elif tmp['file_type'] == 'document':
		bot.send_document(message.chat.id, tmp['file_id'], caption=middleware.create_draw_progress(message.chat.id, tmp), reply_markup=keyboard.get_draw_keyboard(message.chat.id))
	else:
		bot.send_message(message.chat.id, middleware.create_draw_progress(message.chat.id, tmp), reply_markup=keyboard.get_draw_keyboard(message.chat.id))


# -------------------------------------- # change start time # -------------------------------------- #
@bot.message_handler(func=lambda message: True and middleware.check_post(message.chat.id) != None and message.text == language_check(message.chat.id)[1]['draw']['draw_buttons'][0])
def change_start_time(message):
	text = language_check(str(message.chat.id))[1]['draw']
	fsm.set_state(message.chat.id, 'change_post_time')
	bot.send_message(message.chat.id, text['post_time'], reply_markup=keyboard.back_button(message.chat.id))


@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'change_post_time')
def confirm_change_start_time(message):
	text = language_check(str(message.chat.id))[1]['draw']
	try:																	# Проверяет правильно ли ввёл время юзер
		print(time.strptime(message.text, '%Y-%m-%d %H:%M'))
	except:
		bot.send_message(message.chat.id, text['invalid_format_time'])
		return 'gg'

	if time.strptime(datetime.now().strftime('%Y-%m-%d %H:%M'), '%Y-%m-%d %H:%M') >= time.strptime(message.text, '%Y-%m-%d %H:%M'):
		bot.send_message(message.chat.id, text['over_time'])
		return 'gg'

	tmp = base.get_one(models.DrawProgress, user_id=str(message.chat.id))
	if time.strptime(message.text, '%Y-%m-%d %H:%M') >= time.strptime(tmp.end_time, '%Y-%m-%d %H:%M'):
		bot.send_message(message.chat.id, text['post_biger'])
		return 'gg'

	base.update(models.DrawProgress, {'post_time': message.text}, user_id=str(message.chat.id))
	middleware.send_draw_info(message.chat.id)


# -------------------------------------- # change end time # -------------------------------------- #
@bot.message_handler(func=lambda message: True and middleware.check_post(message.chat.id) != None and message.text == language_check(message.chat.id)[1]['draw']['draw_buttons'][1])
def change_end_time(message):
	text = language_check(str(message.chat.id))[1]['draw']
	fsm.set_state(message.chat.id, 'change_end_time')
	bot.send_message(message.chat.id, text['end_time'], reply_markup=keyboard.back_button(message.chat.id))


@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'change_end_time')
def confirm_change_end_time(message):
	text = language_check(str(message.chat.id))[1]['draw']
	try:																
		print(time.strptime(message.text, '%Y-%m-%d %H:%M'))
	except:
		bot.send_message(message.chat.id, text['invalid_format_time'])
		return 'gg'

	if time.strptime(datetime.now().strftime('%Y-%m-%d %H:%M'), '%Y-%m-%d %H:%M') >= time.strptime(message.text, '%Y-%m-%d %H:%M'):
		bot.send_message(message.chat.id, text['over_time'])
		return 'gg'

	tmp = base.get_one(models.DrawProgress, user_id=str(message.chat.id))
	if time.strptime(message.text, '%Y-%m-%d %H:%M') <= time.strptime(tmp.post_time, '%Y-%m-%d %H:%M'):
		bot.send_message(message.chat.id, text['post_biger'])
		return 'gg'

	base.update(models.DrawProgress, {'end_time': message.text}, user_id=str(message.chat.id))
	middleware.send_draw_info(message.chat.id)


# -------------------------------------- # change winers count # -------------------------------------- #
@bot.message_handler(func=lambda message: True and middleware.check_post(message.chat.id) != None and message.text == language_check(message.chat.id)[1]['draw']['draw_buttons'][2])
def change_winers_count(message):
	text = language_check(str(message.chat.id))[1]['draw']
	fsm.set_state(message.chat.id, 'change_winers_count')
	bot.send_message(message.chat.id, text['winers_count'], reply_markup=keyboard.back_button(message.chat.id))


@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'change_winers_count')
def confirm_change_wines_count(message):
	try:
		print(int(message.text))
	except:
		bot.send_message(message.chat.id, language_check(message.chat.id)[1]['draw']['not_int'])
		return 'gg'
	
	base.update(models.DrawProgress, {'winers_count': message.text}, user_id=str(message.chat.id))
	middleware.send_draw_info(message.chat.id)


# -------------------------------------- # change text # -------------------------------------- #
@bot.message_handler(func=lambda message: True and middleware.check_post(message.chat.id) != None and message.text == language_check(message.chat.id)[1]['draw']['draw_buttons'][3])
def change_text(message):
	text = language_check(str(message.chat.id))[1]['draw']
	fsm.set_state(message.chat.id, 'change_draw_text')
	bot.send_message(message.chat.id, text['draw_text'], reply_markup=keyboard.back_button(message.chat.id))


@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'change_draw_text')
def confirm_change_draw_text(message):
	base.update(models.DrawProgress, {'text': message.text}, user_id=str(message.chat.id))
	middleware.send_draw_info(message.chat.id)


# -------------------------------------- # change photo # -------------------------------------- #
@bot.message_handler(func=lambda message: True and middleware.check_post(message.chat.id) != None and message.text == language_check(message.chat.id)[1]['draw']['draw_buttons'][4])
def change_photo(message):
	text = language_check(str(message.chat.id))[1]['draw']
	fsm.set_state(message.chat.id, 'change_draw_photo')
	bot.send_message(message.chat.id, text['file'], reply_markup=keyboard.back_button(message.chat.id))


@bot.message_handler(content_types=['text', 'photo', 'document'], func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'change_draw_photo')
def confirm_change_draw_photo(message):
	file_id = ''
	file_type = ''
	if message.content_type == 'photo':
		file_id = message.photo[0].file_id
		file_type = 'photo'
	elif message.content_type == 'document':
		file_id = message.document.file_id
		file_type = 'document'
	else:
		file_id = ''
		file_type = 'text'
	base.update(models.DrawProgress, {'file_id': file_id, 'file_type': file_type}, user_id=str(message.chat.id))
	middleware.send_draw_info(message.chat.id)


# -------------------------------------- # add channel check # -------------------------------------- #
@bot.message_handler(func=lambda message: True and middleware.check_post(message.chat.id) != None and message.text == language_check(message.chat.id)[1]['draw']['draw_buttons'][5])
def add_chanel(message):
	text = language_check(str(message.chat.id))[1]['draw']
	fsm.set_state(message.chat.id, 'add_check_channel')
	bot.send_message(message.chat.id, text['chanel_id_check'], reply_markup=keyboard.back_button(message.chat.id))


@bot.message_handler(func=lambda message: True and fsm.get_state(message.chat.id)[0] == 'add_check_channel')
def add_check_channel(message):
	text = language_check(str(message.chat.id))[1]['draw']
	try:
		status = ['creator', 'administrator']
		if str(bot.get_chat_member(chat_id=message.text, user_id=message.from_user.id).status) not in status:
			bot.send_message(text['not_admin'])
			return ''
	except:
		bot.send_message(message.chat.id, text['not_in_chanel'])
		return ''
	tmp = base.get_one(models.DrawProgress, user_id=str(message.chat.id))
	base.new(models.SubscribeChannel, tmp.id, str(message.chat.id), message.text)
	middleware.send_draw_info(message.chat.id)
	print(base.select_all(models.SubscribeChannel))











if __name__ == '__main__':
	bot.polling(none_stop=True)