Use type=str instead of Choice for spec names

sdss · Oct 16, 2023 · ef4f3d9 · ef4f3d9
1 parent bcd058f
commit ef4f3d9
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 14 deletions.
diff --git a/python/lvmieb/actor/commands/hartmann.py b/python/lvmieb/actor/commands/hartmann.py
@@ -41,7 +41,7 @@ def hartmann(*args):
     default="all",
     help="all, right, or left",
 )
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def open(
     command: IEBCommand,
     controllers: ControllersType,
@@ -90,7 +90,7 @@ async def open(
     default="all",
     help="all, right, or left",
 )
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def close(
     command: IEBCommand,
     controllers: ControllersType,
@@ -132,7 +132,7 @@ async def close(
 
 
 @hartmann.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def status(
     command: IEBCommand,
     controllers: ControllersType,
@@ -182,7 +182,7 @@ async def status(
 
 
 @hartmann.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def init(
     command: IEBCommand,
     controllers: ControllersType,
@@ -215,7 +215,7 @@ async def init(
 
 
 @hartmann.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def home(
     command: IEBCommand,
     controllers: ControllersType,

diff --git a/python/lvmieb/actor/commands/shutter.py b/python/lvmieb/actor/commands/shutter.py
@@ -35,7 +35,7 @@ def shutter(*args):
 
 
 @shutter.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def open(
     command: IEBCommand,
     controllers: ControllersType,
@@ -68,7 +68,7 @@ async def open(
 
 
 @shutter.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def close(
     command: IEBCommand,
     controllers: ControllersType,
@@ -101,7 +101,7 @@ async def close(
 
 
 @shutter.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def status(
     command: IEBCommand,
     controllers: ControllersType,
@@ -146,7 +146,7 @@ async def status(
 
 
 @shutter.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def init(
     command: IEBCommand,
     controllers: ControllersType,
@@ -178,7 +178,7 @@ async def init(
 
 
 @shutter.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def home(
     command: IEBCommand,
     controllers: ControllersType,

diff --git a/python/lvmieb/actor/commands/transducer.py b/python/lvmieb/actor/commands/transducer.py
@@ -18,7 +18,7 @@ def transducer(*args):
 
 
 @transducer.command()
-@click.argument("spectro", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("spectro", type=str, required=False)
 async def status(
     command: IEBCommand,
     controllers: ControllersType,

diff --git a/python/lvmieb/actor/commands/wago.py b/python/lvmieb/actor/commands/wago.py
@@ -27,7 +27,7 @@ def wago(*args):
 
 
 @wago.command()
-@click.argument("SPECTRO", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("SPECTRO", type=str, required=False)
 async def status(
     command: IEBCommand,
     controllers: ControllersType,
@@ -66,7 +66,7 @@ async def status(
 
 
 @wago.command()
-@click.argument("SPECTRO", type=click.Choice(["sp1", "sp2", "sp3"]), required=False)
+@click.argument("SPECTRO", type=str, required=False)
 async def getpower(
     command: IEBCommand,
     controllers: ControllersType,
@@ -111,7 +111,7 @@ async def getpower(
 )
 @click.argument(
     "SPECTRO",
-    type=click.Choice(["sp1", "sp2", "sp3"]),
+    type=str,
     required=False,
 )
 @click.option(