subscription

This commit is contained in:
j 2011-01-11 20:26:08 +05:30
commit c577e6930b
7 changed files with 129 additions and 83 deletions

View file

@ -7,7 +7,7 @@ from django.db.models import Q, Manager
import models
def parseCondition(condition):
def parseCondition(condition, user):
'''
condition: {
value: "war"
@ -35,10 +35,9 @@ def parseCondition(condition):
exclude = False
if k == 'subscribed':
print "FXIME, subscribed needs work"
k = 'public'
v = True
if isinstance(v, bool): #featured and public flag
key = 'subscribed_users__username'
v = user.username
elif isinstance(v, bool): #featured and public flag
key = k
else:
if op == '=':
@ -58,7 +57,7 @@ def parseCondition(condition):
q = Q(**{key: v})
return q
def parseConditions(conditions, operator):
def parseConditions(conditions, operator, user):
'''
conditions: [
{
@ -81,14 +80,14 @@ def parseConditions(conditions, operator):
for condition in conditions:
if 'conditions' in condition:
q = parseConditions(condition['conditions'],
condition.get('operator', '&'))
condition.get('operator', '&'), user)
if q:
conn.append(q)
pass
else:
if condition.get('value', '') != '' or \
condition.get('operator', '') == '=':
conn.append(parseCondition(condition))
conn.append(parseCondition(condition, user))
if conn:
q = conn[0]
for c in conn[1:]:
@ -129,8 +128,9 @@ class ListManager(Manager):
#join query with operator
qs = self.get_query_set()
conditions = parseConditions(data['query']['conditions'],
data['query'].get('operator', '&'))
conditions = parseConditions(data['query'].get('conditions', []),
data['query'].get('operator', '&'),
user)
if conditions:
qs = qs.filter(conditions)

View file

@ -19,7 +19,7 @@ class List(models.Model):
created = models.DateTimeField(auto_now_add=True)
modified = models.DateTimeField(auto_now=True)
user = models.ForeignKey(User)
user = models.ForeignKey(User, related_name='lists')
name = models.CharField(max_length=255)
status = models.CharField(max_length=20, default='private')
_status = ['private', 'public', 'featured']
@ -28,6 +28,8 @@ class List(models.Model):
items = models.ManyToManyField('item.Item', related_name='lists',
through='ListItem')
subscribed_users = models.ManyToManyField(User, related_name='subscribed_lists')
objects = managers.ListManager()
def save(self, *args, **kwargs):
@ -53,22 +55,29 @@ class List(models.Model):
def __unicode__(self):
return u'%s (%s)' % (self.name, self.user)
def get_id(self):
return '%s:%s' % (self.user.username, self.name)
def editable(self, user):
#FIXME: make permissions work
if self.user == user or user.has_perm('Ox.admin'):
if self.user == user or user.is_staff:
return True
return False
def json(self, keys, user=None):
def json(self, keys=['id', 'name', 'user', 'query', 'status'], user=None):
response = {}
for key in keys:
if key == 'items':
response[key] = self.get_number_of_items(user)
elif key == 'id':
response[key] = self.get_id()
elif key == 'user':
response[key] = self.user.username
elif key == 'query':
if not self.query.get('static', False):
response[key] = self.query
elif key == 'ui':
response[key] = site_conf['uiDefaults']['list']
else:
response[key] = getattr(self, key)
return response
@ -81,3 +90,4 @@ class ListItem(models.Model):
def __unicode__(self):
return u'%s in %s' % (self.item, self.list)

View file

@ -160,7 +160,7 @@ def addList(request):
}
'''
data = json.loads(request.POST['data'])
if models.List.filter(name=data['name'], user=request.user).count() == 0:
if models.List.objects.filter(name=data['name'], user=request.user).count() == 0:
list = models.List(name = data['name'], user=request.user)
list.save()
response = json_response(status=200, text='created')
@ -186,8 +186,10 @@ def editList(request):
}
'''
data = json.loads(request.POST['data'])
list = get_object_or_404_json(models.List, pk=data['list'])
username, listname = data['id'].split(':')
list = get_object_or_404_json(models.List, user__username=username, name=listname)
if list.editable(request.user):
response = json_response()
for key in data:
if key in ('name', 'status', 'query'):
if key in data:
@ -200,7 +202,7 @@ def editList(request):
setattr(list, key, value)
else:
setattr(list, key, data[key])
if user.has_perm('Ox.admin') and 'featured' in data:
if request.user.is_staff and 'featured' in data:
list.featured = data['featured']
else:
response = json_response(status=403, text='not allowed')
@ -223,7 +225,7 @@ def removeList(request):
'''
data = json.loads(request.POST['data'])
user = request.user.username
if user.has_perm('Ox.admin') and 'user' in data:
if user.is_staff and 'user' in data:
user = data.get('user')
list = get_object_or_404_json(models.List, name=data['name'], user__username=user)
if list.editable(request.user):
@ -232,3 +234,49 @@ def removeList(request):
response = json_response(status=403, text='not allowed')
return render_to_json_response(response)
actions.register(removeList)
@login_required_json
def subscribeToList(request):
'''
param data {
id: listId,
user: username(only admins)
}
return {
status: {'code': int, 'text': string},
data: {
}
}
'''
data = json.loads(request.POST['data'])
username, listname = data['id'].split(':')
list = get_object_or_404_json(models.List, user__username=username, name=listname)
user = request.user
if list.subscribed_users.filter(username=user.username).count() == 0:
list.subscribed_users.add(user)
response = json_response()
return render_to_json_response(response)
actions.register(subscribeToList)
@login_required_json
def unsubscribeFromList(request):
'''
param data {
id: listId,
user: username(only admins)
}
return {
status: {'code': int, 'text': string},
data: {
}
}
'''
data = json.loads(request.POST['data'])
username, listname = data['id'].split(':')
list = get_object_or_404_json(models.List, user__username=username, name=listname)
user = request.user
list.subscribed_users.remove(user)
response = json_response()
return render_to_json_response(response)
actions.register(unsubscribeFromList)