feat(web): add nexus environment (FXC-1750) (#2643)

* feat(web): add use_nexus function

* add docstring

* remove use_nexus function and add env_vars to EnvironmentConfig

Author: frederikschubertflex

tidy3d/web/core/environment.py

@@ -4,6 +4,7 @@
 
 import os
 import ssl
+from typing import Optional
 
 from pydantic.v1 import BaseSettings, Field
 
@@ -21,8 +22,9 @@ def __hash__(self):
     website_endpoint: str
     s3_region: str
     ssl_verify: bool = Field(True, env="TIDY3D_SSL_VERIFY")
-    enable_caching: bool = None
-    ssl_version: ssl.TLSVersion = None
+    enable_caching: Optional[bool] = None
+    ssl_version: Optional[ssl.TLSVersion] = None
+    env_vars: Optional[dict[str, str]] = None
 
     def active(self) -> None:
         """Activate the environment instance."""
@@ -73,6 +75,17 @@ def get_real_url(self, path: str) -> str:
 )
 
 
+nexus = EnvironmentConfig(
+    name="nexus",
+    web_api_endpoint="http://127.0.0.1:5000",
+    ssl_verify=False,
+    enable_caching=False,
+    s3_region="us-east-1",
+    website_endpoint="http://127.0.0.1/tidy3d",
+    env_vars={"AWS_ENDPOINT_URL_S3": "http://127.0.0.1:9000"},
+)
+
+
 class Environment:
     """Environment decorator for user interactive.
 
@@ -88,11 +101,13 @@ class Environment:
         "dev": dev,
         "uat": uat,
         "prod": prod,
+        "nexus": nexus,
     }
 
     def __init__(self):
         log = get_logger()
         """Initialize the environment."""
+        self._previous_env_vars = {}
         env_key = os.environ.get("TIDY3D_ENV")
         env_key = env_key.lower() if env_key else env_key
         log.info(f"env_key is {env_key}")
@@ -107,6 +122,11 @@ def __init__(self):
             )
             self._current = prod
 
+        if self._current.env_vars:
+            for key, value in self._current.env_vars.items():
+                self._previous_env_vars[key] = os.environ.get(key)
+                os.environ[key] = value
+
     @property
     def current(self) -> EnvironmentConfig:
         """Get the current environment.
@@ -162,6 +182,17 @@ def prod(self) -> EnvironmentConfig:
         """
         return prod
 
+    @property
+    def nexus(self) -> EnvironmentConfig:
+        """Get the nexus environment.
+
+        Returns
+        -------
+        EnvironmentConfig
+            The config for the nexus environment.
+        """
+        return nexus
+
     def set_current(self, config: EnvironmentConfig) -> None:
         """Set the current environment.
 
@@ -170,6 +201,19 @@ def set_current(self, config: EnvironmentConfig) -> None:
         config : EnvironmentConfig
             The environment to set to current.
         """
+        for key, value in self._previous_env_vars.items():
+            if value is None:
+                if key in os.environ:
+                    del os.environ[key]
+            else:
+                os.environ[key] = value
+        self._previous_env_vars = {}
+
+        if config.env_vars:
+            for key, value in config.env_vars.items():
+                self._previous_env_vars[key] = os.environ.get(key)
+                os.environ[key] = value
+
         self._current = config
 
     def enable_caching(self, enable_caching: bool = True) -> None: