Hoist out helper fns
Note that since __main__ is in global scope, no closure conversion is required
This commit is contained in:
parent
b18edf1c4a
commit
522f1b6628
1 changed files with 152 additions and 144 deletions
|
@ -111,80 +111,8 @@ nfsboot = "/var/lib/clusterctrl/boot/"
|
|||
nfsroot = "/var/lib/clusterctrl/nfs/"
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
args = len(sys.argv)
|
||||
|
||||
if (args == 1 or sys.argv[1] == "help" or sys.argv[1] == "--help" or sys.argv[1] == "-h" or sys.argv[1] == "/?"):
|
||||
print("""Usage :{0} <cmd>
|
||||
|
||||
<devices> can be a single device 'p1' or a list 'p2 p3 p5'
|
||||
<order> is the order listed by '{0} status' (default 20)
|
||||
|
||||
# Power on/off all or listed device(s)
|
||||
{0} on|off [<devices>]
|
||||
|
||||
# Show status of ClusterHAT/CTRL
|
||||
{0} status
|
||||
|
||||
# Get number of controllable Pi
|
||||
{0} maxpi
|
||||
|
||||
# Create/update symlinks for rpiboot [root]"
|
||||
sudo {0} init
|
||||
|
||||
# Turn ALERT LED on/off for all or listed device(s)"
|
||||
{0} alert on|off [<devices>]
|
||||
|
||||
# Enable LED (Power/pX/etc.)
|
||||
{0} led on
|
||||
|
||||
# Disable LED (Power/pX/etc.)
|
||||
{0} led off
|
||||
|
||||
# Turns on/off or resets the USB HUB
|
||||
{0} hub off|on|reset
|
||||
|
||||
## The following are only available on ClusterCTRL devices
|
||||
# Set order on device <old> to <new>
|
||||
{0} setorder <old> <new>
|
||||
|
||||
# Get USB path to Px
|
||||
{0} getpath <device>
|
||||
|
||||
# Turns FAN on/off for CTRL with <order>
|
||||
{0} fan on|off <order>
|
||||
|
||||
# Save current settings to EEPROM
|
||||
{0} save <order>
|
||||
|
||||
# Save current order to EEPROM
|
||||
{0} saveorder <order>
|
||||
|
||||
# Save current Power On State to EEPROM
|
||||
{0} savepos <order>
|
||||
|
||||
# Save factory default settings to EEPROM
|
||||
{0} savedefaults <order>
|
||||
""" .format(sys.argv[0]))
|
||||
sys.exit()
|
||||
|
||||
# Read configruation file
|
||||
config = {}
|
||||
if os.path.isfile("/etc/default/clusterctrl"):
|
||||
with open ("/etc/default/clusterctrl") as configfile:
|
||||
for line in configfile:
|
||||
if (line[:1] != "#"):
|
||||
k, v = line.partition("=")[::2]
|
||||
config[k.strip().lower()] = v.split("#")[0].strip(" \"'\n\t")
|
||||
|
||||
# If we're not a controller of some sort exit cleanly
|
||||
if ("type" not in config or not (config["type"] == "c" or config["type"] == "cnat")):
|
||||
print("Unable to load config, or invalid config loaded", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
# Functions
|
||||
# Send command to ClusterCTRL via I2C
|
||||
def send_cmd(c, cmd, data0=None, data1=None, data2=None, data3=None, data4=None, data5=None, data6=None, data7=None):
|
||||
def send_cmd(c, cmd, data0=None, data1=None, data2=None, data3=None, data4=None, data5=None, data6=None, data7=None):
|
||||
"""Send command to ClusterCTRL via I2C."""
|
||||
#print("CMD: {} - {} {} {} {} {} {} {} {}"format(cmd, data0, data1, data2, data3, data4, data5, data6, data7))
|
||||
if (data7 is not None): c[1].write_byte_data(I2C_ADDRESS, REG_DATA7, data7)
|
||||
if (data6 is not None): c[1].write_byte_data(I2C_ADDRESS, REG_DATA6, data6)
|
||||
|
@ -199,32 +127,40 @@ if __name__ == "__main__":
|
|||
except IOError:
|
||||
return False
|
||||
|
||||
# Read register from ClusterCTRL via I2C
|
||||
def read_reg(c, offset, len=1):
|
||||
|
||||
def read_reg(c, offset, len=1):
|
||||
"""Read register from ClusterCTRL via I2C."""
|
||||
|
||||
if (len>1):
|
||||
tmp = c[1].read_i2c_block_data(I2C_ADDRESS, offset, len)
|
||||
else:
|
||||
tmp = c[1].read_byte_data(I2C_ADDRESS, offset)
|
||||
return tmp
|
||||
|
||||
# Get throttled status
|
||||
def get_throttled():
|
||||
|
||||
def get_throttled():
|
||||
"""Get throttled status."""
|
||||
if (not os.path.isfile(vcgencmdpath) or not os.access(vcgencmdpath, os.X_OK)):
|
||||
return "NA"
|
||||
return ((os.popen(vcgencmdpath + " get_throttled").readline()).split("=", 1)[-1].strip())
|
||||
|
||||
# Get USB path (eg 1-1.4.1) for I2C bus
|
||||
def usbpathfrombus(bus):
|
||||
|
||||
def usbpathfrombus(bus):
|
||||
"""Get USB path (eg 1-1.4.1) for I2C bus."""
|
||||
|
||||
for device in glob.glob("/sys/bus/usb/drivers/i2c-tiny-usb/*/i2c*"):
|
||||
parts = device.split("/")
|
||||
path = parts[6].split(":")[0]
|
||||
id = parts[7][4:]
|
||||
if int(id) == bus:
|
||||
return path
|
||||
|
||||
return False
|
||||
|
||||
# Build list of pi zero numbers to get USB path of
|
||||
def getusbpaths():
|
||||
|
||||
def getusbpaths():
|
||||
"""Build list of pi zero numbers to get USB path of."""
|
||||
|
||||
paths = {}
|
||||
zeros = []
|
||||
|
||||
|
@ -319,13 +255,85 @@ if __name__ == "__main__":
|
|||
paths[str(zero)] = cache_clusterctrl[c[0]]+pathdata
|
||||
return paths
|
||||
|
||||
def is_float(n):
|
||||
def is_float(n):
|
||||
try:
|
||||
float(n)
|
||||
return True
|
||||
except ValueError:
|
||||
return False
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
args = len(sys.argv)
|
||||
|
||||
if (args == 1 or sys.argv[1] == "help" or sys.argv[1] == "--help" or sys.argv[1] == "-h" or sys.argv[1] == "/?"):
|
||||
print("""Usage :{0} <cmd>
|
||||
|
||||
<devices> can be a single device 'p1' or a list 'p2 p3 p5'
|
||||
<order> is the order listed by '{0} status' (default 20)
|
||||
|
||||
# Power on/off all or listed device(s)
|
||||
{0} on|off [<devices>]
|
||||
|
||||
# Show status of ClusterHAT/CTRL
|
||||
{0} status
|
||||
|
||||
# Get number of controllable Pi
|
||||
{0} maxpi
|
||||
|
||||
# Create/update symlinks for rpiboot [root]"
|
||||
sudo {0} init
|
||||
|
||||
# Turn ALERT LED on/off for all or listed device(s)"
|
||||
{0} alert on|off [<devices>]
|
||||
|
||||
# Enable LED (Power/pX/etc.)
|
||||
{0} led on
|
||||
|
||||
# Disable LED (Power/pX/etc.)
|
||||
{0} led off
|
||||
|
||||
# Turns on/off or resets the USB HUB
|
||||
{0} hub off|on|reset
|
||||
|
||||
## The following are only available on ClusterCTRL devices
|
||||
# Set order on device <old> to <new>
|
||||
{0} setorder <old> <new>
|
||||
|
||||
# Get USB path to Px
|
||||
{0} getpath <device>
|
||||
|
||||
# Turns FAN on/off for CTRL with <order>
|
||||
{0} fan on|off <order>
|
||||
|
||||
# Save current settings to EEPROM
|
||||
{0} save <order>
|
||||
|
||||
# Save current order to EEPROM
|
||||
{0} saveorder <order>
|
||||
|
||||
# Save current Power On State to EEPROM
|
||||
{0} savepos <order>
|
||||
|
||||
# Save factory default settings to EEPROM
|
||||
{0} savedefaults <order>
|
||||
""" .format(sys.argv[0]))
|
||||
sys.exit()
|
||||
|
||||
# Read configruation file
|
||||
config = {}
|
||||
if os.path.isfile("/etc/default/clusterctrl"):
|
||||
with open ("/etc/default/clusterctrl") as configfile:
|
||||
for line in configfile:
|
||||
if (line[:1] != "#"):
|
||||
k, v = line.partition("=")[::2]
|
||||
config[k.strip().lower()] = v.split("#")[0].strip(" \"'\n\t")
|
||||
|
||||
# If we're not a controller of some sort exit cleanly
|
||||
if ("type" not in config or not (config["type"] == "c" or config["type"] == "cnat")):
|
||||
print("Unable to load config, or invalid config loaded", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
##########
|
||||
# Init #
|
||||
##########
|
||||
|
|
Loading…
Reference in a new issue