feat: 默认从系统变量pycdc中读取路径

2024-06-04 17:47:25 +08:00 · 2024-06-04 17:47:25 +08:00 · 977841837d
commit 977841837d
parent 42135c516c
1 changed files with 4 additions and 1 deletions
--- a/detection/main.py
+++ b/detection/main.py
@ -420,7 +420,10 @@ def main():
        "-m", "--mode", help="Mode of operation:[regex,llm]", default="regex"
    )
    parser.add_argument(
-        "-p", "--pycdc", help="Path to pycdc.exe to decompile", default=None
+        "-p",
+        "--pycdc",
+        help="Path to pycdc.exe to decompile",
+        default=os.getenv("pycdc"),
    )
    args = parser.parse_args()
    output_format = "txt"  # Default output format