sql row factory
This commit is contained in:
parent
fa152f4d46
commit
e3eaba33ed
|
@ -31,11 +31,12 @@ class handler(asyncTornado.asyncRequestHandler):
|
||||||
raise Exception("Missing required arguments")
|
raise Exception("Missing required arguments")
|
||||||
|
|
||||||
method = args_filter[0]
|
method = args_filter[0]
|
||||||
method_value = self.request.arguments[method]
|
method_value = self.get_argument(method)
|
||||||
|
|
||||||
cur = glob.sql.cursor()
|
conn = glob.new_sql()
|
||||||
|
cur = conn.cursor()
|
||||||
|
|
||||||
sql = SQL_STRUCT["main"] % SQL_STRUCT["method"]
|
sql = SQL_STRUCT["main"] % SQL_STRUCT[method]
|
||||||
if method == "timestamp":
|
if method == "timestamp":
|
||||||
sql = sql % method_value
|
sql = sql % method_value
|
||||||
else:
|
else:
|
||||||
|
@ -45,12 +46,14 @@ class handler(asyncTornado.asyncRequestHandler):
|
||||||
data = cur.fetchone()
|
data = cur.fetchone()
|
||||||
|
|
||||||
cur.close()
|
cur.close()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
status_code = 200
|
status_code = 200
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
status_code = 400
|
status_code = 400
|
||||||
data["status"] = status_code
|
data["status"] = status_code
|
||||||
data["message"] = str(e)
|
data["message"] = str(e)
|
||||||
|
print("Err: %s" % e)
|
||||||
finally:
|
finally:
|
||||||
self.write( json.dumps(data) )
|
self.write( json.dumps(data) )
|
||||||
self.set_header("Content-Type", "application/json")
|
self.set_header("Content-Type", "application/json")
|
||||||
|
|
0
helpers/__init__.py
Normal file
0
helpers/__init__.py
Normal file
5
helpers/databaseHelper.py
Normal file
5
helpers/databaseHelper.py
Normal file
|
@ -0,0 +1,5 @@
|
||||||
|
def dict_factory(cursor, row):
|
||||||
|
d = {}
|
||||||
|
for idx, col in enumerate(cursor.description):
|
||||||
|
d[col[0]] = row[idx]
|
||||||
|
return d
|
4
main.py
4
main.py
|
@ -53,7 +53,7 @@ def build_database():
|
||||||
f = open("wayback.db", "w")
|
f = open("wayback.db", "w")
|
||||||
f.close()
|
f.close()
|
||||||
|
|
||||||
glob.sql = sqlite3.connect("wayback.db")
|
glob.sql = glob.new_sql()
|
||||||
cur = glob.sql.cursor()
|
cur = glob.sql.cursor()
|
||||||
|
|
||||||
with open("database_structs.sql", "r") as f:
|
with open("database_structs.sql", "r") as f:
|
||||||
|
@ -69,7 +69,7 @@ if __name__ == "__main__":
|
||||||
build_database()
|
build_database()
|
||||||
|
|
||||||
if glob.sql == None:
|
if glob.sql == None:
|
||||||
glob.sql = sqlite3.connect("wayback.db")
|
glob.sql = glob.new_sql()
|
||||||
|
|
||||||
print("Serving at %s:%s" % (glob.config["server"]["host"], glob.config["server"]["port"]))
|
print("Serving at %s:%s" % (glob.config["server"]["host"], glob.config["server"]["port"]))
|
||||||
print("To stop server press CTRL + C")
|
print("To stop server press CTRL + C")
|
||||||
|
|
|
@ -1,5 +1,7 @@
|
||||||
import json
|
import json
|
||||||
|
import sqlite3
|
||||||
from multiprocessing.pool import ThreadPool
|
from multiprocessing.pool import ThreadPool
|
||||||
|
from helpers import databaseHelper
|
||||||
|
|
||||||
with open("config.json", "r") as f:
|
with open("config.json", "r") as f:
|
||||||
config = json.load(f)
|
config = json.load(f)
|
||||||
|
@ -7,3 +9,8 @@ with open("config.json", "r") as f:
|
||||||
app = None
|
app = None
|
||||||
sql = None
|
sql = None
|
||||||
pool = ThreadPool(config["threads"])
|
pool = ThreadPool(config["threads"])
|
||||||
|
|
||||||
|
def new_sql():
|
||||||
|
db = sqlite3.connect("wayback.db")
|
||||||
|
db.row_factory = databaseHelper.dict_factory
|
||||||
|
return db
|
Loading…
Reference in New Issue
Block a user