iploc: Use IP2Location_cached.py

Change-Id: Ic0cee6ccaa262312bcc10111f13c0f6f81e5de85

1 files changed, 3 insertions, 1 deletions

diff --git a/iploc.py b/iploc.py
index 3b6085a..b88e36c 100644
--- a/iploc.py
+++ b/iploc.py
@@ -1,7 +1,8 @@
 #!/usr/bin/python
 # Usage: iploc.py </path/to/log> </path/to/output>
 
-import IP2Location
+#import IP2Location
+import IP2Location_cached as IP2Location
 import atexit
 import fileinput
 import re
@@ -80,6 +81,7 @@ def main(file_names):
 
     IP2LocObj = IP2Location.IP2Location()
     IP2LocObj.open(get_path(config["LOC_DB_FILE"]))
+    IP2LocObj.load_index()
 
     REVERSE_DNS_DB.open(get_path(config["DNSHISTORY_DB"]), dbtype=db.DB_BTREE)
     atexit.register(REVERSE_DNS_DB.close)