diff options
| -rw-r--r-- | crocoite/behavior.py | 2 | ||||
| -rw-r--r-- | crocoite/test_behavior.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/crocoite/behavior.py b/crocoite/behavior.py index 3ecc92f..2d1dd0d 100644 --- a/crocoite/behavior.py +++ b/crocoite/behavior.py @@ -311,7 +311,7 @@ class Click (JsOnload): def __init__ (self, loader, logger): super ().__init__ (loader, logger) with pkg_resources.resource_stream (__name__, os.path.join ('data', 'click.yaml')) as fd: - self.options['sites'] = list (yaml.load_all (fd)) + self.options['sites'] = list (yaml.safe_load_all (fd)) class ExtractLinksEvent: __slots__ = ('links', ) diff --git a/crocoite/test_behavior.py b/crocoite/test_behavior.py index 2c13804..8a0dff4 100644 --- a/crocoite/test_behavior.py +++ b/crocoite/test_behavior.py @@ -32,7 +32,7 @@ from .behavior import Scroll, Behavior, ExtractLinks, ExtractLinksEvent from .controller import SinglePageController, EventHandler with pkg_resources.resource_stream (__name__, os.path.join ('data', 'click.yaml')) as fd: - sites = list (yaml.load_all (fd)) + sites = list (yaml.safe_load_all (fd)) clickParam = [] for o in sites: for s in o['selector']: |