cli_auth: extract method

grid5000/cli.py

@@ -40,10 +40,19 @@ def main():
     IPython.embed(header=motd)
 
 
-def auth():
-    import argparse
+def auth(user: str):
     from getpass import getpass
 
+    path = pathlib.Path(CONF_PATH)
+
+    password = getpass("Grid'5000 password: ")
+    path.write_text("\n".join([f"username: {user}", f"password: {password}"]))
+    print(f"{CONF_PATH} created successfully")
+
+
+def cli_auth():
+    import argparse
+
     parser = argparse.ArgumentParser(description=f"Check or create {CONF_PATH}")
     parser.add_argument("--user", "-u", help="Username on Grid'5000", required=True)
     args = parser.parse_args()
@@ -52,11 +61,4 @@ def auth():
     if path.exists():
         print(f"{CONF_PATH} file already exists, not overwriting")
         return
-    user = args.user
-
-    password = getpass("Grid'5000 password: ")
-    path.write_text("\n".join([
-        f"username: {user}",
-        f"password: {password}"
-    ]))
-    print(f"{CONF_PATH} created successfully")
+    auth(args.user)

setup.cfg

@@ -35,4 +35,4 @@ exclude =
 [options.entry_points]
 console_scripts =
     grid5000 = grid5000.cli:main
-    grid5000-auth = grid5000.cli:auth
+    grid5000-auth = grid5000.cli:cli_auth
\ No newline at end of file