mirror of
https://github.com/Telecominfraproject/wlan-testing.git
synced 2025-10-31 10:57:50 +00:00
96 lines
3.8 KiB
Python
96 lines
3.8 KiB
Python
import os
|
|
import pprint
|
|
import time
|
|
import argparse
|
|
from typing import Sequence
|
|
from typing import Optional
|
|
|
|
try:
|
|
import dlipower
|
|
except:
|
|
print('Please wait we are installing DLI Power')
|
|
os.system('pip install dlipower')
|
|
|
|
|
|
class setup:
|
|
def __init__(self, hostname, user, password):
|
|
self.hostname = hostname
|
|
self.user = user
|
|
self.password = password
|
|
self.power_switch = dlipower.PowerSwitch(hostname=self.hostname, userid=self.user, password=self.password)
|
|
|
|
|
|
class switch_on(setup):
|
|
def __init__(self, hostname, user, password, port=None):
|
|
super().__init__(hostname, user, password)
|
|
self.port = port
|
|
if self.port != 'all':
|
|
self.i = 0
|
|
for i in self.power_switch:
|
|
# print(i.description)
|
|
if i.description == self.port:
|
|
self.power_switch[self.i].state = "ON"
|
|
self.i += 1
|
|
else:
|
|
for outlet in self.power_switch:
|
|
outlet.state = 'ON'
|
|
|
|
|
|
class switch_off(setup):
|
|
def __init__(self, hostname, user, password, port=None):
|
|
super().__init__(hostname, user, password)
|
|
self.port = port
|
|
if self.port != 'all':
|
|
self.i = 0
|
|
for i in self.power_switch:
|
|
# print(i.description)
|
|
if i.description == self.port:
|
|
self.power_switch[self.i].state = "OFF"
|
|
self.i += 1
|
|
# self.power_switch[int(self.port)-1].state = "OFF"
|
|
else:
|
|
for outlet in self.power_switch:
|
|
outlet.state = 'OFF'
|
|
|
|
class print_status(setup):
|
|
def __init__(self, hostname, user, password):
|
|
super().__init__(hostname, user, password)
|
|
print(self.power_switch)
|
|
|
|
def main(argv: Optional[Sequence[str]]=None):
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--host', help='Please provide host name eg: 192.168.200.65')
|
|
parser.add_argument('--username', help='Please provide username eg: admin')
|
|
parser.add_argument('--password', help='Please provide password eg: 1234')
|
|
parser.add_argument('--action', help='Switches all Outlets in ON Mode eg: --on_all\n'
|
|
'Switches all Outlets in OFF Mode eg: --off_all\n'
|
|
'Cycle all Outlets in OFF Mode and then ON Mode eg: --cycle_all\n'
|
|
'Switches the target Outlets in ON Mode eg: --on\n'
|
|
'Switches the target Outlets in OFF Mode --off\n')
|
|
parser.add_argument('--port', help='Please provide port name eg: --port lanforge')
|
|
args = parser.parse_args(argv)
|
|
dic = vars(args)
|
|
# print(dic)
|
|
if dic['action'] == 'on':
|
|
set = setup(dic['host'], dic['username'], dic['password'])
|
|
on = switch_on(dic['host'], dic['username'], dic['password'], dic['port'])
|
|
elif dic['action'] == 'off':
|
|
set = setup(dic['host'], dic['username'], dic['password'])
|
|
off = switch_off(dic['host'], dic['username'], dic['password'], dic['port'])
|
|
# off = switch_on(dic['action'])
|
|
elif dic['action'] == 'cycle':
|
|
set = setup(dic['host'], dic['username'], dic['password'])
|
|
on = switch_off(dic['host'], dic['username'], dic['password'], dic['port'])
|
|
off = switch_on(dic['host'], dic['username'], dic['password'], dic['port'])
|
|
else:
|
|
print('Command not found')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|
|
# Command line to be used as
|
|
# python pdu_automation.py --host 192.168.200.90 --user admin --password 1234 --action on --port all
|
|
# python pdu_automation.py --host 192.168.200.90 --user admin --password 1234 --action off --port 'Outlet 1'
|
|
# python pdu_automation.py --host 192.168.200.90 --user admin --password 1234 --action cycle --port all
|