Merge pull request #23 from byuccl/usb-power-cli

USB Power CLI

setup.py

@@ -3,11 +3,16 @@
 setup(
     name="yinstruments",
     packages=find_packages(),
-    version="1.5.0",
+    version="1.6.0",
     description="Experiment device control scripts for BYU's Configurable Computing Lab (https://ccl.byu.edu/)",
     author="Jeff Goeders",
     author_email="jeff.goeders@gmail.com",
     url="https://github.com/byuccl/yinstruments",
     install_requires=["pyudev", "pyserial", "pysnmp", "python-vxi11", "pyhubctl"],
-    entry_points={"console_scripts": ["pdu = yinstruments.pdu.cli:main"]},
+    entry_points={
+        "console_scripts": [
+            "pdu = yinstruments.pdu.cli:main",
+            "usb_power = yinstruments.usb_power:cli",
+        ]
+    },
 )

yinstruments/usb_power.py

@@ -97,6 +97,22 @@ def power_on(usb_phys_port, print_output=False):
         return True
 
 
+def cli():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "usb_physical_port", help="The physical port of the USB (1-1.1 for example)"
+    )
+    parser.add_argument("action", choices=["on", "off"])
+
+    args = parser.parse_args()
+
+    p = USBPortPower(args.usb_physical_port)
+    if args.action == "on":
+        p.on()
+    else:
+        p.off()
+
+
 def main():
     #  Power cycle UART USB port
     print("Power cycling UART")