Skip to content
This repository was archived by the owner on Oct 14, 2018. It is now read-only.
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions pyzipcode/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from settings import db_location
from .settings import db_location
try:
import sqlite3
except ImportError:
Expand All @@ -24,7 +24,7 @@ def query(self, sql, args):
# then just give up...
try:
conn = sqlite3.connect(db_location)
except sqlite3.OperationalError, x:
except sqlite3.OperationalError as x:
retry_count += 1
time.sleep(0.001)

Expand Down
4 changes: 2 additions & 2 deletions pyzipcode/import.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import os
import csv
try:
from settings import db_location
from .settings import db_location
except:
from pyzipcode.settings import db_location

Expand All @@ -17,7 +17,7 @@
c.execute("CREATE INDEX state_index ON ZipCodes(state);")

reader = csv.reader(open('zipcode.csv', "rb"))
reader.next() # prime it
next(reader) # prime it

for row in reader:
zip, city, state, lat, longt, timezone, dst = row
Expand Down
12 changes: 6 additions & 6 deletions pyzipcode/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@ def setUp(self):

def test_retrieves_zip_code_information(self):
zip = self.db['54115']
self.assertEquals(zip.zip, '54115')
self.assertEquals(zip.city, "De Pere")
self.assertEquals(zip.state, "WI")
self.assertEqual(zip.zip, '54115')
self.assertEqual(zip.city, "De Pere")
self.assertEqual(zip.state, "WI")

def test_correct_longitude_value(self):
zip = self.db[54115]
Expand All @@ -24,19 +24,19 @@ def test_correct_latitude_value(self):

def test_correct_timezone(self):
zip = self.db[54115]
self.assertEquals(zip.timezone, -6)
self.assertEqual(zip.timezone, -6)

def test_correct_dst(self):
zip = self.db[54115]
self.assertEquals(zip.dst, 1)
self.assertEqual(zip.dst, 1)

def test_radius(self):
zips = self.db.get_zipcodes_around_radius('54115', 30)
self.assertTrue('54304' in [zip.zip for zip in zips])

def test_find_zip_by_city(self):
zip = self.db.find_zip(city="De Pere")[0]
self.assertEquals('54115', zip.zip)
self.assertEqual('54115', zip.zip)

def test_find_zip_by_city_with_multiple_zips(self):
zips = self.db.find_zip(city="Green Bay")
Expand Down
8 changes: 0 additions & 8 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,6 @@

version = '0.4'

try:
import sqlite3
except ImportError:
requires = ['pysqlite']
else:
requires = []

setup(name='pyzipcode',
version=version,
description="query zip codes and location data",
Expand All @@ -24,7 +17,6 @@
package_data={'pyzipcode': ['zipcodes.db']},
include_package_data=True,
zip_safe=False,
install_requires=requires,
entry_points="""
# -*- Entry points: -*-
""",
Expand Down