From ae2f3fbb493fdaad56d6e2b1b7a526482f917fd5 Mon Sep 17 00:00:00 2001 From: Jeremy Epstein Date: Wed, 8 Apr 2015 14:12:23 +1000 Subject: [PATCH] fix keyerror if S3_HEADERS isn't set in config --- flask_s3.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/flask_s3.py b/flask_s3.py index c54a156..e116605 100644 --- a/flask_s3.py +++ b/flask_s3.py @@ -148,12 +148,14 @@ def _write_files(app, static_url_loc, static_folder, files, bucket, k = Key(bucket=bucket, name=key_name) # Set custom headers - for header, value in app.config['S3_HEADERS'].iteritems(): - k.set_metadata(header, value) + headers = app.config.get('S3_HEADERS') + if headers: + for header, value in headers.iteritems(): + k.set_metadata(header, value) # Set more custom headers if the filepath matches certain # configured regular expressions. - filepath_headers = app.config['S3_FILEPATH_HEADERS'] + filepath_headers = app.config.get('S3_FILEPATH_HEADERS') if filepath_headers: for filepath_regex, headers in filepath_headers.iteritems(): if re.search(filepath_regex, file_path):