Fix type hinting for Python3.9

pull/41/head
Kevin Nobel 2 years ago
parent 0846f50463
commit bdb62795b7
No known key found for this signature in database
GPG Key ID: 3315E34C37CA7D87

@ -8,7 +8,7 @@ import re
import sys import sys
import traceback import traceback
import urllib import urllib
from typing import Any, Sequence from typing import Any, Optional, Sequence, Union
import ruamel.yaml import ruamel.yaml
from mitmproxy.exceptions import FlowReadException from mitmproxy.exceptions import FlowReadException
@ -63,7 +63,7 @@ def detect_input_format(file_path):
return MitmproxyCaptureReader(file_path, progress_callback) return MitmproxyCaptureReader(file_path, progress_callback)
def main(override_args: Sequence[str] | None = None): def main(override_args: Optional[Sequence[str]] = None):
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description="Converts a mitmproxy dump file or HAR to a swagger schema." description="Converts a mitmproxy dump file or HAR to a swagger schema."
) )
@ -102,7 +102,7 @@ def main(override_args: Sequence[str] | None = None):
yaml = ruamel.yaml.YAML() yaml = ruamel.yaml.YAML()
capture_reader: MitmproxyCaptureReader | HarCaptureReader capture_reader: Union[MitmproxyCaptureReader, HarCaptureReader]
if args.format == "flow" or args.format == "mitmproxy": if args.format == "flow" or args.format == "mitmproxy":
capture_reader = MitmproxyCaptureReader(args.input, progress_callback) capture_reader = MitmproxyCaptureReader(args.input, progress_callback)
elif args.format == "har": elif args.format == "har":

@ -1,7 +1,7 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import tempfile import tempfile
from typing import Any, List from typing import Any, List, Optional
import ruamel.yaml as ruamel import ruamel.yaml as ruamel
@ -21,7 +21,7 @@ def get_nested_key(obj: Any, path: str) -> Any:
def mitmproxy2swagger_e2e_test( def mitmproxy2swagger_e2e_test(
input_file: str, url_prefix: str, extra_args: List[str] | None = None input_file: str, url_prefix: str, extra_args: Optional[List[str]] = None
) -> Any: ) -> Any:
"""Runs mitmproxy2swagger on the given input file twice, and returns the """Runs mitmproxy2swagger on the given input file twice, and returns the
detected endpoints.""" detected endpoints."""

Loading…
Cancel
Save