Spaces:
Running
Running
from IPython.core.magic import Magics, line_magic, magics_class # type: ignore | |
from IPython.core.magic_arguments import (argument, magic_arguments, # type: ignore | |
parse_argstring) # type: ignore | |
from .main import find_dotenv, load_dotenv | |
class IPythonDotEnv(Magics): | |
def dotenv(self, line): | |
args = parse_argstring(self.dotenv, line) | |
# Locate the .env file | |
dotenv_path = args.dotenv_path | |
try: | |
dotenv_path = find_dotenv(dotenv_path, True, True) | |
except IOError: | |
print("cannot find .env file") | |
return | |
# Load the .env file | |
load_dotenv(dotenv_path, verbose=args.verbose, override=args.override) | |
def load_ipython_extension(ipython): | |
"""Register the %dotenv magic.""" | |
ipython.register_magics(IPythonDotEnv) | |