import os
import requests
from XML2Dict import encoder
from sys import argv

apikey = argv[1] # Get your API key from http://www.thetvdb.com/wiki/index.php?title=Programmers_API
dsid = "71256"
crid = "79274"
apiurl = "http://thetvdb.com/api/GetEpisodeByAirDate.php?apikey=%s&seriesid=%s&airdate=%s"
xml = encoder.XML2Dict()
for filename in os.listdir('.'):
  if ".py" in filename or filename == "XML2Dict":
    continue
  airdate = "-".join(filename.split(".")[3:6])
  if "The.Colbert.Report" in filename:
    data = xml.parse(requests.get(apiurl % (apikey, crid, airdate)).text)
    os.rename(filename, "The Colbert Report s{SeasonNumber}e{EpisodeNumber}.mp4".format(**data["Data"]["Episode"]))
  else:
    data = xml.parse(requests.get(apiurl % (apikey, dsid, airdate)).text)
    os.rename(filename, "The Daily Show s{SeasonNumber}e{EpisodeNumber}.mp4".format(**data["Data"]["Episode"]))