import tracing

* add argument to parser
* add out_path as fn parameter
* set necessary env variables for import tracing

2 files changed, 25 insertions, 0 deletions

diff --git a/library/python/import_tracing/constructor/__init__.py b/library/python/import_tracing/constructor/__init__.py
new file mode 100644
index 0000000000..47ae8527d4
--- /dev/null
+++ b/library/python/import_tracing/constructor/__init__.py
@@ -0,0 +1,10 @@
+import os
+
+
+def init():
+    if "Y_PYTHON_TRACE_FILE" in os.environ:
+        import atexit
+        import library.python.import_tracing.lib.regulator as regulator
+
+        regulator.enable(os.getenv("Y_PYTHON_TRACE_FILE"))
+        atexit.register(regulator.disable)
diff --git a/library/python/import_tracing/constructor/ya.make b/library/python/import_tracing/constructor/ya.make
new file mode 100644
index 0000000000..8ba6403897
--- /dev/null
+++ b/library/python/import_tracing/constructor/ya.make
@@ -0,0 +1,15 @@
+PY23_LIBRARY()
+
+STYLE_PYTHON()
+
+PY_CONSTRUCTOR(library.python.import_tracing.constructor)
+
+PY_SRCS(
+    __init__.py
+)
+
+PEERDIR(
+    library/python/import_tracing/lib
+)
+
+END()