CompanionScripts/ups_ctl.py3 at master · ZachIndigo/CompanionScripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#!/usr/bin/env python3

from time import sleep
from telnetlib import Telnet as tn
import argparse

parser = argparse.ArgumentParser(
    description='Control TrippLite PowerAlert UPSes')
parser.add_argument('-a', '--all-ports', type=str, default=None,
                    help='Power state for all outlets (overwritten by -1 or \
                            -2). (*None)')
parser.add_argument('-i', '--ip', type=str, default=None,
                    help='IP address of the UPS. (*None)')
parser.add_argument('-p', '--passwd', type=str, default=None,
                    help='PADM password for localadmin. (*None)')
parser.add_argument('-u', '--ups-version', type=int, default=15,
                    choices=[0, 12, 15],
                    help='UPS PADM version to use (0 for JuiceGoose) \
                            (0,12,*15)')
parser.add_argument('-1', '--port-1', type=str, default=None,
                    help='Power state for the first outlet. (*None)')
parser.add_argument('-2', '--port-2', type=str, default=None,
                    help='Power state for the second outlet. (*None)')
parser.add_argument('-3', '--port-3', type=str, default=None,
                    help='Power state for the third outlet (only applicable \
                            for JuiceGeese). (*None)')
parser.add_argument('-q', '--quiet', action='store_true',
                    help='Suppress standard output.')
parser.add_argument('-v', '--verbose', action='store_true',
                    help='Prints telnet output.')
args = parser.parse_args()

if (args.passwd is None and args.ups_version != 0) or args.ip is None:
    print("ups_ctl needs a password (if not a JuiceGoose) and ip address!")
    quit()

timeout = 5

act1 = args.all_ports
act2 = args.all_ports
if args.ups_version == 0:
    act3 = args.all_ports
if args.port_1:
    act1 = args.port_1
if args.port_2:
    act2 = args.port_2
if args.ups_version == 0 and args.port_3:
    act3 = args.port_3
if act1 is None and act2 is None and act3 is None:
    print("Not doing anything, because no action provided.")


log = ""

ups = tn(args.ip, 5214, timeout)
if not args.quiet:
    print("Connecting...")
if not args.ups_version == 0:
    log = ups.read_until(b"login: ", 15)
    if args.verbose:
        print(repr(log))
    if not log:
        print("Connection failed somehow.")
        quit()
    ups.write(b"localadmin\r\n")
    log = ups.read_until(b"Password: ", timeout)
    if args.verbose:
        print(repr(log))
    ups.write((args.passwd + "\r\n").encode("ascii"))
    if args.ups_version == 15:
        log = ups.read_until(b">> ", timeout)
        if args.verbose:
            print(repr(log))
        ups.write(b"e\r\n")
    log = ups.read_until(b"$> ", timeout)
    if args.verbose:
        print(repr(log))
else:
    sleep(timeout)

if act1 == "on" or act1 == "off":
    if not args.quiet:
        print("Turning load 1 " + act1 + ".")
    if not args.ups_version == 0:
        ups.write(("loadctl {} -o 1 --force\r\n".format(act1)).encode("ascii"))
        log = ups.read_until(b'$> ', timeout)
    else:
        ups.write(("POD1{}\r\n".format(act1)).encode("ascii"))
        log = ups.read_until(b'->', timeout)
    if args.verbose:
        print(repr(log))
if act2 == "on" or act2 == "off":
    if not args.quiet:
        print("Turning load 2 " + act2 + ".")
    if not args.ups_version == 0:
        ups.write(("loadctl {} -o 2 --force\r\n".format(act2)).encode("ascii"))
        log = ups.read_until(b'$> ', timeout)
    else:
        ups.write(("POD2{}\r\n".format(act2)).encode("ascii"))
        log = ups.read_until(b'->', timeout)
    if args.verbose:
        print(repr(log))
if args.ups_version == 0 and act3 == "on":
    if not args.quiet:
        print("Turning load 3 " + act1 + ".")
    ups.write(("POD3{}\r\n".format(act2)).encode("ascii"))
    log = ups.read_until(b'->', timeout)
    if args.verbose:
        print(repr(log))

if not args.quiet:
    print("Exiting...")
if not args.ups_version == 0:
    ups.write(b"exit\r\n")
    if args.ups_version == 15:
        log = ups.read_until(b"Farewell, localadmin.\r\r\n")
        if args.verbose:
            print(repr(log))
    elif args.ups_version == 12:
        log = ups.read_until(b"Goodbye.\r\n")
        if args.verbose:
            print(repr(log))
ups.close()