Pyyaml5.1 (#24529)

* Migrated to PyYAML 5.1

* More intelligent fixing of yaml safe_load

Based on https://circleci.com/gh/home-assistant/home-assistant/34831?utm_campaign=vcs-integration-link&utm_medium=referral&utm_source=github-build-link

homeassistant/components/kira/__init__.py

@@ -60,7 +60,7 @@ def load_codes(path):
     codes = []
     if os.path.exists(path):
         with open(path) as code_file:
-            data = yaml.load(code_file) or []
+            data = yaml.safe_load(code_file) or []
         for code in data:
             try:
                 codes.append(CODE_SCHEMA(code))

homeassistant/package_constraints.txt

@@ -17,7 +17,7 @@ netdisco==2.6.0
 pip>=8.0.3
 python-slugify==3.0.2
 pytz>=2019.01
-pyyaml>=3.13,<4
+pyyaml==5.1
 requests==2.22.0
 ruamel.yaml==0.15.97
 sqlalchemy==1.3.3

requirements_all.txt

@@ -12,7 +12,7 @@ cryptography==2.6.1
 pip>=8.0.3
 python-slugify==3.0.2
 pytz>=2019.01
-pyyaml>=3.13,<4
+pyyaml==5.1
 requests==2.22.0
 ruamel.yaml==0.15.97
 voluptuous==0.11.5

setup.py

@@ -46,7 +46,7 @@ REQUIRES = [
     'pip>=8.0.3',
     'python-slugify==3.0.2',
     'pytz>=2019.01',
-    'pyyaml>=3.13,<4',
+    'pyyaml==5.1',
     'requests==2.22.0',
     'ruamel.yaml==0.15.97',
     'voluptuous==0.11.5',

tests/components/apns/test_notify.py

@@ -400,7 +400,7 @@ def test_write_device():
     device = apns.ApnsDevice('123', 'name', 'track_id', True)
 
     apns._write_device(out, device)
-    data = yaml.load(out.getvalue())
+    data = yaml.safe_load(out.getvalue())
     assert data == {
         123: {
             'name': 'name',

tests/components/scene/test_init.py

@@ -90,7 +90,7 @@ class TestScene(unittest.TestCase):
                 self.light_1.entity_id, self.light_2.entity_id)
 
         with io.StringIO(config) as file:
-            doc = yaml_loader.yaml.load(file)
+            doc = yaml_loader.yaml.safe_load(file)
 
         assert setup_component(self.hass, scene.DOMAIN, doc)
         common.activate(self.hass, 'scene.test')