Browse Source

Add support for .env loading

main
isms 9 years ago
parent
commit
f90e83a968
  1. 1
      {{ cookiecutter.repo_name }}/requirements.txt
  2. 8
      {{ cookiecutter.repo_name }}/src/make_dataset.py

1
{{ cookiecutter.repo_name }}/requirements.txt

@ -3,3 +3,4 @@ Sphinx
coverage
s3cmd==1.6.0
flake8
python-dotenv

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

@ -4,8 +4,8 @@ import logging
@click.command()
@click.argument('input_filepath', type=str)
@click.argument('output_filepath', type=str)
@click.argument('input_filepath', type=click.Path(exists=True))
@click.argument('output_filepath', type=click.Path())
def main(input_filepath, output_filepath):
logger = logging.getLogger(__name__)
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'
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()

Loading…
Cancel
Save