Add options to make WAN port Internet State enable or disable

internet-status.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python3
+
+import sys
+from omada import Omada
+
+def main():
+	if len(sys.argv) < 2 or (len(sys.argv) > 1 and sys.argv[1] not in ('enable','disable')):
+		print( f"usage: {sys.argv[0]} [enable|disable]" )
+		return
+
+	omada = Omada()
+	omada.login()
+
+	devices = omada.getSiteDevices()
+	if len(devices) > 0:
+		gateway = next((device for device in devices if device['type'] == 'gateway'), None)
+		print(f"Selected gateway: {gateway}")
+		if gateway:
+			mac = gateway['mac']
+		else:
+			print( 'No gateway device found' )
+			return
+
+		omada.setInternetStatus(mac=mac, port_id=1, enable=sys.argv[1] == 'enable')
+		gateway = omada.getGatewayDevice(mac=mac)
+		print( f'Updated gateway: {gateway}' )
+
+	omada.logout()
+
+if __name__ == '__main__':
+	main()

omada/omada.py

@@ -590,3 +590,20 @@ def getWirelessGroups(self, site=None):
 	##
 	def getWirelessNetworks(self, group, site=None):
 		return self.__get( f'/sites/{self.__findKey(site)}/setting/wlans/{group}/ssids' )
+
+	##
+	## Returns the gateway device specified by the MAC address.
+	##
+	## This is a Gateway device listed on Device List.
+	##
+	def getGatewayDevice(self, mac, site=None):
+		return self.__get( f'/sites/{self.__findKey(site)}/gateways/{mac}' )
+
+	##
+	## Control the internet status of a WAN port in a gateway device specified by the MAC address.
+	##
+	## enable = True, Make WAN port connected.
+	## enable = False, Make WAN port disconnected.
+	##
+	def setInternetStatus(self, mac, port_id, enable, site=None):
+		return self.__post( f'/sites/{self.__findKey(site)}/cmd/gateways/{mac}/internetState', json={'portId': port_id, 'operation': 1 if enable else 0} )