mygpo/db/couchdb/common.py

   1 from mygpo.db.couchdb import get_main_database
   2 from mygpo.db import QueryParameterMissing
   3 from mygpo.db.couchdb.utils import multi_request_view
   4
   5
   6 def missing_slug_count(doc_type, start, end):
   7
   8     if not doc_type:
   9         raise QueryParameterMissing('doc_type')
  10
  11     if not start:
  12         raise QueryParameterMissing('start')
  13
  14     if not end:
  15         raise QueryParameterMissing('end')
  16
  17
  18     db = get_main_database()
  19     res = db.view('slugs/missing',
  20             startkey     = [doc_type] + end,
  21             endkey       = [doc_type] + start,
  22             descending   = True,
  23             reduce       = True,
  24             group        = True,
  25             group_level  = 1,
  26         )
  27     return res.first()['value'] if res else 0
  28
  29
  30 def missing_slugs(doc_type, start, end, wrapper, **kwargs):
  31
  32     if not doc_type:
  33         raise QueryParameterMissing('doc_type')
  34
  35     if not start:
  36         raise QueryParameterMissing('start')
  37
  38     if not end:
  39         raise QueryParameterMissing('end')
  40
  41
  42     db = get_main_database()
  43     return multi_request_view(db, 'slugs/missing',
  44             startkey     = [doc_type] + end,
  45             endkey       = [doc_type] + start,
  46             descending   = True,
  47             include_docs = True,
  48             reduce       = False,
  49             wrapper      = wrapper,
  50             auto_advance = False,
  51             **kwargs
  52         )