have a single function to serialize unix timestamps as redis keys

autovpn/acct.py

@@ -4,6 +4,10 @@ import redis
 import time
 
 
+def time2day(unix_stamp):
+    return time.strftime('%Y%m%d', time.gmtime(unix_stamp))
+
+
 class Accounting(object):
 
     def __init__(self, local_db, aggr_db=None):
@@ -44,7 +48,7 @@ class Accounting(object):
         for data in conns:
             pipe.lpush('connections:%s' % cn, data)
             conn_info = json.loads(data)
-            day = time.strftime('%Y%m%d', time.gmtime(conn_info['end_time']))
+            day = time2day(conn_info['end_time'])
             aggr_day = aggr.setdefault(day, {'conn_time': 0,
                                              'bytes_sent': 0,
                                              'bytes_recv': 0})
@@ -98,7 +102,7 @@ class Accounting(object):
     def get_aggregate_counts(self, cn, when=None):
         if not when:
             when = time.time()
-        day = time.strftime('%Y%m%d', time.gmtime(when))
+        day = time2day(when)
         data = self._aggr_db.hget('aggr:%s' % cn, day)
         if data:
             return json.loads(data)