Merge pull request #2026 from Kriechi/filename-matching

add filename-matching linter

1 files changed, 33 insertions, 0 deletions

diff --git a/test/helper_tools/inspect_dumpfile.py b/test/helper_tools/inspect_dumpfile.py
new file mode 100644
index 00000000..b2201f40
--- /dev/null
+++ b/test/helper_tools/inspect_dumpfile.py
@@ -0,0 +1,33 @@
+from pprint import pprint
+
+import click
+
+from mitmproxy import tnetstring
+
+
+def read_tnetstring(input):
+    # tnetstring throw a ValueError on EOF, which is hard to catch
+    # because they raise ValueErrors for a couple of other reasons.
+    # Check for EOF to avoid this.
+    if not input.read(1):
+        return None
+    else:
+        input.seek(-1, 1)
+    return tnetstring.load(input)
+
+
+@click.command()
+@click.argument("input", type=click.File('rb'))
+def inspect(input):
+    """
+    pretty-print a dumpfile
+    """
+    while True:
+        data = read_tnetstring(input)
+        if not data:
+            break
+        pprint(data)
+
+
+if __name__ == "__main__":
+    inspect()