round hour when inserting

1 files changed, 7 insertions, 4 deletions

diff --git a/src/perform_queries.py b/src/perform_queries.py
index 0d77355..688f4c5 100755
--- a/src/perform_queries.py
+++ b/src/perform_queries.py
@@ -15,7 +15,8 @@ class dns_queries:
         self.services = services
 
 class single_query:
-    def __init__(self, cursor, vpn_id, dns_id, service_id):
+    def __init__(self, hour, cursor, vpn_id, dns_id, service_id):
+        self.hour = hour
         self.cursor = cursor
         self.vpn_id = vpn_id
         self.dns_id = dns_id
@@ -67,9 +68,10 @@ def resolve_call_back(mydata, status, result):
     # write to database
     query.cursor.execute('''
     INSERT INTO user_side_responses (date, result, dns_id, service_id, vpn_id)
-    VALUES (current_timestamp, %s, %s, %s, %s)
+    VALUES (%s, %s, %s, %s, %s)
     RETURNING id
-    ''', (result_info, query.dns_id, query.service_id, query.vpn_id))
+    ''', (query.hour, result_info, query.dns_id,
+          query.service_id, query.vpn_id))
 
     responses_id = query.cursor.fetchone()[0]
 
@@ -93,7 +95,8 @@ def query_dns(dns_queries):
     for service_id, service_name in dns_queries.services:
         print("starting resolution of {} through {}".format(service_name,
                                                             dns_queries.dns_IP))
-        query = single_query(cursor, vpn_id, dns_queries.dns_id, service_id)
+        query = single_query(hour, cursor, vpn_id,
+                             dns_queries.dns_id, service_id)
         
         ctx.resolve_async(service_name, query, resolve_call_back,
                           unbound.RR_TYPE_A, unbound.RR_CLASS_IN)