userhandler/views.py

   1 from django.shortcuts import render, redirect
   2 from sf_t3d.decorators import login_required
   3 from django.http import HttpResponse
   4 import json
   5 import logging
   6 import authosm.utils as osmutils
   7 from lib.osm.osmclient.clientv2 import Client
   8
   9 logging.basicConfig(level=logging.DEBUG)
  10 log = logging.getLogger(__name__)
  11
  12
  13 @login_required
  14 def user_list(request):
  15     user = osmutils.get_user(request)
  16     client = Client()
  17     result = client.user_list(user.get_token())
  18     result = {
  19         'users': result['data'] if result and result['error'] is False else []
  20     }
  21     return __response_handler(request, result, 'user_list.html')
  22
  23
  24 @login_required
  25 def create(request):
  26     user = osmutils.get_user(request)
  27     client = Client()
  28     user_data ={
  29         "username": request.POST['username'],
  30         "password": request.POST['password'],
  31         "projects": request.POST.getlist('projects')
  32     }
  33
  34     result = client.user_create(user.get_token(), user_data)
  35
  36     return __response_handler(request, result, 'users:list', to_redirect=True)
  37
  38
  39 @login_required
  40 def delete(request, user_id=None):
  41     user = osmutils.get_user(request)
  42     try:
  43         client = Client()
  44         del_res = client.user_delete(user.get_token(), user_id)
  45     except Exception as e:
  46         log.exception(e)
  47     return __response_handler(request, {}, 'users:list', to_redirect=True, )
  48
  49 @login_required
  50 def update(request, user_id=None):
  51     user = osmutils.get_user(request)
  52     try:
  53         client = Client()
  54         projects_old = request.POST.get('projects_old').split(',')
  55         projects_new = request.POST.getlist('projects')
  56         default_project = request.POST.get('default_project')
  57         projects_new.append(default_project)
  58         projects_to_add = list(set(projects_new) - set(projects_old))
  59         projects_to_remove = list(set(projects_old) - set(projects_new))
  60
  61         project_payload = {}
  62
  63         for p in projects_to_remove:
  64             project_payload["$"+str(p)] = None
  65         for p in projects_to_add:
  66             if p not in projects_old:
  67                 project_payload["$+"+str(p)] = str(p)
  68         project_payload["$" + default_project] = None
  69         project_payload["$+[0]"] = default_project
  70         payload = {}
  71         if project_payload:
  72             payload["projects"] = project_payload
  73         if request.POST.get('password') and request.POST.get('password') is not '':
  74             payload["password"] = request.POST.get('password')
  75
  76         update_res = client.user_update(user.get_token(), user_id, payload)
  77     except Exception as e:
  78         log.exception(e)
  79         update_res = {'error': True, 'data': str(e)}
  80     if update_res['error']:
  81         return __response_handler(request, update_res['data'], url=None,
  82                                   status=update_res['data']['status'] if 'status' in update_res['data'] else 500)
  83     else:
  84         return __response_handler(request, {}, url=None, status=200)
  85         #return __response_handler(request, {}, 'users:list', to_redirect=True, )
  86
  87
  88 def __response_handler(request, data_res, url=None, to_redirect=None, *args, **kwargs):
  89     raw_content_types = request.META.get('HTTP_ACCEPT', '*/*').split(',')
  90     if 'application/json' in raw_content_types or url is None:
  91         return HttpResponse(json.dumps(data_res), content_type="application/json", *args, **kwargs)
  92     elif to_redirect:
  93         return redirect(url, *args, **kwargs)
  94     else:
  95         return render(request, url, data_res)