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             "k8s_version": request.POST.get('k8s_version'),
  59             "nets": yaml.load(request.POST.get('nets')),
  60         }
  61
  62         # new_k8sc['vim_account'] = get_vim_account_id(new_k8sc['vim_account'])
  63
  64     except Exception as e:
  65         return __response_handler(request, {'status': 400, 'code': 'BAD_REQUEST', 'detail': e.message}, url=None, status=400)
  66     result = client.k8sc_create(user.get_token(), new_k8sc)
  67     if result['error']:
  68         return __response_handler(request, result['data'], url=None, status=result['data']['status'] if 'status' in result['data'] else 500)
  69     else:
  70         return __response_handler(request, result, 'k8sc:list', to_redirect=True)
  71
  72
  73 @login_required
  74 def update(request, k8sc_id=None):
  75     user = osmutils.get_user(request)
  76     try:
  77         update_k8sc_dict = request.POST.dict()
  78         client = Client()
  79         res = client.k8sc_update(user.get_token(), k8sc_id, update_k8sc_dict)
  80     except Exception as e:
  81         log.exception(e)
  82     return __response_handler(request, res, 'k8sc:list', to_redirect=True)
  83
  84
  85 @login_required
  86 def show(request, k8sc_id=None):
  87     user = osmutils.get_user(request)
  88     project_id = user.project_id
  89     client = Client()
  90     result_client = client.k8sc_get(user.get_token(), k8sc_id)
  91
  92     return __response_handler(request, result_client)
  93
  94
  95 @login_required
  96 def delete(request, k8sc_id=None):
  97     user = osmutils.get_user(request)
  98     try:
  99         client = Client()
 100         del_res = client.k8sc_delete(user.get_token(), k8sc_id)
 101     except Exception as e:
 102         log.exception(e)
 103     return __response_handler(request, del_res, 'k8sc:list', to_redirect=True)
 104
 105
 106 def __response_handler(request, data_res, url=None, to_redirect=None, *args, **kwargs):
 107     raw_content_types = request.META.get('HTTP_ACCEPT', '*/*').split(',')
 108     if 'application/json' in raw_content_types or url is None:
 109         return HttpResponse(json.dumps(data_res), content_type="application/json", *args, **kwargs)
 110     elif to_redirect:
 111         return redirect(url, *args, **kwargs)
 112     else:
 113         return render(request, url, data_res)