make version of ts_mon compatible with appengine

appengine/sheriff_o_matic/ts_alerts.py

 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import alerts_history
 import datetime_encoder
 import hashlib
 import json
 import logging
 import webapp2
 import zlib
 
+from alerts import increment_monarch
 from datetime import datetime as dt
 from google.appengine.api import memcache
 from google.appengine.api import users
 from google.appengine.datastore import datastore_query
 from google.appengine.ext import ndb
 
 
 ALLOWED_APP_IDS = ('google.com:monarch-email-alerts-parser')
 INBOUND_APP_ID = 'X-Appengine-Inbound-Appid'
 
@@ skipping 60 matching lines @@
 
 
 class TimeSeriesAlertsHandler(webapp2.RequestHandler):
   ALERT_TYPE = 'ts-alerts'
   MEMCACHE_COMPRESSION_LEVEL = 9
   # Alerts which have continued to fire are re-sent every 5 minutes, so stale
   # alerts older than 300 seconds are replaced by incoming alerts.
   STALE_ALERT_TIMEOUT = 300
 
   def get(self, key=None):
+    increment_monarch('ts-alerts')
     self.remove_expired_alerts()
     if not users.get_current_user():
       results = {'date': dt.utcnow(),
                  'redirect-url': users.create_login_url(self.request.uri)}
       self.write_json(results)
       return
 
     if key:
       logging.info('getting the key: ' + key)
       try:
@@ skipping 114 matching lines @@
   def set_memcache(self, key, data):
     json_data = generate_json_dump(data, False)
     compression_level = self.MEMCACHE_COMPRESSION_LEVEL
     compressed = zlib.compress(json_data, compression_level)
     memcache.set(key, compressed)
 
 
 class TimeSeriesAlertsHistory(alerts_history.AlertsHistory):
 
   def get(self, timestamp=None):
+    increment_monarch('ts-alerts-history')
     result_json = {}
     if not users.get_current_user():
       result_json['login-url'] = users.create_login_url(self.request.uri)
       return result_json
 
     alerts = TSAlertsJSON.query_active().fetch()
     if timestamp:
       try:
         time = dt.fromtimestamp(int(timestamp))
       except ValueError:
@@ skipping 27 matching lines @@
     self.response.headers['Content-Type'] = 'application/json'
     data = generate_json_dump(data)
     self.response.write(data)
 
 
 app = webapp2.WSGIApplication([
     ('/ts-alerts', TimeSeriesAlertsHandler),
     ('/ts-alerts/(.*)', TimeSeriesAlertsHandler),
     ('/ts-alerts-history', TimeSeriesAlertsHistory),
     ('/ts-alerts-history/(.*)', TimeSeriesAlertsHistory)])