import requests
import telebot
from bs4 import BeautifulSoup as bs
from saturn_cli import modulo_scarica
import csv

bot = telebot.TeleBot('5082139972:AAFIcYvf3kjLcm2msTHPODaNuXyIjPFrv9s', parse_mode=None)

#########################func per whitelist
whitelist=["Arcnold_Wp","PlugInSheep",]
def si_puo(message):
	user=message.from_user.username
	if user in whitelist:
		return True
	else:
		bot.reply_to(message, "Nice try "+str(user))
		return False
####################
def send_welcome(message):
	user=message.from_user.username
	if si_puo(message):
		bot.reply_to(message, get_ip())
@bot.message_handler(commands=['scarica_animu'])
def animu(message):
	user=message.from_user.username
	if si_puo(message):
		bot.reply_to(message, "Avvio lo scaricamento")
		modulo_scarica()
		bot.reply_to(message, "Ho finito di scaricare")
############blocco new anime
@bot.message_handler(commands=['new_animu'])
def new_animu(message):
	user=message.from_user.username
	if si_puo(message):
		msg=bot.send_message(message.chat.id, """Vuoi inserire un nuovo anime?
		Se si scrivi la cosa da cercare, altrimenti scrivi no""")
		bot.register_next_step_handler(msg, process_scegli_anime)
def process_scegli_anime(message):
	r=requests.get('https://www.animesaturn.tv/animelist?search='+message.text.strip())
	soup=bs(r.text,'html.parser')
	links=soup.find_all('a',class_='badge badge-archivio badge-light')
	nomi=[nome.text for nome in links]
	risposta=''
	for i in range(len(nomi)):
		risposta=risposta+str(i)+') '+nomi[i]+'\n'
	msg=bot.send_message(message.chat.id,'Quale anime inserisco (inserisci il numero oppure un "no")?\n'+risposta)
	bot.register_next_step_handler(msg,process_link_step,links)
def process_link_step(message,links):
	if message.text.lower()!="no":
		new_entry=[]
		new_entry.append(links[int(message.text)]['href'])
		print(new_entry)
		msg=bot.send_message(message.chat.id, """In che cartella lo salvo?""")
		bot.register_next_step_handler(msg, process_folder_step, new_entry)
	else:
		bot.send_message(message.chat.id, """Niente allora""")
def process_folder_step(message, new_entry):
	new_entry.append(message.text.strip())
	print(new_entry)
	msg=bot.send_message(message.chat.id, """Qual'è la stagione di questo anime? Inserisci solo un numero per piacere""")
	bot.register_next_step_handler(msg, process_nome_step, new_entry)
def process_nome_step(message, new_entry):
	seas_num=format(int(message.text.strip().title()),'02d')
	new_entry.append('Episode S'+seas_num)
	print(new_entry)
	bot.send_message(message.chat.id, str(new_entry))
	msg=bot.send_message(message.chat.id, """Così va bene?(si/NO)""")
	bot.register_next_step_handler(msg, process_verifica_step, new_entry)
def process_verifica_step(message, new_entry):
	if message.text.lower()=="si":
		with open("animelist.csv",mode='a') as animecsv: #link,cartella,nome,1,1
			csvfile=csv.writer(animecsv)
			new_entry.append("1") #scaricamento attivo
			new_entry.append('1') #new naming convention
			csvfile.writerow(new_entry)
			bot.send_message(message.chat.id, """Anime aggiunto :)""")
	else:
		bot.send_message(message.chat.id, """Niente allora""")
def smartsearch_anime(ricerca):
	r=requests.get('https://www.animesaturn.in/animelist?search='+ricerca)
	soup=bs(r.text,'html.parser')
	links=soup.find_all('a',class_='badge badge-archivio badge-light')
bot.infinity_polling(10)