From f020fcf9770b540c3a6c35e6388b02d37ad5b20b Mon Sep 17 00:00:00 2001
From: Neil Booth <kyuupichan@gmail.com>
Date: Tue, 8 Nov 2016 08:31:43 +0900
Subject: [PATCH] Update query.py

---
 query.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/query.py b/query.py
index faea87a..8fe7c94 100755
--- a/query.py
+++ b/query.py
@@ -13,11 +13,10 @@ Not currently documented; might become easier to use in future.
 '''
 
 
-import os
 import sys
 
 from server.env import Env
-from server.block_processor import BlockProcessor
+from server.DB import DB
 from lib.hash import hash_to_str
 
 
@@ -40,9 +39,8 @@ def count_entries(db):
 
 def main():
     env = Env()
+    bp = DB(env)
     coin = env.coin
-    os.chdir(env.db_dir)
-    bp = BlockProcessor(env, None)
     if len(sys.argv) == 1:
         count_entries(bp.db)
         return