TOSCA/kubernetes-cluster-TOSCA/scripts/tasks.py

   1 #!/usr/bin/env python
   2
   3 # ============LICENSE_START==========================================
   4 # ===================================================================
   5 # Copyright © 2017 AT&T
   6 #
   7 # Licensed under the Apache License, Version 2.0 (the "License");
   8 # you may not use this file except in compliance with the License.
   9 # You may obtain a copy of the License at
  10 #
  11 #         http://www.apache.org/licenses/LICENSE-2.0
  12 #
  13 # Unless required by applicable law or agreed to in writing, software
  14 # distributed under the License is distributed on an "AS IS" BASIS,
  15 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 # See the License for the specific language governing permissions and
  17 # limitations under the License.
  18 #============LICENSE_END============================================
  19
  20 # here we define some tasks
  21
  22 from fabric.api import run
  23
  24
  25 def label_node(labels, hostname):
  26     if labels:
  27         label_list = []
  28         for key, value in labels.items():
  29             label_pair_string = '%s=%s' % (key, value)
  30             label_list.append(label_pair_string)
  31         label_string = ' '.join(label_list)
  32         command = 'kubectl label nodes %s %s' % (hostname, label_string)
  33         run(command)
  34
  35
  36 def stop_node(hostname):
  37     command = 'kubectl drain %s' % (hostname)
  38     run(command)
  39
  40
  41 def delete_node(hostname):
  42     command = 'kubectl delete no %s' % (hostname)
  43     run(command)