From cea41547886e0a46e7bebabad9de04821f2967b0 Mon Sep 17 00:00:00 2001 From: Michael Kaye <1917473+michaelkaye@users.noreply.github.com> Date: Wed, 5 Jun 2019 10:01:22 +0100 Subject: [PATCH] Move to safe_yaml per https://msg.pyyaml.org/load --- scripts/s3_media_upload | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/s3_media_upload b/scripts/s3_media_upload index b8e08d8..672c175 100644 --- a/scripts/s3_media_upload +++ b/scripts/s3_media_upload @@ -311,7 +311,7 @@ def get_postgres_conn(parser): """Attempt to get a postgres connection based on database.yaml, or exit. """ try: - database_yaml = yaml.load(open("database.yaml")) + database_yaml = yaml.safe_load(open("database.yaml")) except FileNotFoundError: parser.error("Could not find database.yaml") except yaml.YAMLError as e: