From: a_magical_me <andrew@turnipmints.mooo.com>
Date: Sun, 3 Apr 2011 12:16:15 +0000 (-0700)
Subject: Add script for adding B/W locations
X-Git-Tag: veekun-promotions/2011041101~6^2~3
X-Git-Url: http://git.veekun.com/zzz-pokedex.git/commitdiff_plain/b1e513dbb14e2f514da1bb43f9c1ab2b2aa8ea80

Add script for adding B/W locations

in the hope that it will be useful in future generations.
---

diff --git a/scripts/add-bw-locations.py b/scripts/add-bw-locations.py
new file mode 100755
index 0000000..5f031a0
--- /dev/null
+++ b/scripts/add-bw-locations.py
@@ -0,0 +1,49 @@
+#!/usr/bin/env python2
+
+from codecs import open
+
+from pokedex.db import connect, identifier_from_name
+from pokedex.db.tables import Language
+from pokedex.db.tables import Location, LocationGameIndex
+
+session = connect()
+
+en = session.query(Language).filter_by(identifier='en').one() # English
+ja = session.query(Language).filter_by(identifier='ja').one() # Japanese
+
+with open("bw-location-names-en", "r", "utf-8") as f:
+    en_names = [line.rstrip("\n") for line in f]
+with open("bw-location-names-kanji", "r", "utf-8") as f:
+    ja_names = [line.rstrip("\n") for line in f]
+
+locations = {}
+for i, name in enumerate(zip(en_names, ja_names)):
+    if i == 0:
+        continue
+
+    en_name, ja_name = name
+    if not en_name:
+        continue
+
+    if name in locations:
+        loc = locations[name]
+    else:
+        loc = Location()
+        if en_name:
+            loc.name_map[en] = en_name
+        if ja_name:
+            loc.name_map[ja] = ja_name
+        loc.region_id = 5 # Unova
+        loc.identifier = identifier_from_name(en_name)
+
+        locations[name] = loc
+
+    lgi = LocationGameIndex()
+    lgi.location = loc
+    lgi.generation_id = 5 # Gen 5
+    lgi.game_index = i
+
+    session.add(loc)
+    session.add(lgi)
+
+session.commit()