from pymavlink import mavutil
import time

# Create a connection to the MAVLink device
# For USB, use '/dev/ttyUSB0' (Linux) or 'comX' (Windows).
connection = mavutil.mavlink_connection('/dev/ttyUSB0', baud=115200)

# Function to send a heartbeat
def send_heartbeat():
    # Create the heartbeat message
    connection.mav.heartbeat_send(
        type=mavutil.mavlink.MAV_TYPE_GCS,         # MAV_TYPE_GCS for Ground Control Station
        autopilot=mavutil.mavlink.MAV_AUTOPILOT_INVALID,
        base_mode=0,
        custom_mode=0,
        system_status=mavutil.mavlink.MAV_STATE_STANDBY
    )

# Send a heartbeat every second
while True:
    send_heartbeat()
    print("Heartbeat sent")
    time.sleep(1)

from pymavlink import mavutil

# Create a connection to the MAVLink device
connection = mavutil.mavlink_connection('/dev/ttyUSB0', baud=115200)

# Wait for a heartbeat from the MAVLink device
print("Waiting for heartbeat...")
heartbeat = connection.wait_heartbeat(timeout=5)

if heartbeat:
    print(f"Heartbeat received from system {heartbeat.get_srcSystem()}, component {heartbeat.get_srcComponent()}")
else:
    print("No heartbeat received within the timeout period.")