###########################################################################
# CLOUD SERVER API DOCUMENTATION
#
# Name: addFirewallServer
# Language: Python
#
# Description: adds firewall policy to an IP
###########################################################################
from urllib.request import Request, urlopen
import urllib
import json

#ID access to API
TOKEN = "{YOUR_API_TOKEN}" # e.g.: "f03c3c76cc853ee690b879909c9c6f2a"
url = "https://cloudpanel-api.1and1.com/v1"

def _addFirewallServer(id, ip_id, content):
    #Configure the request
    _command = url + "/servers/" + id + "/ips/" + ip_id + "/firewall_policy"
    _method = 'PUT'
    request = Request(_command, data=content.encode(encoding='utf_8'), 
                      headers={'X-TOKEN':TOKEN, 'content-type':'application/json'}, 
                      method=_method)
    
    #Try to get the response
    try:
        response = urlopen(request)
        content = response.read()
        return (content.decode())
    #Fetch error
    except urllib.error.URLError as e:
        return("Error " + str(e.code) + ":" + e.reason) 

#PARAMETERS
id = "{YOUR_SERVER_ID}" # e.g.: "5340033E7FBBC308BC329414A0DF3C20"
ip_id = "{YOUR_IP_ID}" # e.g.: "5340033E7FBBC308BC329414A0DF3C20"
firewall_policy_id = "{YOUR_FIREWALL_ID}" # e.g.: "5340033E7FBBC308BC329414A0DF3C20"

data = json.dumps({'id':firewall_policy_id})

#Add firewall policy
print(_addFirewallServer(id, ip_id, data))