import os
import requests
import XML2Dict # The PyPI installer doesn't work; get this from http://blha303.com.au/XML2Dict.zip
from sys import argv

apikey = argv[1] # Get your API key from http://www.thetvdb.com/wiki/index.php?title=Programmers_API
ids = {"The.Daily.Show": "71256", "The.Colbert.Report": "79274", "Craig.Ferguson": "73387"}
eplookupurl = "http://thetvdb.com/api/GetEpisodeByAirDate.php?apikey=%s&seriesid=%s&airdate=%s"
srslookupurl = "http://thetvdb.com/data/series/%s/"
xml = XML2Dict.encoder.XML2Dict()
continuing = False
for filename in os.listdir('.'):
  for k,v in ids.items():
    if k in filename:
      airdate = "-".join(filename.split(".")[len(k.split(".")):len(k.split("."))+3])
      print "[%s] Processing %s" % (k, filename)
      data = xml.parse(requests.get(eplookupurl % (apikey, v, airdate)).text)
      data["Data"]["Episode"]["SeriesName"] = xml.parse(requests.get(srslookupurl % v).text)["Data"]["Series"]["SeriesName"]
      out = "{SeriesName} s{SeasonNumber}e{EpisodeNumber}.".format(**data["Data"]["Episode"]) + filename.split(".")[-1]
      os.rename(filename, out)
      print "[%s] Renamed to %s" % (k, out)
      continuing = True
      continue
  if continuing:
    continuing = False
    continue
  print "Skipping " + filename