Upload key updates

* Record the key id for a specific unit and don't allow multiple keys for that unit.
panoptes · Jan 10, 2024 · 3d7ec52 · 3d7ec52
1 parent 9e2a788
commit 3d7ec52
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/panoptes/pocs/utils/cli/network.py b/src/panoptes/pocs/utils/cli/network.py
@@ -17,7 +17,8 @@
 
 
 @app.command('get-key')
-def get_key_cmd(secret_password: str = typer.Option(..., prompt=True, hide_input=True),
+def get_key_cmd(unit_id: str = typer.Option(..., prompt=True),
+                secret_password: str = typer.Option(..., prompt=True, hide_input=True),
                 save_dir: Path = Path('~/keys').expanduser(),
                 key_name: str = 'panoptes-upload-key.json',
                 enable_image_upload: bool = True,
@@ -34,7 +35,7 @@ def get_key_cmd(secret_password: str = typer.Option(..., prompt=True, hide_input
         return
 
     # Make a request for the key.
-    res = requests.post(url, json=dict(pwd=secret_password))
+    res = requests.post(url, json=dict(pwd=secret_password, unit_id=unit_id))
     if not res.ok:
         print(f'[red]Error getting key: {res.text}[/]')
         return