Test Logstash connection in python

check_logstash.py

#!/usr/bin/env python

import json
import socket
import sys
import datetime

def send_data_to_logstash(ls_data, host, port):
  try:
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    data_json = json.dumps(ls_data)
    sock.sendto(data_json.encode('utf-8'), (host, port))
    print("Data sent successfully")
  except Exception as e:
    print(f"Error sending data: {e}")

if __name__ == "__main__":
  if len(sys.argv) < 3:
    print("Usage: check-logstash.py <host> <port>")
    sys.exit(1)
  logstash_host = sys.argv[1]
  logstash_port = int(sys.argv[2])
  data = {
      "message": "Hello from Python",
      "timestamp": str(datetime.datetime.now())
  }
  send_data_to_logstash(data, logstash_host, logstash_port)