python-setenv 0.0.1__tar.gz

python_setenv-0.0.1/.gitignore

@@ -0,0 +1,10 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv

python_setenv-0.0.1/.python-version

@@ -0,0 +1 @@
+3.12

python_setenv-0.0.1/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: python-setenv
+Version: 0.0.1
+Summary: Set env (e.g. HF_TOKEN) from colab/kaggle secrets or dotenv .env env files
+Author-email: ffreemt <yucongo+fmt@gmail.com>
+Requires-Python: >=3.10
+Requires-Dist: python-dotenv>=1.2.1

python_setenv-0.0.1/pyproject.toml

@@ -0,0 +1,19 @@
+[project]
+name = "python-setenv"
+version = "0.0.1"
+description = "Set env (e.g. HF_TOKEN) from colab/kaggle secrets or dotenv .env env files"
+readme = "README.md"
+authors = [
+    { name = "ffreemt", email = "yucongo+fmt@gmail.com" }
+]
+requires-python = ">=3.10"
+dependencies = [
+    "python-dotenv>=1.2.1",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/setenv"]

python_setenv-0.0.1/src/setenv/__init__.py

@@ -0,0 +1,131 @@
+"""Set an environ var in colab/kaggle/dotenv(search for .env dotenv env).
+
+!pip install -q python-dotenv
+"""
+# pylint: disable=import-outside-toplevel,too-many-statements,too-many-branches
+
+import os
+# import sys
+
+from pathlib import Path
+from dotenv import dotenv_values, find_dotenv
+# from loguru import logger
+
+__version__ = "0.0.1"
+
+
+def hello() -> str:
+    return "Hello from python-setenv!"
+
+
+# def set_env(
+def setenv(
+    env_var="HF_TOKEN", source_var=None, envfile=None, override=False,
+    # verbose=False
+):
+    """
+    Set environ var via google userdat/kaggle_secrets/[.env/dotenv/env].
+
+    Args:
+    ----
+    env_var:
+        env var to set, default HF_TOKEN.
+    source_var:
+        var from google/kaggle secret/userdata
+        or defined in .env/dotenv/env
+    envfile: file to read from, default, search
+        for ['.env', 'dotenv', 'env']
+    override:
+        reset if set to True
+    x verbose:
+        for loguru.logging, print TRACE if set
+
+    Returns:
+    -------
+    value of the env var.
+    """
+    _ = '''
+    if verbose:
+        logger.remove()
+        logger.add(sys.stderr, level="TRACE")  # default "DEBUG"
+    # '''
+    if not override and os.getenv(env_var):
+        return os.getenv(env_var)
+
+    # if override:
+    save_env_var = os.getenv(env_var)
+    if os.getenv(env_var) is not None:
+        del os.environ[env_var]
+
+    if source_var is None:
+        source_var = env_var
+
+    # if KAGGLE
+    if "KAGGLE_KERNEL" in " ".join(os.environ.keys()):
+        # import kaggle_web_client  # type: ignore
+        from kaggle_secrets import UserSecretsClient  # type: ignore
+
+        user_secrets = UserSecretsClient()
+        try:
+            os.environ[env_var] = user_secrets.get_secret(source_var)
+            return os.getenv(env_var)
+        except Exception:
+            return ""
+
+    # try colab COLAB_NOTEBOOK
+    if "COLAB_NOTEBOOK" in " ".join(os.environ.keys()):
+        from google.colab import userdata  # noqa  # type: ignore
+        try:
+            os.environ[env_var] = userdata.get(source_var)
+            return os.getenv(env_var)
+        except Exception:
+            return ""
+
+    # .env dotenv env
+    # envfile = None
+
+    # cwd
+    print(f"cwd: {Path.cwd()}, {envfile=}")
+
+    if envfile is None:
+        for _ in [".env", "dotenv", "env"]:
+            # filepath = Path.cwd() / _
+            # envfile = find_dotenv(filepath.as_posix())
+            filepath = _
+            envfile = find_dotenv(filepath)
+            if envfile:
+                # logger.trace(f"Found {envfile=}")
+                print(f"Found {envfile=}")
+                break
+
+    if envfile:
+        print(f"loading {envfile=} with dotenv_values(envfile)")
+        if dotenv_values(envfile).get(source_var):
+            _ = dotenv_values(envfile).get(source_var)
+            if _:  # need to be a str
+                os.environ[env_var] = _
+            print(f"{envfile=}: set {env_var}={source_var}")
+    if os.getenv(env_var):
+        return os.getenv(env_var)
+
+    # try manual input?
+    print(
+        f"""
+        Unable to set {env_var}={source_var},
+        not in colab or Secrets not set, not kaggle
+        or Secrets not set, no .env/dotenv/env file
+        in the current working dir or parent dirs or env not present."""
+    )
+
+    # restore?
+    if override:
+        # save_env_var may be None
+        if save_env_var:
+            if os.getenv(env_var) is not None:
+                os.environ[env_var] = save_env_var
+
+        print(f"Restore {env_var=}")
+
+    return ""
+
+__all__ = ("__version__", "setenv",)

python_setenv-0.0.1/uv.lock

@@ -0,0 +1,23 @@
+version = 1
+revision = 2
+requires-python = ">=3.10"
+
+[[package]]
+name = "python-dotenv"
+version = "1.2.1"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/f0/26/19cadc79a718c5edbec86fd4919a6b6d3f681039a2f6d66d14be94e75fb9/python_dotenv-1.2.1.tar.gz", hash = "sha256:42667e897e16ab0d66954af0e60a9caa94f0fd4ecf3aaf6d2d260eec1aa36ad6", size = 44221, upload_time = "2025-10-26T15:12:10.434Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/14/1b/a298b06749107c305e1fe0f814c6c74aea7b2f1e10989cb30f544a1b3253/python_dotenv-1.2.1-py3-none-any.whl", hash = "sha256:b81ee9561e9ca4004139c6cbba3a238c32b03e4894671e181b671e8cb8425d61", size = 21230, upload_time = "2025-10-26T15:12:09.109Z" },
+]
+
+[[package]]
+name = "python-setenv"
+version = "0.1.0"
+source = { editable = "." }
+dependencies = [
+    { name = "python-dotenv" },
+]
+
+[package.metadata]
+requires-dist = [{ name = "python-dotenv", specifier = ">=1.2.1" }]