feat: add --notebook flag

crunch/__version__.py

@@ -1,6 +1,6 @@
 __title__ = 'crunch-cli'
 __description__ = 'crunch-cli - CLI of the CrunchDAO Platform'
-__version__ = '0.15.2'
+__version__ = '0.16.0'
 __author__ = 'Enzo CACERES'
 __author_email__ = '[email protected]'
 __url__ = 'https://github.com/crunchdao/crunch-cli'

crunch/main.py

@@ -12,10 +12,12 @@
 @click.option("--debug", "enable_debug", envvar="CRUNCH_DEBUG", is_flag=True, help="Enable debug output.")
 @click.option("--api-base-url", envvar="API_BASE_URL", required=True, help="Set the API base url.")
 @click.option("--web-base-url", envvar="WEB_BASE_URL", required=True, help="Set the Web base url.")
+@click.option("--notebook", is_flag=True, help="Tell the CLI you are running the command while inside a notebook.")
 def cli(
     enable_debug: bool,
     api_base_url: str,
     web_base_url: str,
+    notebook: bool,
 ):
     global debug
     debug = enable_debug
@@ -24,7 +26,8 @@ def cli(
     session = utils.CustomSession(
         web_base_url,
         api_base_url,
-        debug
+        debug,
+        notebook,
     )
 
 

crunch/utils.py

@@ -16,11 +16,12 @@
 class CustomSession(requests.Session):
     # https://stackoverflow.com/a/51026159/7292958
 
-    def __init__(self, web_base_url=None, api_base_url=None, debug=False):
+    def __init__(self, web_base_url=None, api_base_url=None, debug=False, notebook=False):
         super().__init__()
         self.web_base_url = web_base_url
         self.api_base_url = api_base_url
         self.debug = debug
+        self.notebook = notebook
 
     def request(self, method, url, *args, **kwargs):
         response = super().request(
@@ -61,9 +62,11 @@ def format_web_url(self, path: str):
         )
 
     def print_recopy_command(self):
+        tab = "notebook" if self.notebook else "cli"
+
         print("---")
         print("please follow this link to copy and paste your new setup command:")
-        print(self.format_web_url('/submit?tab=cli'))
+        print(self.format_web_url(f'/submit?tab={tab}'))
         print("")