Untitled

#!/usr/bin/env python3
import psutil
import json
import requests

INPUT_JSON = input()
SHUFFLE_WEBHOOK_URL = "http://192.168.0.108:3001/api/v1/hooks/webhook_2030196e-c52a-45f4-813d-977f47696789"

def get_memory_info():
    return {
        "total_memory": psutil.virtual_memory().total / (1024.0 ** 3),
        "available_memory": psutil.virtual_memory().available / (1024.0 ** 3),
        "used_memory": psutil.virtual_memory().used / (1024.0 ** 3),
        "memory_percentage": psutil.virtual_memory().percent
    }

def get_cpu_info():
    return {
        "physical_cores": psutil.cpu_count(logical=False),
        "total_cores": psutil.cpu_count(logical=True),
        "processor_speed": psutil.cpu_freq().current,
        "cpu_usage_per_core": dict(enumerate(psutil.cpu_percent(percpu=True, interval=1))),
        "total_cpu_usage": psutil.cpu_percent(interval=1)
    }

def get_disk_info():
    partition_usage = psutil.disk_usage('/')
    return {
       "total_space": partition_usage.total / (1024.0 ** 3),
       "used_space": partition_usage.used / (1024.0 ** 3),
       "free_space": partition_usage.free / (1024.0 ** 3),
       "usage_percentage": partition_usage.percent
    }

def get_alert_info():
    disk_data = json.loads(INPUT_JSON)
    parameters = disk_data.get("parameters").get("extra_args")
    return {
        "text": parameters[0],
        "timestamp": parameters[1],
	"srcip": parameters[2],
	"dstip": parameters[3],
	"agent_id": parameters[4],
	"agent_name": parameters[5],
	"agent_ip": parameters[6],
	"api_key": parameters[7]
    }

def send_to_shuffle(data):
  headers = {'Content-Type': 'application/json'}
  payload = {'system': json.dumps(data, indent=4)}

  response = requests.post(SHUFFLE_WEBHOOK_URL, headers=headers, json=payload)

if __name__ == '__main__':
  data = {
    "cpu_info": get_cpu_info(),
    "memory_info": get_memory_info(),
    "disk_info": get_disk_info(),
    "alert_info": get_alert_info()
  }
  send_to_shuffle(data)