call count

tfrce · Jan 30, 2014 · ed6c368 · ed6c368
1 parent f6a7676
commit ed6c368
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -9,7 +9,7 @@
 from raven.contrib.flask import Sentry
 from twilio import TwilioRestException
 
-from models import aggregate_stats, log_call
+from models import aggregate_stats, log_call, call_count
 from utils import play_or_say
 from political_data import PoliticalData
 
@@ -281,7 +281,7 @@ def demo():
 
 @app.route('/count')
 def count():
-    return db.session.query(Call).count()
+    return call_count(db)
 
 
 @app.route('/stats')

diff --git a/models.py b/models.py
@@ -82,6 +82,11 @@ def log_call(db, params, campaign, request):
     except:
         logging.error('Failed to log call: {}'.format(kwds))
 
+def call_count(db):
+    try:
+        return db.session.query(Call).count()
+    except:
+        return 0
 
 def aggregate_stats(cid):
     zipcodes = db.session.query(Call.zipcode, func.Count(Call.zipcode)) \