mygpo/api/backend.py

   1 import uuid
   2
   3 from django.db import transaction, IntegrityError
   4
   5 from mygpo.users.settings import STORE_UA
   6 from mygpo.users.models import Client
   7
   8 import logging
   9 logger = logging.getLogger(__name__)
  10
  11
  12 def get_device(user, uid, user_agent, undelete=True):
  13     """
  14     Loads or creates the device indicated by user, uid.
  15
  16     If the device has been deleted and undelete=True, it is undeleted.
  17     """
  18
  19     store_ua = user.profile.settings.get_wksetting(STORE_UA)
  20
  21     # list of fields to update -- empty list = no update
  22     update_fields = []
  23
  24     try:
  25         with transaction.atomic():
  26             client = Client(id=uuid.uuid1(), user=user, uid=uid)
  27             client.clean_fields()
  28             client.clean()
  29             client.save()
  30
  31     except IntegrityError:
  32         client = Client.objects.get(user=user, uid=uid)
  33
  34     if client.deleted and undelete:
  35         client.deleted = False
  36         update_fields.append('deleted')
  37
  38     if store_ua and user_agent and client.user_agent != user_agent:
  39         client.user_agent = user_agent
  40         update_fields.append('user_agent')
  41
  42     if update_fields:
  43         client.save(update_fields=update_fields)
  44
  45     return client