userhandler/views.py

   1 #
   2 #   Copyright 2018 EveryUP Srl
   3 #
   4 #   Licensed under the Apache License, Version 2.0 (the "License");
   5 #   you may not use this file except in compliance with the License.
   6 #   You may obtain a copy of the License at
   7 #
   8 #       http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 #   Unless required by applicable law or agreed to in writing, software
  11 #   distributed under the License is distributed on an  BASIS,
  12 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 #   See the License for the specific language governing permissions and
  14 #   limitations under the License.
  15 #
  16
  17 from django.shortcuts import render, redirect
  18 from sf_t3d.decorators import login_required
  19 from django.http import HttpResponse
  20 import json
  21 import logging
  22 import authosm.utils as osmutils
  23 from lib.osm.osmclient.clientv2 import Client
  24
  25 logging.basicConfig(level=logging.DEBUG)
  26 log = logging.getLogger(__name__)
  27
  28
  29 @login_required
  30 def user_list(request):
  31     user = osmutils.get_user(request)
  32     client = Client()
  33     result = client.user_list(user.get_token())
  34
  35     result = {
  36         'users': result['data'] if result and result['error'] is False else []
  37     }
  38
  39     return __response_handler(request, result, 'user_list.html')
  40
  41
  42 @login_required
  43 def create(request):
  44     user = osmutils.get_user(request)
  45     client = Client()
  46     user_data = {
  47         "username": request.POST['username'],
  48         "password": request.POST['password'],
  49         "domain_name": request.POST['domain_name']
  50     }
  51
  52     result = client.user_create(user.get_token(), user_data)
  53     if result['error']:
  54         return __response_handler(request, result['data'], url=None,
  55                                   status=result['data']['status'] if 'status' in result['data'] else 500)
  56     else:
  57         return __response_handler(request, {}, url=None, status=200)
  58
  59
  60 @login_required
  61 def delete(request, user_id=None):
  62     user = osmutils.get_user(request)
  63     try:
  64         client = Client()
  65         result = client.user_delete(user.get_token(), user_id)
  66     except Exception as e:
  67         log.exception(e)
  68         result = {'error': True, 'data': str(e)}
  69     if result['error']:
  70         return __response_handler(request, result['data'], url=None,
  71                                   status=result['data']['status'] if 'status' in result['data'] else 500)
  72     else:
  73         return __response_handler(request, {}, url=None, status=200)
  74
  75
  76 @login_required
  77 def user_info(request, user_id=None):
  78     user = osmutils.get_user(request)
  79     try:
  80         client = Client()
  81
  82         info_res = client.get_user_info(user.get_token(), user_id)
  83     except Exception as e:
  84         log.exception(e)
  85         info_res = {'error': True, 'data': str(e)}
  86     if info_res['error']:
  87         return __response_handler(request, info_res['data'], url=None,
  88                                   status=info_res['data']['status'] if 'status' in info_res['data'] else 500)
  89     else:
  90         return __response_handler(request, info_res['data'], url=None, status=200)
  91
  92
  93 @login_required
  94 def update(request, user_id=None):
  95     user = osmutils.get_user(request)
  96     try:
  97         client = Client()
  98         payload = {}
  99
 100         if request.POST.get('password') and request.POST.get('password') is not '':
 101             payload["password"] = request.POST.get('password')
 102
 103         if request.POST.getlist('map_project_name') and request.POST.getlist('map_role_name'):
 104             project_param_name = request.POST.getlist('map_project_name')
 105             role_param_ip = request.POST.getlist('map_role_name')
 106             payload["project_role_mappings"] = []
 107             for i, project in enumerate(project_param_name):
 108                 payload["project_role_mappings"].append({
 109                     'project': project,
 110                     'role': role_param_ip[i],
 111                 })
 112
 113         update_res = client.user_update(user.get_token(), user_id, payload)
 114     except Exception as e:
 115         log.exception(e)
 116         update_res = {'error': True, 'data': str(e)}
 117     if update_res['error']:
 118         return __response_handler(request, update_res['data'], url=None,
 119                                   status=update_res['data']['status'] if 'status' in update_res['data'] else 500)
 120     else:
 121         return __response_handler(request, {}, url=None, status=200)
 122
 123
 124 def __response_handler(request, data_res, url=None, to_redirect=None, *args, **kwargs):
 125     raw_content_types = request.META.get('HTTP_ACCEPT', '*/*').split(',')
 126     if 'application/json' in raw_content_types or url is None:
 127         return HttpResponse(json.dumps(data_res), content_type="application/json", *args, **kwargs)
 128     elif to_redirect:
 129         return redirect(url, *args, **kwargs)
 130     else:
 131         return render(request, url, data_res)