mygpo/db/couchdb/user.py

   1 from mygpo.cache import cache_result
   2 from mygpo.counter import Counter
   3 from mygpo.decorators import repeat_on_conflict
   4 from mygpo.couch import get_main_database
   5 from mygpo.users.settings import FLATTR_TOKEN, FLATTR_AUTO, FLATTR_MYGPO, \
   6          FLATTR_USERNAME
   7 from mygpo.db import QueryParameterMissing
   8 from mygpo.db.couchdb.episode import episodes_by_id
   9
  10
  11 @cache_result(timeout=60)
  12 def get_num_listened_episodes(user):
  13
  14     if not user:
  15         raise QueryParameterMissing('user')
  16
  17     db = get_main_database()
  18     r = db.view('listeners/by_user_podcast',
  19             startkey    = [user._id, None],
  20             endkey      = [user._id, {}],
  21             reduce      = True,
  22             group_level = 2,
  23         )
  24
  25     return map(_wrap_num_listened, r)
  26
  27
  28 def _wrap_num_listened(obj):
  29     count = obj['value']
  30     podcast = obj['key'][1]
  31     return (podcast, count)
  32
  33
  34 @cache_result(timeout=60)
  35 def get_num_played_episodes(user, since=None, until={}):
  36     """ Number of played episodes in interval """
  37
  38     if not user:
  39         raise QueryParameterMissing('user')
  40
  41     since_str = since.strftime('%Y-%m-%d') if since else None
  42     until_str = until.strftime('%Y-%m-%d') if until else {}
  43
  44     startkey = [user._id, since_str]
  45     endkey   = [user._id, until_str]
  46
  47     db = get_main_database()
  48     res = db.view('listeners/by_user',
  49             startkey = startkey,
  50             endkey   = endkey,
  51             reduce   = True,
  52         )
  53
  54     val = res.one()
  55     return val['value'] if val else 0
  56
  57
  58
  59
  60 @cache_result(timeout=60)
  61 def get_latest_episodes(user, count=10):
  62     """ Returns the latest episodes that the user has accessed """
  63
  64     if not user:
  65         raise QueryParameterMissing('user')
  66
  67     startkey = [user._id, {}]
  68     endkey   = [user._id, None]
  69
  70     db = get_main_database()
  71     res = db.view('listeners/by_user',
  72             startkey     = startkey,
  73             endkey       = endkey,
  74             include_docs = True,
  75             descending   = True,
  76             limit        = count,
  77             reduce       = False,
  78         )
  79
  80     keys = [r['value'] for r in res]
  81     return episodes_by_id(keys)
  82
  83
  84
  85 @cache_result(timeout=60)
  86 def get_seconds_played(user, since=None, until={}):
  87     """ Returns the number of seconds that the user has listened
  88
  89     Can be selected by timespan, podcast and episode """
  90
  91     if not user:
  92         raise QueryParameterMissing('user')
  93
  94     since_str = since.strftime('%Y-%m-%dT%H:%M:%S') if since else None
  95     until_str = until.strftime('%Y-%m-%dT%H:%M:%S') if until else {}
  96
  97     startkey = [user._id, since_str]
  98     endkey   = [user._id, until_str]
  99
 100     db = get_main_database()
 101     res = db.view('listeners/times_played_by_user',
 102             startkey = startkey,
 103             endkey   = endkey,
 104             reduce   = True,
 105         )
 106
 107     val = res.one()
 108     return val['value'] if val else 0
 109
 110
 111
 112 @cache_result(timeout=60*60)
 113 def suggestions_for_user(user):
 114
 115     if not user:
 116         raise QueryParameterMissing('user')
 117
 118     from mygpo.users.models import Suggestions
 119     r = Suggestions.view('suggestions/by_user',
 120                 key          = user._id,
 121                 include_docs = True,
 122             )
 123
 124     if r:
 125         return r.first()
 126
 127     else:
 128         s = Suggestions()
 129         s.user = user._id
 130         return s
 131
 132
 133 @cache_result(timeout=60*60)
 134 def user_agent_stats():
 135     from mygpo.users.models import User
 136     res = User.view('clients/by_ua_string',
 137         wrap_doc    = False,
 138         group_level = 1,
 139         stale       = 'update_after',
 140     )
 141
 142     return Counter(dict((r['key'], r['value']) for r in res))
 143
 144
 145 def deleted_users():
 146     from mygpo.users.models import User
 147     users = User.view('users/deleted',
 148             include_docs = True,
 149             reduce       = False,
 150         )
 151     return list(users)
 152
 153
 154 def deleted_user_count():
 155     total = User.view('users/deleted',
 156             reduce = True,
 157         )
 158     return list(total)[0]['value'] if total else 0
 159
 160
 161
 162 @cache_result(timeout=60)
 163 def user_history(user, start, length):
 164
 165     if not user:
 166         raise QueryParameterMissing('user')
 167
 168     if length <= 0:
 169         return []
 170
 171     db = get_main_database()
 172     res = db.view('history/by_user',
 173             descending = True,
 174             startkey   = [user._id, {}],
 175             endkey     = [user._id, None],
 176             limit      = length,
 177             skip       = start,
 178         )
 179
 180     return map(_wrap_historyentry, res)
 181
 182
 183 @cache_result(timeout=60)
 184 def device_history(user, device, start, length):
 185
 186     if not user:
 187         raise QueryParameterMissing('user')
 188
 189     if not device:
 190         raise QueryParameterMissing('device')
 191
 192     if length <= 0:
 193         return []
 194
 195     db = get_main_database()
 196
 197     res = db.view('history/by_device',
 198             descending = True,
 199             startkey   = [user._id, device.id, {}],
 200             endkey     = [user._id, device.id, None],
 201             limit      = length,
 202             skip       = start,
 203         )
 204
 205     return map(_wrap_historyentry, res)
 206
 207
 208 @repeat_on_conflict(['user'])
 209 def update_flattr_settings(user, token, enabled=None, flattr_mygpo=False,
 210         username=None):
 211     """ Updates the Flattr settings of a user """
 212
 213     if enabled is not None:
 214         user.settings[FLATTR_AUTO.name] = enabled
 215
 216     if token is not None:
 217         user.settings[FLATTR_TOKEN.name] = token
 218
 219     if flattr_mygpo is not None:
 220         user.settings[FLATTR_MYGPO.name] = flattr_mygpo
 221
 222     if username is not None:
 223         user.settings[FLATTR_USERNAME.name] = username
 224
 225     user.save()
 226
 227
 228 def _wrap_historyentry(action):
 229     from mygpo.users.models import HistoryEntry
 230     return HistoryEntry.from_action_dict(action['value'])