mygpo/db/couchdb/user.py

   1 from mygpo.cache import cache_result
   2 from mygpo.counter import Counter
   3
   4
   5 @cache_result(timeout=60)
   6 def get_num_listened_episodes(user):
   7     db = get_main_database()
   8     r = db.view('listeners/by_user_podcast',
   9             startkey    = [user._id, None],
  10             endkey      = [user._id, {}],
  11             reduce      = True,
  12             group_level = 2,
  13         )
  14
  15     return map(_wrap_num_listened, r)
  16
  17
  18 def _wrap_num_listened(obj):
  19     count = obj['value']
  20     podcast = obj['key'][1]
  21     return (podcast, count)
  22
  23
  24 @cache_result(timeout=60)
  25 def get_num_played_episodes(self, since=None, until={}):
  26     """ Number of played episodes in interval """
  27
  28     since_str = since.strftime('%Y-%m-%d') if since else None
  29     until_str = until.strftime('%Y-%m-%d') if until else {}
  30
  31     startkey = [self._id, since_str]
  32     endkey   = [self._id, until_str]
  33
  34     db = get_main_database()
  35     res = db.view('listeners/by_user',
  36             startkey = startkey,
  37             endkey   = endkey,
  38             reduce   = True,
  39         )
  40
  41     val = res.one()
  42     return val['value'] if val else 0
  43
  44
  45
  46
  47 @cache_result(timeout=60)
  48 def get_latest_episodes(self, count=10):
  49     """ Returns the latest episodes that the user has accessed """
  50
  51     startkey = [self._id, {}]
  52     endkey   = [self._id, None]
  53
  54     db = get_main_database()
  55     res = db.view('listeners/by_user',
  56             startkey     = startkey,
  57             endkey       = endkey,
  58             include_docs = True,
  59             descending   = True,
  60             limit        = count,
  61             reduce       = False,
  62         )
  63
  64     keys = [r['value'] for r in res]
  65     return list(episodes_by_id(keys))
  66
  67
  68
  69 @cache_result(timeout=60)
  70 def get_seconds_played(self, since=None, until={}):
  71     """ Returns the number of seconds that the user has listened
  72
  73     Can be selected by timespan, podcast and episode """
  74
  75     since_str = since.strftime('%Y-%m-%dT%H:%M:%S') if since else None
  76     until_str = until.strftime('%Y-%m-%dT%H:%M:%S') if until else {}
  77
  78     startkey = [self._id, since_str]
  79     endkey   = [self._id, until_str]
  80
  81     db = get_main_database()
  82     res = db.view('listeners/times_played_by_user',
  83             startkey = startkey,
  84             endkey   = endkey,
  85             reduce   = True,
  86         )
  87
  88     val = res.one()
  89     return val['value'] if val else 0
  90
  91
  92
  93 @cache_result(timeout=60*60)
  94 def suggestions_for_user(user):
  95     r = Suggestions.view('suggestions/by_user',
  96                 key          = user._id,
  97                 include_docs = True,
  98             )
  99
 100     if r:
 101         return r.first()
 102
 103     else:
 104         s = Suggestions()
 105         s.user = user._id
 106         return s
 107
 108
 109 @cache_result(timeout=60*60)
 110 def user_agent_stats():
 111     res = User.view('clients/by_ua_string',
 112         wrap_doc    = False,
 113         group_level = 1,
 114         stale       = 'update_after',
 115     )
 116
 117     return Counter(dict((r['key'], r['value']) for r in res))
 118
 119
 120 def deleted_users():
 121     users = User.view('users/deleted',
 122             include_docs = True,
 123             reduce       = False,
 124         )
 125     return list(users)
 126
 127
 128 def deleted_user_count():
 129     total = User.view('users/deleted',
 130             reduce = True,
 131         )
 132     return list(total)[0]['value'] if total else 0
 133
 134
 135
 136 @cache_result(timeout=60)
 137 def user_history(user, start, length):
 138     db = get_main_database()
 139     res = db.view('history/by_user',
 140             descending = True,
 141             startkey   = [user._id, None],
 142             endkey     = [user._id, {}],
 143             limit      = length,
 144             skip       = start,
 145         )
 146
 147     return map(_wrap_historyentry, res)
 148
 149
 150 @cache_result(timeout=60)
 151 def device_history(user, device, start, length):
 152     db = get_main_database()
 153
 154     res = self._db.view('history/by_device',
 155             descending = True,
 156             startkey   = [user._id, device.id, None],
 157             endkey     = [user._id, device.id, {}],
 158             limit      = length,
 159             skip       = start,
 160         )
 161
 162     return map(_wrap_historyentry, res)
 163
 164
 165 def _wrap_historyentry(action):
 166     return HistoryEntry.from_action_dict(action['action'])