Fixed get_country_code

ogn/ognutils.py

@@ -42,5 +42,8 @@ def get_ddb(csvfile=None):
 def get_country_code(latitude, longitude):
     geolocator = Nominatim()
     location = geolocator.reverse("%f, %f" % (latitude, longitude))
-    country_code = location.raw["address"]["country_code"]
+    try:
+        country_code = location.raw["address"]["country_code"]
+    except KeyError:
+        country_code = None
     return country_code

tests/test_ognutils.py

@@ -27,3 +27,9 @@ class TestStringMethods(unittest.TestCase):
         longitude = 11.0
         country_code = get_country_code(latitude, longitude)
         self.assertEquals(country_code, 'de')
+
+    def test_get_country_code_bad(self):
+        latitude = 0.0002274
+        longitude = -0.0009119
+        country_code = get_country_code(latitude, longitude)
+        self.assertEqual(country_code, None)