96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
# Copyright 2015-2017 Espressif Systems (Shanghai) PTE LTD
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http:#www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
"""
|
|
Internal use only.
|
|
|
|
This file implements controlling APC PDU via telnet.
|
|
"""
|
|
|
|
import telnetlib
|
|
|
|
|
|
class Control(object):
|
|
""" control APC via telnet """
|
|
|
|
@classmethod
|
|
def apc_telnet_make_choice(cls, telnet, choice):
|
|
""" select a choice """
|
|
telnet.read_until(b"Event Log")
|
|
telnet.read_until(b">")
|
|
telnet.write(choice.encode() + b"\r\n")
|
|
|
|
@classmethod
|
|
def apc_telnet_common_action(cls, telnet, check_str, action):
|
|
""" wait until a pattern and then write a line """
|
|
telnet.read_until(check_str.encode())
|
|
telnet.write(action.encode() + b"\r\n")
|
|
|
|
@classmethod
|
|
def control(cls, apc_ip, control_dict):
|
|
"""
|
|
control APC
|
|
|
|
:param apc_ip: IP of APC
|
|
:param control_dict: dict with outlet ID and "ON" or "OFF"
|
|
"""
|
|
|
|
for _outlet in control_dict:
|
|
assert 0 < _outlet < 9
|
|
assert control_dict[_outlet] in ["ON", "OFF"]
|
|
|
|
# telnet
|
|
# set timeout as 2s so that it won't waste time even can't access APC
|
|
tn = telnetlib.Telnet(host=apc_ip, timeout=5)
|
|
# log on
|
|
cls.apc_telnet_common_action(tn, "User Name :", "apc")
|
|
cls.apc_telnet_common_action(tn, "Password :", "apc")
|
|
# go to Device Manager
|
|
cls.apc_telnet_make_choice(tn, "1")
|
|
# go to Outlet Management
|
|
cls.apc_telnet_make_choice(tn, "2")
|
|
# go to Outlet Control/Configuration
|
|
cls.apc_telnet_make_choice(tn, "1")
|
|
|
|
# do select Outlet and control
|
|
for _outlet in control_dict:
|
|
# choose Outlet
|
|
cls.apc_telnet_make_choice(tn, str(_outlet))
|
|
# choose Control Outlet
|
|
cls.apc_telnet_make_choice(tn, "1")
|
|
# choose action
|
|
_action = control_dict[_outlet]
|
|
if "ON" in _action:
|
|
cls.apc_telnet_make_choice(tn, "1")
|
|
else:
|
|
cls.apc_telnet_make_choice(tn, "2")
|
|
# do confirm
|
|
cls.apc_telnet_common_action(tn, "cancel :", "YES")
|
|
cls.apc_telnet_common_action(tn, "continue...", "")
|
|
# return to Outlet Control/Configuration
|
|
cls.apc_telnet_make_choice(tn, "\033")
|
|
cls.apc_telnet_make_choice(tn, "\033")
|
|
|
|
# exit to main menu and logout
|
|
tn.write(b"\033\r\n")
|
|
tn.write(b"\033\r\n")
|
|
tn.write(b"\033\r\n")
|
|
tn.write(b"4\r\n")
|
|
|
|
@classmethod
|
|
def control_rest(cls, apc_ip, outlet, action):
|
|
outlet_list = list(range(1, 9)) # has to be a list if we want to remove from it under Python 3
|
|
outlet_list.remove(outlet)
|
|
cls.control(apc_ip, dict.fromkeys(outlet_list, action))
|