import latitudesh_python_sdk
from latitudesh_python_sdk import Latitudesh
import os


with Latitudesh(
    bearer=os.getenv("LATITUDESH_BEARER", ""),
) as latitudesh:

    res = latitudesh.firewalls.assign(firewall_id="fw_Ee8pKq05DWAob", data={
        "type": latitudesh_python_sdk.CreateFirewallAssignmentFirewallsType.FIREWALL_ASSIGNMENTS,
        "attributes": {
            "server_id": "sv_aKXgRdR3qv9k5",
        },
    })

    # Handle response
    print(res)

{
  "data": {
    "id": "fwasg_6VE1Wd37dXnZJ",
    "type": "firewall_assignments",
    "attributes": {
      "server": {
        "id": "sv_aKXgRdR3qv9k5",
        "hostname": "Ergonomic Wooden Keyboard",
        "primary_ipv4": "132.192.74.198"
      },
      "firewall_id": "fw_Ee8pKq05DWAob"
    }
  }
}