WIP feat(cli) add option to disable api in command line

Author: prmths128

codecarbon/cli/cli_utils.py

@@ -37,6 +37,21 @@ def get_api_endpoint(path: Optional[Path] = None):
     return "https://api.codecarbon.io"
 
 
+def get_api_enabled(path: Optional[Path] = None):
+    p = path or Path.cwd().resolve() / ".codecarbon.config"
+    if p.exists():
+        config = configparser.ConfigParser()
+        config.read(str(p))
+        if "codecarbon" in config.sections():
+            d = dict(config["codecarbon"])
+            if "api_enabled" in d:
+                return "1"
+            else:
+                with p.open("a") as f:
+                    f.write("api_enabled=0\n")
+    return "0"
+
+
 def get_existing_local_exp_id(path: Optional[Path] = None):
     p = path or Path.cwd().resolve() / ".codecarbon.config"
     if p.exists():

codecarbon/cli/main.py

@@ -2,6 +2,7 @@
 import time
 from pathlib import Path
 from typing import Optional
+from uuid import uuid4
 
 import questionary
 import requests
@@ -15,6 +16,7 @@
 from codecarbon import __app_name__, __version__
 from codecarbon.cli.cli_utils import (
     create_new_config_file,
+    get_api_enabled,
     get_api_endpoint,
     get_config,
     get_existing_local_exp_id,
@@ -177,13 +179,29 @@ def config():
     else:
         file_path = create_new_config_file()
 
+    api_enabled = get_api_enabled()
+    api_enabled = typer.prompt(
+        f"API is currently {'enabled' if api_enabled == '1' else 'disabled'}. Press enter to continue or change to 0/1",
+        type=str,
+        default="0",
+    )
+    overwrite_local_config("api_enabled", api_enabled, path=file_path)
+
+    if api_enabled == "0":
+        overwrite_local_config("experiment_id", str(uuid4()), path=file_path)
+        return
+
     api_endpoint = get_api_endpoint(file_path)
     api_endpoint = typer.prompt(
-        f"Current API endpoint is {api_endpoint}. Press enter to continue or input other url",
+        f"Current API endpoint is at {api_endpoint}. Press enter to continue or input other url",
         type=str,
         default=api_endpoint,
     )
     overwrite_local_config("api_endpoint", api_endpoint, path=file_path)
+
+    print(f"Logging in to auth server at {AUTH_SERVER_URL}")
+    fief_auth.authorize()
+
     api = ApiClient(endpoint_url=api_endpoint)
     api.set_access_token(_get_access_token())
     organizations = api.get_list_organizations()