hectoragent.py

#!/usr/bin/env python

'''
 * Author : Hutter Valentin
 * Date : 08.05.2019
 * Description : Hector agent monitoring
 * Help :
 *  - Requests: https://2.python-requests.org/en/master/
'''

import sys
import pyfiglet
import os
import platform
import configparser
import requests
import json
import datetime
import helpers
import psutil
# Import modules
from modules.colors import colors
from modules.memory import memory
from modules.swap import swap
from modules.disk import disk
from modules.network import network
from modules.load import load
from modules.process import process
from modules.ping import ping
from modules.cpu import cpu
from modules.ip import ip

AGENT_VERSION = '1.0.0'
API_ENDPOINT = 'https://api.hector.valentinhutter.ch'

'''
Class who manages the agent
'''
class HectorAgent:

  def __init__(self):
    print(colors.ORANGE + "Starting hector agent..." + colors.NORMAL)

  def _server_platform(self):
    return platform.platform()

  # Collect and sendit to API
  def collect_data(self):
    if os.path.exists('/opt/hector-agent/hectoragent.ini'):
      config = configparser.ConfigParser()
      config.read('/opt/hector-agent/hectoragent.ini') # Loading configuration
      token = config['API']['token'].strip('"') # Get token from .ini configuration
      is_agent_installed = config['GENERAL'].getboolean('installed')
      
      if is_agent_installed:
        # Collecting data
        print('\nCollecting Data...')
        mem = memory().collect()
        swap_mem = swap().collect()
        disk_usage_overall = disk().collect_overall_disk_usage()
        disk_partitions = disk().collect_partitions()
        disk_io = disk().collect_io()
        net = network().collect_network_addrs()
        server_loads = load().collect()
        processes = process().collect()
        extern_ping = ping().collect()
        server_cpu = cpu().collect()
        boot_time = float(psutil.boot_time())
        servers_ips = ip().collect_ips()
        
        # Sending data to the API
        try:
          print('Sending data to API...')
          res = requests.post(API_ENDPOINT + "/servers/sendmonitoring?server_token=" + token, data={
            'os_fullname': self._server_platform(),
            'boot_time': boot_time,
            'memory': helpers.dict_to_base64(mem),
            'swap': helpers.dict_to_base64(swap_mem),
            'disk_usage_overall': helpers.dict_to_base64(disk_usage_overall),
            'disk_partitions': helpers.dict_to_base64(disk_partitions),
            'disk_io': helpers.dict_to_base64(disk_io),
            'network': helpers.dict_to_base64(net),
            'loads': helpers.dict_to_base64(server_loads),
            'processes': helpers.dict_to_base64(processes),
            'extern_ping': helpers.dict_to_base64(extern_ping),
            'cpu': helpers.dict_to_base64(server_cpu),
            'ips': helpers.dict_to_base64(servers_ips),
            'request_send_at': datetime.datetime.now(),
          })
          
          if res.status_code == 200 and json.loads(res.text)['success']:
            print(colors.GREEN + '\nThe data has been correctly sent to the API!' + colors.NORMAL)

        except requests.exceptions.RequestException:
          sys.exit(1)
      else:
        print(colors.RED + "The agent does not seem to be correctly installed, please restart the installation script!" + colors.NORMAL)
    
  # Start the agent
  def start(self):
    print(colors.GREEN + 'Hector agent has started!' + colors.NORMAL)
    self.collect_data()

'''
Print ascii logo of hector
'''
def getLogo():
  ascii_logo = pyfiglet.figlet_format("Hector Agent")
  print(colors.CBLUE + ascii_logo + colors.NORMAL)
  print('')

'''
Entry point of agent
'''
def main():
  getLogo()

  if len(sys.argv) > 1:
    # Help argument
    if sys.argv[1] == 'help':
      print ('\n'.join((
          'To start the agent, run it without specifying an argument.',
          'Available optional arguments (hectoragent.py <argument>):',
          '  help',
          '  version',
      )))
      sys.exit()
    if sys.argv[1] == 'version':
      print('Agent run on version: ' + AGENT_VERSION)
      sys.exit()
  else:
    HectorAgent().start()

if __name__ == '__main__':
    main()