__init__.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. from typing import Any, Optional
  2. from .main import dotenv_values, find_dotenv, get_key, load_dotenv, set_key, unset_key
  3. def load_ipython_extension(ipython: Any) -> None:
  4. from .ipython import load_ipython_extension
  5. load_ipython_extension(ipython)
  6. def get_cli_string(
  7. path: Optional[str] = None,
  8. action: Optional[str] = None,
  9. key: Optional[str] = None,
  10. value: Optional[str] = None,
  11. quote: Optional[str] = None,
  12. ):
  13. """Returns a string suitable for running as a shell script.
  14. Useful for converting a arguments passed to a fabric task
  15. to be passed to a `local` or `run` command.
  16. """
  17. command = ["dotenv"]
  18. if quote:
  19. command.append(f"-q {quote}")
  20. if path:
  21. command.append(f"-f {path}")
  22. if action:
  23. command.append(action)
  24. if key:
  25. command.append(key)
  26. if value:
  27. if " " in value:
  28. command.append(f'"{value}"')
  29. else:
  30. command.append(value)
  31. return " ".join(command).strip()
  32. __all__ = [
  33. "get_cli_string",
  34. "load_dotenv",
  35. "dotenv_values",
  36. "get_key",
  37. "set_key",
  38. "unset_key",
  39. "find_dotenv",
  40. "load_ipython_extension",
  41. ]