[fix] Do not fail if protocol is not specified

This commit is contained in:
Kload 2014-06-09 14:27:58 +02:00
parent cddae8b54a
commit cdc777671d

View file

@ -70,8 +70,11 @@ def firewall_allow(port=None, protocol=['TCP'], ipv6=False, no_upnp=False):
"""
port = int(port)
ipv = "ipv4"
if isinstance(protocol, list):
protocols = protocol
protocol = protocol[0]
else:
protocols = [protocol]
protocol = protocols[0]
firewall = firewall_list(raw=True)
@ -109,8 +112,11 @@ def firewall_disallow(port=None, protocol=['TCP'], ipv6=False):
"""
port = int(port)
ipv = "ipv4"
if isinstance(protocol, list):
protocols = protocol
protocol = protocol[0]
else:
protocols = [protocol]
protocol = protocols[0]
firewall = firewall_list(raw=True)