diff options
author | Anthon van der Neut <anthon@mnt.org> | 2017-03-16 08:50:30 +0100 |
---|---|---|
committer | Anthon van der Neut <anthon@mnt.org> | 2017-03-16 08:50:30 +0100 |
commit | f75754aa90ea4d53f87aba5fce692185a25514c3 (patch) | |
tree | 678c37afebe1efd786d513e222a053a3cfe76a07 /composer.py | |
parent | 65537ac45d2fe35a98322aea5e6056baa1aefd8a (diff) | |
download | ruamel.yaml-f75754aa90ea4d53f87aba5fce692185a25514c3.tar.gz |
some changes for mypy --strict
Diffstat (limited to 'composer.py')
-rw-r--r-- | composer.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/composer.py b/composer.py index 791bf50..bbaa62f 100644 --- a/composer.py +++ b/composer.py @@ -5,6 +5,8 @@ from __future__ import print_function import warnings +from typing import Any, Dict, Optional, List # NOQA + from ruamel.yaml.error import MarkedYAMLError, ReusedAnchorWarning from ruamel.yaml.compat import utf8 @@ -25,9 +27,11 @@ class ComposerError(MarkedYAMLError): class Composer(object): def __init__(self): - self.anchors = {} + # type: () -> None + self.anchors = {} # type: Dict[Any, Any] def check_node(self): + # type: () -> Any # Drop the STREAM-START event. if self.check_event(StreamStartEvent): self.get_event() @@ -36,11 +40,13 @@ class Composer(object): return not self.check_event(StreamEndEvent) def get_node(self): + # type: () -> Any # Get the root node of the next document. if not self.check_event(StreamEndEvent): return self.compose_document() def get_single_node(self): + # type: () -> Any # Drop the STREAM-START event. self.get_event() @@ -63,6 +69,7 @@ class Composer(object): return document def compose_document(self): + # type: (Any) -> Any # Drop the DOCUMENT-START event. self.get_event() @@ -76,6 +83,7 @@ class Composer(object): return node def compose_node(self, parent, index): + # type: (Any, Any) -> Any if self.check_event(AliasEvent): event = self.get_event() alias = event.anchor @@ -107,6 +115,7 @@ class Composer(object): return node def compose_scalar_node(self, anchor): + # type: (Any) -> Any event = self.get_event() tag = event.tag if tag is None or tag == u'!': @@ -119,6 +128,7 @@ class Composer(object): return node def compose_sequence_node(self, anchor): + # type: (Any) -> Any start_event = self.get_event() tag = start_event.tag if tag is None or tag == u'!': @@ -144,6 +154,7 @@ class Composer(object): return node def compose_mapping_node(self, anchor): + # type: (Any) -> Any start_event = self.get_event() tag = start_event.tag if tag is None or tag == u'!': @@ -172,6 +183,7 @@ class Composer(object): return node def check_end_doc_comment(self, end_event, node): + # type: (Any, Any) -> None if end_event.comment and end_event.comment[1]: # pre comments on an end_event, no following to move to if node.comment is None: |