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     users = result['data'] if result and result['error'] is False else []
  36     result = {
  37         'users': result['data'] if result and result['error'] is False else []
  38     }
  39
  40     return __response_handler(request, result, 'user_list.html')
  41
  42
  43 @login_required
  44 def create(request):
  45     user = osmutils.get_user(request)
  46     client = Client()
  47     user_data ={
  48         "username": request.POST['username'],
  49         "password": request.POST['password']
  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 @login_required
  76 def user_info(request, user_id=None):
  77     user = osmutils.get_user(request)
  78     try:
  79         client = Client()
  80
  81         info_res = client.get_user_info(user.get_token(), user_id)
  82     except Exception as e:
  83         log.exception(e)
  84         info_res = {'error': True, 'data': str(e)}
  85     if info_res['error']:
  86         return __response_handler(request, info_res['data'], url=None,
  87                                   status=info_res['data']['status'] if 'status' in info_res['data'] else 500)
  88     else:
  89         return __response_handler(request, info_res['data'], url=None, status=200)
  90
  91 @login_required
  92 def update(request, user_id=None):
  93     user = osmutils.get_user(request)
  94     try:
  95         client = Client()
  96         payload = {}
  97
  98         if request.POST.get('password') and request.POST.get('password') is not '':
  99             payload["password"] = request.POST.get('password')
 100
 101         if request.POST.getlist('map_project_name') and request.POST.getlist('map_role_name'):
 102             project_param_name = request.POST.getlist('map_project_name')
 103             role_param_ip = request.POST.getlist('map_role_name')
 104             payload["project_role_mappings"] = []
 105             for i, project in enumerate(project_param_name):
 106                 payload["project_role_mappings"].append({
 107                     'project': project,
 108                     'role': role_param_ip[i],
 109                 })
 110
 111         update_res = client.user_update(user.get_token(), user_id, payload)
 112     except Exception as e:
 113         log.exception(e)
 114         update_res = {'error': True, 'data': str(e)}
 115     if update_res['error']:
 116         return __response_handler(request, update_res['data'], url=None,
 117                                   status=update_res['data']['status'] if 'status' in update_res['data'] else 500)
 118     else:
 119         return __response_handler(request, {}, url=None, status=200)
 120
 121
 122 def __response_handler(request, data_res, url=None, to_redirect=None, *args, **kwargs):
 123     raw_content_types = request.META.get('HTTP_ACCEPT', '*/*').split(',')
 124     if 'application/json' in raw_content_types or url is None:
 125         return HttpResponse(json.dumps(data_res), content_type="application/json", *args, **kwargs)
 126     elif to_redirect:
 127         return redirect(url, *args, **kwargs)
 128     else:
 129         return render(request, url, data_res)