mygpo/db/couchdb/pubsub.py

   1 from couchdbkit import MultipleResultsFound
   2
   3 from mygpo.pubsub.models import Subscription
   4 from mygpo.db.couchdb import get_pubsub_database, get_single_result
   5 from mygpo.decorators import repeat_on_conflict
   6
   7 import logging
   8 logger = logging.getLogger(__name__)
   9
  10
  11 def subscription_for_topic(topic):
  12     """ return the subscription for the given topic, one None """
  13
  14     db = get_pubsub_database()
  15     sub = get_single_result(db, 'subscriptions/by_topic',
  16             key          = topic,
  17             include_docs = True,
  18             reduce       = False,
  19             schema       = Subscription
  20         )
  21
  22     return sub
  23
  24
  25 @repeat_on_conflict(['subscription'])
  26 def set_subscription_verified(subscription):
  27     """ marks the pubsub subscription as verified """
  28     pdb = get_pubsub_database()
  29     subscription.verified = True
  30     pdb.save_doc(subscription)