empress.py

#!/usr/bin/env python3
# Power/Reset Control of Systems via GPIO
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

import argparse
import RPi.GPIO as GPIO
from time import sleep

#     Raspberry Pi Pinout
#      3V3 (1) (2) 5V
#    GPIO2 (3) (4) 5V
#    GPIO3 (5) (6) GND
#    GPIO4 (7) (8) GPIO14
#      GND (9) (10) GPIO15
#   GPIO17 (11) (12) GPIO18
#   GPIO27 (13) (14) GND
#   GPIO22 (15) (16) GPIO23
#      3V3 (17) (18) GPIO24
#   GPIO10 (19) (20) GND
#    GPIO9 (21) (22) GPIO25
#   GPIO11 (23) (24) GPIO8
#      GND (25) (26) GPIO7
#    GPIO0 (27) (28) GPIO1
#    GPIO5 (29) (30) GND
#    GPIO6 (31) (32) GPIO12
#   GPIO13 (33) (34) GND
#   GPIO19 (35) (36) GPIO16
#   GPIO26 (37) (38) GPIO20
#      GND (39) (40) GPIO21


# System Targets List
systems = ["miner01", "miner02", "miner03", "miner04"]
# System Power Pin List
powerpin = [17, 23, 6, 19]
# System Reset Pin List
resetpin = [22, 18, 12, 16]
# All the pins that are used
pins = powerpin + resetpin

def press(pin, seconds):
    GPIO.output(pin, GPIO.HIGH)
    sleep(seconds)
    GPIO.output(pin, GPIO.LOW)

# Setup ArgParser
parser = argparse.ArgumentParser(
    prog='empress',
    description='External Management of Power and RESet Switchs')

parser.add_argument(
    '--target', '-t',
    required=True,
    action='store',
    dest="target",
    help='Target System (required)')

group = parser.add_mutually_exclusive_group(required=True)
group.add_argument(
    '--poweron', '-P',
    action='store_true',
    dest="poweron",
    default=False,
    help='Flag indicates Power Toggle Target System')

group.add_argument(
    '--poweroff', '-p',
    action='store_true',
    dest="poweroff",
    default=False,
    help='Flag indicates Power Off Target System')

group.add_argument(
    '--reset', '-r',
    action='store_true',
    dest="reset",
    default=False,
    help='Flag indicates Reset Target System')

parser.add_argument(
    '--version', '-v',
    action='version',
    version='%(prog)s v0.1 Empress')

# Parse Args
args = parser.parse_args()

# Setup GPIO
GPIO.setmode(GPIO.BCM) # Set Board Layout
GPIO.setwarnings(False) # Turn Warnings Off
GPIO.setup(pins, GPIO.OUT, initial=GPIO.LOW) # Mark pins as Output

# Check if target is valid
if args.target.lower() in systems:
    # Get target index
    index = systems.index(args.target.lower())
    # If Poweron Flag Raised
    if args.poweron:
        # Apply power to 'power' pin for 200 Milliseconds
        print(powerpin[index])
        press(powerpin[index], .2)
    # If Poweroff Flag Raised
    elif args.poweroff:
        # Apply power to 'power' pin for 5 seconds
        print(powerpin[index])
        press(powerpin[index], 5)
    # If Reset Flag Raised
    elif args.reset:
        # Apply power to 'reset' pin for 200 Milliseconds
        print(resetpin[index])
        press(resetpin[index], .2)
else:
    print("Target System not recognized. Known Systems listed below")
    print(systems)
# Set GPIO pins to default
GPIO.cleanup(pins)