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
  47     new_user_dict = request.POST.dict()
  48     keys = ["username", "password", "domain_name"]
  49     user_data = dict(filter(lambda i: i[0] in keys and len(i[1]) > 0, new_user_dict.items()))
  50     result = client.user_create(user.get_token(), user_data)
  51     if result['error']:
  52         return __response_handler(request, result['data'], url=None,
  53                                   status=result['data']['status'] if 'status' in result['data'] else 500)
  54     else:
  55         return __response_handler(request, {}, url=None, status=200)
  56
  57
  58 @login_required
  59 def delete(request, user_id=None):
  60     user = osmutils.get_user(request)
  61     try:
  62         client = Client()
  63         result = client.user_delete(user.get_token(), user_id)
  64     except Exception as e:
  65         log.exception(e)
  66         result = {'error': True, 'data': str(e)}
  67     if result['error']:
  68         return __response_handler(request, result['data'], url=None,
  69                                   status=result['data']['status'] if 'status' in result['data'] else 500)
  70     else:
  71         return __response_handler(request, {}, url=None, status=200)
  72
  73
  74 @login_required
  75 def user_info(request, user_id=None):
  76     user = osmutils.get_user(request)
  77     try:
  78         client = Client()
  79
  80         info_res = client.get_user_info(user.get_token(), user_id)
  81     except Exception as e:
  82         log.exception(e)
  83         info_res = {'error': True, 'data': str(e)}
  84     if info_res['error']:
  85         return __response_handler(request, info_res['data'], url=None,
  86                                   status=info_res['data']['status'] if 'status' in info_res['data'] else 500)
  87     else:
  88         return __response_handler(request, info_res['data'], url=None, status=200)
  89
  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)