Merge pull request #16 from jennier0107/dev

[ENH] Provides command line tool for pydumpling

Author: cocolato

.gitignore

@@ -5,4 +5,5 @@ __pycache__
 *.dump
 test.py
 /my
-.vscode
+.vscode
+.idea

dist/pydumpling-0.1.1-py2.py3-none-any.whl

@@ -0,0 +1,3 @@
+from .cli import main
+
+main()

dist/pydumpling-0.1.1.tar.gz

@@ -0,0 +1,53 @@
+import argparse
+import os.path
+from .debug_dumpling import debug_dumpling, load_dumpling
+from .helpers import print_traceback_and_except
+
+DUMP_FILE_EXTENSION: str = ".dump"
+
+
+def validate_file_name(file_name: str) -> str:
+    """check file extension name and exists"""
+    if not file_name.endswith(DUMP_FILE_EXTENSION):
+        raise argparse.ArgumentTypeError("File must be .dump file")
+    if not os.path.exists(file_name):
+        raise argparse.ArgumentTypeError(f"File {file_name}  not found")
+    return file_name
+
+
+parser = argparse.ArgumentParser(
+    description="pydumpling cli tools",
+    prog="pydumpling",
+    usage="%(prog)s [options] filename"
+)
+
+# print or debug
+pydumpling_cli_action_group = parser.add_mutually_exclusive_group(required=True)
+
+pydumpling_cli_action_group.add_argument(
+    "--print",
+    action="store_true",
+    help="print traceback information"
+)
+
+pydumpling_cli_action_group.add_argument(
+    "--debug",
+    action="store_true",
+    help="enter pdb debugging interface"
+)
+
+parser.add_argument(
+    "filename",
+    type=validate_file_name,
+    help="the .dump file"
+)
+
+
+def main() -> None:
+    args = parser.parse_args()
+    file_name = args.filename
+    if args.print:
+        dumpling_ = load_dumpling(file_name)
+        print_traceback_and_except(dumpling_)
+    elif args.debug:
+        debug_dumpling(file_name)

poetry.lock

@@ -0,0 +1,26 @@
+import linecache
+
+
+def print_traceback_and_except(dumpling_result):
+    traceback_obj = dumpling_result["traceback"]
+    except_extra = dumpling_result["exc_extra"]
+    except_type = except_extra["exc_type"]
+    except_value = except_extra["exc_value"]
+
+    print("Traceback (most recent call last):")
+    while traceback_obj:
+        frame = traceback_obj.tb_frame
+        lineno = traceback_obj.tb_lineno
+        filename = frame.f_code.co_filename
+        function_name = frame.f_code.co_name
+        line = linecache.getline(filename, lineno).strip()
+        mid_index = len(line) // 2
+
+        print(f"File \"{filename}\", line {lineno}, in {function_name}")
+        print(f"  {line}")
+        print(" " * 4 + "~" * mid_index + "^" + "~" * mid_index)
+        traceback_obj = traceback_obj.tb_next
+
+        if traceback_obj is None:
+            break
+    print(f"{except_type.__name__}: {except_value}")