Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Fix for missing episode names.

  • Loading branch information...
commit 68ec61388ca54729b69bffb674da97f9877c4d91 1 parent 554ba4a
@skorokithakis authored
Showing with 6 additions and 1 deletion.
  1. +6 −1 importer.py
View
7 importer.py
@@ -11,7 +11,8 @@
def import_data(filename):
"""Import episode names and ratings from a file."""
- regex = re.compile(""""(?P<show_name>.*?)"\s+\((?P<year>\d+)(?:|/.*?)\)\s+\{(?P<episode_name>.*?) \(\#(?P<season_no>\d+)\.(?P<episode_no>\d+)\)\}""")
+ regex = re.compile(""""(?P<show_name>.*?)"\s+\((?P<year>\d+)(?:|/.*?)\)\s+\{(?P<episode_name>.*?)\s?\(\#(?P<season_no>\d+)\.(?P<episode_no>\d+)\)\}""")
+
with codecs.open(filename, "r", "latin-1") as ratings:
# Generate all the lines that matched.
matches = (match for match in (regex.search(line.strip()) for line in ratings) if match)
@@ -24,6 +25,10 @@ def import_data(filename):
for field in ["show_name", "year", "episode_name", "episode_no", "season_no"]:
episode[field] = match.group(field)
+ # If the episode has no name it is given the same name as on imdb.com for consistency.
+ if not episode["episode_name"]:
+ episode["episode_name"] = "Episode #%s.%s" % (episode["season_no"], episode["episode_no"])
+
try:
show = session.query(Show).filter_by(name=episode["show_name"], year=episode["year"]).one()
except sqlalchemy.orm.exc.NoResultFound:
Please sign in to comment.
Something went wrong with that request. Please try again.