diff --git a/main.py b/main.py index d7aaac9..1c1245e 100644 --- a/main.py +++ b/main.py @@ -16,8 +16,12 @@ from flask import Flask, request RULE_PREFIX = "[auto]::" -def create_rule(label, address, port): - return asusrouter.modules.port_forwarding.PortForwardingRule(RULE_PREFIX + label, address, None, "BOTH", None, port) +def create_rule(label, address, ports): + + if isinstance(ports, list): + ports = ",".join(ports) + + return asusrouter.modules.port_forwarding.PortForwardingRule(RULE_PREFIX + label, address, None, "BOTH", None, ports) def rule_to_string(rule: asusrouter.modules.port_forwarding.PortForwardingRule):