SGJP / Kuznia

#-*- coding:utf-8 -*-

from django.contrib.auth.models import User, Group
from common.decorators import ajax
from dictionary.models import Vocabulary

@ajax(method='post')
def set_group(request, user_id, group_id, set):
  user = User.objects.get(pk=user_id)
  group = Group.objects.get(pk=group_id)
  if set:
    user.groups.add(group)
  else:
    user.groups.remove(group)
  return {}

@ajax(method='post')
def save_default_owner(request, vocab_id):
  us = request.user.usersettings
  us.default_owner = Vocabulary.objects.get(pk=vocab_id)
  us.save()
  return {}