k8sclusterhandler/views.py

   1 #
   2 #   Copyright 2019 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 yaml
  21 import json
  22 import logging
  23 import authosm.utils as osmutils
  24 from lib.osm.osmclient.clientv2 import Client
  25
  26 logging.basicConfig(level=logging.DEBUG)
  27 log = logging.getLogger('k8sclusterhandler/view.py')
  28
  29
  30 @login_required
  31 def list(request):
  32     user = osmutils.get_user(request)
  33     project_id = user.project_id
  34     result = {'project_id': project_id}
  35     raw_content_types = request.META.get('HTTP_ACCEPT', '*/*').split(',')
  36     if 'application/json' not in raw_content_types:
  37         return __response_handler(request, result, 'k8scluster_list.html')
  38     client = Client()
  39     result_client = client.k8sc_list(user.get_token())
  40
  41     result['k8sc'] = result_client['data'] if result_client and result_client['error'] is False else []
  42
  43     return __response_handler(request, result, 'k8scluster_list.html')
  44
  45
  46 @login_required
  47 def create(request):
  48     user = osmutils.get_user(request)
  49     project_id = user.project_id
  50     result = {'project_id': project_id}
  51     client = Client()
  52     try:
  53
  54         new_k8sc = {
  55             "name": request.POST.get('name'),
  56             "credentials": yaml.load(request.POST.get('credentials')),
  57             "vim_account": request.POST.get('vim_account'),
  58             "description": request.POST.get('description'),
  59             "k8s_version": request.POST.get('k8s_version'),
  60             "nets": yaml.load(request.POST.get('nets')),
  61         }
  62
  63         # new_k8sc['vim_account'] = get_vim_account_id(new_k8sc['vim_account'])
  64
  65     except Exception as e:
  66         return __response_handler(request, {'status': 400, 'code': 'BAD_REQUEST', 'detail': e.message}, url=None, status=400)
  67     result = client.k8sc_create(user.get_token(), new_k8sc)
  68     if result['error']:
  69         return __response_handler(request, result['data'], url=None, status=result['data']['status'] if 'status' in result['data'] else 500)
  70     else:
  71         return __response_handler(request, result, 'k8sc:list', to_redirect=True)
  72
  73
  74 @login_required
  75 def update(request, k8sc_id=None):
  76     user = osmutils.get_user(request)
  77     try:
  78         update_k8sc_dict = request.POST.dict()
  79         client = Client()
  80         res = client.k8sc_update(user.get_token(), k8sc_id, update_k8sc_dict)
  81     except Exception as e:
  82         log.exception(e)
  83     return __response_handler(request, res, 'k8sc:list', to_redirect=True)
  84
  85
  86 @login_required
  87 def show(request, k8sc_id=None):
  88     user = osmutils.get_user(request)
  89     project_id = user.project_id
  90     client = Client()
  91     result_client = client.k8sc_get(user.get_token(), k8sc_id)
  92
  93     return __response_handler(request, result_client)
  94
  95
  96 @login_required
  97 def delete(request, k8sc_id=None):
  98     user = osmutils.get_user(request)
  99     try:
 100         client = Client()
 101         del_res = client.k8sc_delete(user.get_token(), k8sc_id)
 102     except Exception as e:
 103         log.exception(e)
 104     return __response_handler(request, del_res, 'k8sc:list', to_redirect=True)
 105
 106
 107 def __response_handler(request, data_res, url=None, to_redirect=None, *args, **kwargs):
 108     raw_content_types = request.META.get('HTTP_ACCEPT', '*/*').split(',')
 109     if 'application/json' in raw_content_types or url is None:
 110         return HttpResponse(json.dumps(data_res), content_type="application/json", *args, **kwargs)
 111     elif to_redirect:
 112         return redirect(url, *args, **kwargs)
 113     else:
 114         return render(request, url, data_res)