cmk_base/donate.py

   1 #!/usr/bin/env python
   2 # -*- encoding: utf-8; py-indent-offset: 4 -*-
   3 # +------------------------------------------------------------------+
   4 # |             ____ _               _        __  __ _  __           |
   5 # |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
   6 # |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
   7 # |           | |___| | | |  __/ (__|   <    | |  | | . \            |
   8 # |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
   9 # |                                                                  |
  10 # | Copyright Mathias Kettner 2014             mk@mathias-kettner.de |
  11 # +------------------------------------------------------------------+
  12 #
  13 # This file is part of Check_MK.
  14 # The official homepage is at http://mathias-kettner.de/check_mk.
  15 #
  16 # check_mk is free software;  you can redistribute it and/or modify it
  17 # under the  terms of the  GNU General Public License  as published by
  18 # the Free Software Foundation in version 2.  check_mk is  distributed
  19 # in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
  20 # out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
  21 # PARTICULAR PURPOSE. See the  GNU General Public License for more de-
  22 # tails. You should have  received  a copy of the  GNU  General Public
  23 # License along with GNU Make; see the file  COPYING.  If  not,  write
  24 # to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
  25 # Boston, MA 02110-1301 USA.
  26
  27 import base64
  28 import os
  29 import sys
  30
  31 import cmk.utils.paths
  32
  33 import cmk_base.config as config
  34 import cmk_base.console as console
  35
  36
  37 def do_donation():
  38     donate = []
  39     cache_files = os.listdir(cmk.utils.paths.tcp_cache_dir)
  40     for host in config.all_active_realhosts():
  41         if config.in_binary_hostlist(host, config.donation_hosts):
  42             for f in cache_files:
  43                 if f == host or f.startswith("%s." % host):
  44                     donate.append(f)
  45
  46     if not donate:
  47         console.error("No hosts specified. You need to set donation_hosts in main.mk.\n")
  48         sys.exit(1)
  49
  50     console.verbose("Donating files %s\n" % " ".join(cache_files))
  51     indata = base64.b64encode(
  52         os.popen(  # nosec
  53             "tar czf - -C %s %s" % (cmk.utils.paths.tcp_cache_dir, " ".join(donate))).read())
  54
  55     output = os.popen(config.donation_command, "w")  # nosec
  56     output.write("\n\n@STARTDATA\n")
  57     while len(indata) > 0:
  58         line = indata[:64]
  59         output.write(line)
  60         output.write('\n')
  61         indata = indata[64:]