public
/
cookiecutter-data-science
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update docs and src for new python-dotenv release

This commit is contained in:
Isaac Slavitt 2016-05-01 14:14:56 -04:00
parent 880c55ee4b
commit 7a9232ff4e
2 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
docs/docs/index.md
View File

@ -203,10 +203,14 @@ If you look at the stub script in `src/data/make_dataset.py`, it uses a package
```python ```python
# src/data/dotenv_example.py # src/data/dotenv_example.py
from os.path import join, dirname from os.path import join, dirname
from dotenv import load_dotenv from dotenv import load_dotenv, find_dotenv
dotenv_path = join(dirname(__file__), os.pardir, os.pardir, '.env') # up two levels to root folder # find .env automagically by walking up directories until it's found
dotenv_path = find_dotenv()
# load up the entries as environment variables
load_dotenv(dotenv_path) load_dotenv(dotenv_path)
database_url = os.environ.get("DATABASE_URL") database_url = os.environ.get("DATABASE_URL")
other_variable = os.environ.get("OTHER_VARIABLE") other_variable = os.environ.get("OTHER_VARIABLE")
``` ```

9
{{ cookiecutter.repo_name }}/src/data/make_dataset.py
View File

@ -1,8 +1,8 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import os import os
import click import click
import dotenv
import logging import logging
from dotenv import find_dotenv, load_dotenv
@click.command() @click.command()
@ -17,9 +17,12 @@ if __name__ == '__main__':
log_fmt = '%(asctime)s - %(name)s - %(levelname)s - %(message)s' log_fmt = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
logging.basicConfig(level=logging.INFO, format=log_fmt) logging.basicConfig(level=logging.INFO, format=log_fmt)
# not used in this stub but often useful for finding various files
project_dir = os.path.join(os.path.dirname(__file__), os.pardir, os.pardir) project_dir = os.path.join(os.path.dirname(__file__), os.pardir, os.pardir)
dotenv_path = os.path.join(project_dir, '.env')
dotenv.load_dotenv(dotenv_path) # find .env automagically by walking up directories until it's found, then
# load up the .env entries as environment variables
load_dotenv(find_dotenv())
main() main()