from bs4 import BeautifulSoup,NavigableString
import urllib2,sys,re,codecs,base64
import json

print('This script will export your anime-planet.com anime list and saves it to anime-planet.xml')
apusername = raw_input("Enter your Anime-Planet username: ")
malusername = raw_input("Enter your MAL username: ")
malpassword = raw_input("Enter your MAL password: ")
headers = { 'User-Agent' : 'Mozilla/5.0' }
baseURL = 'http://www.anime-planet.com/users/'+apusername+'/anime'
req = urllib2.Request(baseURL, None, headers)
html = urllib2.urlopen(req).read()
html = BeautifulSoup(html)
pageNumber = int (html.find('li','next').findPrevious('li').next.contents[0])
delimiter = '\t'

f = codecs.open('anime-planet2.xml', 'w', 'utf-8')
f.write ('<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n')
f.write ('<myanimelist>\n')

print('Exporting rough variant of myanimelist format... \n')
for i in range(1,pageNumber+1):
                baseURL = 'http://www.anime-planet.com/users/'+apusername+'/anime?page='+str(i)
                req = urllib2.Request(baseURL, None, headers)
                html = urllib2.urlopen(req).read()
                html = BeautifulSoup(html)
                for animeItem in html.findAll('tr')[1:]:
                        animeItem = BeautifulSoup(animeItem.renderContents())
                        animeName = '' + animeItem.a.text
                        queryName = re.sub('[^A-Za-z0-9]+', '%20', animeName)
                        titlereq = urllib2.Request("http://myanimelist.net/api/anime/search.xml?q=" + queryName)
                        titlereq.add_header("Authorization", "Basic %s" % base64.encodestring("%s:%s" % (malusername, malpassword)).replace("n", ""))
                        titlereq.add_header("User-Agent", "Mozilla/5.0")
                        queryTitle = urllib2.urlopen(titlereq).read()
                        print(animeName)
                        search=json.loads(queryTitle,"utf8")
                        for x in search:
                                try:
                                        if animeName.lower()==x["title"].lower():
                                                animeID=str(x["id"])
                                        elif animeName.lower()in [y.lower() for y in x["other_titles"]["english"]]:
                                                animeID=str(x["id"])
                                        elif animeName.lower() in [j.lower() for j in x["other_titles"]["synonyms"]]:
                                                animeID=str(x["id"])
                                except KeyError as e:
                                        pass
                        if animeItem.find('td','tableStatus').text.replace('status box','').replace("\t", "").replace("\n", "").replace("\r", "").replace(" ", "")=="Watched":
                                status="Completed"
                        elif animeItem.find('td','tableStatus').text.replace('status box','').replace("\t", "").replace("\n", "").replace("\r", "").replace(" ", "")=="Stalled":
                                status="On-Hold"
                        elif animeItem.find('td','tableStatus').text.replace('status box','').replace("\t", "").replace("\n", "").replace("\r", "").replace(" ", "")=="WanttoWatch":
                                status="Plan to Watch"
                        elif animeItem.find('td','tableStatus').text.replace('status box','').replace("\t", "").replace("\n", "").replace("\r", "").replace(" ", "")=="Won'tWatch":
                                status="Dropped"
                                continue
                        else:
                                status=animeItem.find('td','tableStatus').text.replace('status box','').replace("\t", "").replace("\n", "").replace("\r", "").replace(" ", "")
                        f.write ('\t<anime>\n');
                        f.write ('\t\t<series_animedb_id>'+ animeID +'</series_animedb_id>\n');
                        f.write ('\t\t<series_title><![CDATA['+ animeName +']]></series_title>\n');
                        f.write ('\t\t<series_type>' + animeItem.find('td','tableType').text + '</series_type>\n');
                        f.write ('\t\t<my_id>0</my_id>\n');
                        f.write ('\t\t<my_watched_episodes>'+ animeItem.find('td','tableEps').text.replace('&nbsp;','1').replace("\t", "").replace("\n", "").replace("\r", "").replace(" ", "") +'</my_watched_episodes>\n');
                        f.write ('\t\t<my_start_date>0000-00-00</my_start_date>\n');
                        f.write ('\t\t<my_finish_date>0000-00-00</my_finish_date>\n');
                        f.write ('\t\t<my_fansub_group><![CDATA[]]></my_fansub_group>\n');
                        f.write ('\t\t<my_rated></my_rated>\n');
                        f.write ('\t\t<my_score>' + str(int(float(animeItem.img['name'])*2)).replace("\t", "").replace("\n", "").replace("\r", "").replace(" ", "") + '</my_score>\n');
                        f.write ('\t\t<my_dvd></my_dvd>\n');
                        f.write ('\t\t<my_storage></my_storage>\n');
                        f.write ('\t\t<my_status>' + status +'</my_status>\n');
                        f.write ('\t\t<my_comments><![CDATA[]]></my_comments>\n');
                        f.write ('\t\t<my_times_watched>0</my_times_watched>\n');
                        f.write ('\t\t<my_rewatch_value></my_rewatch_value>\n');
                        f.write ('\t\t<my_downloaded_eps>0</my_downloaded_eps>\n');
                        f.write ('\t\t<my_tags><![CDATA[]]></my_tags>\n');
                        f.write ('\t\t<my_rewatching>0</my_rewatching>\n');
                        f.write ('\t\t<my_rewatching_ep>0</my_rewatching_ep>\n');
                        f.write ('\t\t<update_on_import>1</update_on_import>\n');
                        f.write ('\t</anime>\n\n');



f.write ('</myanimelist>\n')
print('Done, see anime-planet.xml and anime_list.txt')