about summary refs log tree commit diff stats
path: root/tools/capture_transforms.py
diff options
context:
space:
mode:
Diffstat (limited to 'tools/capture_transforms.py')
-rwxr-xr-xtools/capture_transforms.py11
1 files changed, 3 insertions, 8 deletions
diff --git a/tools/capture_transforms.py b/tools/capture_transforms.py
index 5a104c0..552b855 100755
--- a/tools/capture_transforms.py
+++ b/tools/capture_transforms.py
@@ -1,12 +1,10 @@
 #!/usr/bin/env python3
 
 import argparse
-import logging
-import os
 
-from focaccia import parser
+from focaccia import parser, utils
 from focaccia.symbolic import collect_symbolic_trace
-from focaccia.trace import Trace, TraceEnvironment
+from focaccia.trace import TraceEnvironment
 
 def main():
     prog = argparse.ArgumentParser()
@@ -20,10 +18,7 @@ def main():
                       help='Name of output file. (default: trace.out)')
     args = prog.parse_args()
 
-    logging.disable(logging.CRITICAL)
-    env = TraceEnvironment(args.binary,
-                           args.args,
-                           [f'{k}={v}' for k, v in os.environ.items()])
+    env = TraceEnvironment(args.binary, args.args, utils.get_envp())
     trace = collect_symbolic_trace(env, None)
     with open(args.output, 'w') as file:
         parser.serialize_transformations(trace, file)