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

Add support for .env loading

This commit is contained in:
isms 2015-11-02 09:09:45 -05:00
parent 6e0dc39a6c
commit f90e83a968
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
{{ cookiecutter.repo_name }}/requirements.txt
View File

@ -2,4 +2,5 @@ click
Sphinx Sphinx
coverage coverage
s3cmd==1.6.0 s3cmd==1.6.0
flake8 flake8
python-dotenv

8
{{ cookiecutter.repo_name }}/src/make_dataset.py
View File

@ -4,8 +4,8 @@ import logging
@click.command() @click.command()
@click.argument('input_filepath', type=str) @click.argument('input_filepath', type=click.Path(exists=True))
@click.argument('output_filepath', type=str) @click.argument('output_filepath', type=click.Path())
def main(input_filepath, output_filepath): def main(input_filepath, output_filepath):
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
logger.info('making final data set from raw data') logger.info('making final data set from raw data')
@ -15,5 +15,9 @@ 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)
project_dir = os.path.join(os.path.dirname(__file__), os.pardir)
dotenv_path = os.path.join(project_dir, '.env')
dotenv.load_dotenv(dotenv_path)
main() main()