viztracer 1.1.0__cp312-cp312-macosx_11_0_x86_64.whl → 1.1.1__cp312-cp312-macosx_11_0_x86_64.whl

viztracer/__init__.py

@@ -1,7 +1,7 @@
 # Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
 # For details: https://github.com/gaogaotiantian/viztracer/blob/master/NOTICE.txt
 
-__version__ = "1.1.0"
+__version__ = "1.1.1"
 
 
 from .cellmagic import load_ipython_extension

viztracer/main.py

@@ -228,6 +228,8 @@ class VizUI:
                 exec_name = command[0]
             self.ofile = unique_file_name(exec_name)
         if options.output_file:
+            if not options.compress and not options.output_file.endswith((".json", ".html", ".gz")):
+                return False, "Only html, json and gz are supported"
             self.ofile = options.output_file
         elif options.pid_suffix:
             self.ofile = "result.json"

viztracer/modules/eventnode.c

@@ -130,12 +130,24 @@ fprintfeename(FILE* fptr, struct EventNode* node, uint8_t sanitize_function_name
 {
     if (node->data.fee.type == PyTrace_CALL || node->data.fee.type == PyTrace_RETURN) {
 #if PY_VERSION_HEX >= 0x030B0000
+    if (PyUnicode_Check(node->data.fee.code->co_qualname)) {
         fputs(PyUnicode_AsUTF8(node->data.fee.code->co_qualname), fptr);
+    } else {
+        fputs("<unknown>", fptr);
+    }
 #else
+    if (PyUnicode_Check(node->data.fee.code->co_name)) {
         fputs(PyUnicode_AsUTF8(node->data.fee.code->co_name), fptr);
+    } else {
+        fputs("<unknown>", fptr);
+    }
 #endif
         fputs(" (", fptr);
-        fputs_escape(PyUnicode_AsUTF8(node->data.fee.code->co_filename), fptr);
+        if (PyUnicode_Check(node->data.fee.code->co_filename)) {
+            fputs_escape(PyUnicode_AsUTF8(node->data.fee.code->co_filename), fptr);
+        } else {
+            fputs("<unknown>", fptr);
+        }
         fprintf(fptr, ":%d)", node->data.fee.code->co_firstlineno);
     } else {
         const char* ml_name = node->data.fee.ml_name;
@@ -152,7 +164,11 @@ fprintfeename(FILE* fptr, struct EventNode* node, uint8_t sanitize_function_name
         }
         if (node->data.fee.m_module) {
             // The function belongs to a module
-            fputs(PyUnicode_AsUTF8(node->data.fee.m_module), fptr);
+            if (PyUnicode_Check(node->data.fee.m_module)) {
+                fputs(PyUnicode_AsUTF8(node->data.fee.m_module), fptr);
+            } else {
+                fputs("<unknown>", fptr);
+            }
             fputc('.', fptr);
         } else {
             // The function is a class method

viztracer/modules/snaptrace.c

@@ -526,8 +526,9 @@ tracer_pyreturn_callback(TracerObject* self, PyCodeObject* code, PyObject* arg)
 
             if (!PyCode_Check(call_code) || call_code != code) {
                 self->collecting = 0;
-                PyErr_SetString(PyExc_RuntimeError, "VizTracer: Unexpected type. Might be an event mismatch.");
-                return -1;
+                PyErr_WarnEx(PyExc_RuntimeWarning,
+                    "VizTracer: Unexpected function return, tracing is stopped", 1);
+                return 0;
             }
 
             struct EventNode* node = get_next_node(self);
@@ -615,8 +616,9 @@ tracer_creturn_callback(TracerObject* self, PyCodeObject* code, PyObject* arg)
 
             if (!PyCFunction_Check(cfunc)) {
                 self->collecting = 0;
-                PyErr_SetString(PyExc_RuntimeError, "VizTracer: Unexpected type. Might be an event mismatch.");
-                return -1;
+                PyErr_WarnEx(PyExc_RuntimeWarning,
+                    "VizTracer: Unexpected function return, tracing is stopped", 1);
+                return 0;
             }
 
             struct EventNode* node = get_next_node(self);

viztracer/patch.py

@@ -53,6 +53,10 @@ def patch_subprocess(viz_args: list[str]) -> None:
                     # -m mod | -mmod
                     cm_arg = cm_arg or next(args_iter, None)
                     if cm_arg is not None:
+                        if cm_arg.split('.')[0] == "viztracer":
+                            # Avoid tracing viztracer subprocess
+                            # This is mainly used to avoid tracing --open
+                            return None
                         mode.append(cm_arg)
                     else:
                         mode = None
@@ -60,6 +64,12 @@ def patch_subprocess(viz_args: list[str]) -> None:
 
             # -pyopts
             py_args.append(arg)
+            if arg in ("-X", "-W", "--check-hash-based-pycs"):
+                arg_next = next(args_iter, None)
+                if arg_next is not None:
+                    py_args.append(arg_next)
+                else:
+                    return None
 
         if script:
             return [sys.executable, *py_args, "-m", "viztracer", "--quiet", *viz_args, "--", script, *args_iter]

viztracer/viewer.py

@@ -88,6 +88,8 @@ class PerfettoHandler(HttpHandler):
             self.end_headers()
             self.wfile.write(json.dumps(self.server_thread.file_info).encode("utf-8"))
             self.wfile.flush()
+            # Since v1.1, file_info is the last request from the frontend
+            self.server.trace_served = True
         elif self.path.endswith("localtrace"):
             # self.directory is used after 3.8
             # os.getcwd() is used on 3.6
@@ -95,7 +97,6 @@ class PerfettoHandler(HttpHandler):
             with chdir_temp(self.directory):
                 filename = os.path.basename(self.server_thread.path)
                 self.path = f"/{filename}"
-                self.server.trace_served = True
                 return super().do_GET()
         else:
             self.directory = os.path.join(os.path.dirname(__file__), "web_dist")

{viztracer-1.1.0.dist-info → viztracer-1.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: viztracer
-Version: 1.1.0
+Version: 1.1.1
 Summary: A debugging and profiling tool that can trace and visualize python code execution
 Author-email: Tian Gao <gaogaotiantian@hotmail.com>
 License-Expression: Apache-2.0
@@ -301,6 +301,16 @@ normal sampling profilers.
 
 </details>
 
+## Sponsors
+
+We thank our sponsors to support VizTracer. If you want your logo here,
+check our [sponsor page](https://github.com/sponsors/gaogaotiantian) or contact
+the [author](https://github.com/gaogaotiantian) directly.
+
+<a href="https://www.lambdatest.com/?utm_source=viztracer&utm_medium=sponsor" target="_blank">
+    <img src="https://www.lambdatest.com/blue-logo.png" style="vertical-align: middle;" width="250" height="45" />
+</a>
+
 ## Documentation
 
 For full documentation, please see [https://viztracer.readthedocs.io/en/stable](https://viztracer.readthedocs.io/en/stable)

{viztracer-1.1.0.dist-info → viztracer-1.1.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 viztracer/report_builder.py,sha256=CGQ0czLrPDDh6m6MFGArRM1Ewnpbm1KQL2iuUYDLwFA,14186
-viztracer/patch.py,sha256=1KYYrgdDbVfiUux_lv6a-4uCQszS-AdQlhL2u5KD2X4,12460
-viztracer/snaptrace.cpython-312-darwin.so,sha256=MiCy-HCHHUhPZFZhGAqkSIJ_LLivE311Kjs_rUYyD-I,85328
-viztracer/viewer.py,sha256=A_ZUybPYmEgL0aLIqmj4sR4xAtjwhW-bLz0r6CuJfbE,19682
+viztracer/patch.py,sha256=zxSKnv9Wr7p8NVUHy46mIBDNRARvGwnfqMTAsJ6VidU,12955
+viztracer/snaptrace.cpython-312-darwin.so,sha256=quhf2R5HnZ1kSX6y17YXz_034bn5UWLqhTEAPtubQ58,60816
+viztracer/viewer.py,sha256=o2tLt6_xN5GvddXYgeyN2VAv1-9DWVS37yHihMYfC3Q,19752
 viztracer/vizevent.py,sha256=5fKdUp1RDx5SescpffDNvDaV2SaF8H1WgNuvk2ybujY,971
 viztracer/vizplugin.py,sha256=SzoA3skl7cLtgJDtMogUzqn7Yb-C8HGZ1La_jqooEVI,5339
 viztracer/util.py,sha256=s4T0h8jA3Ny4QcHDXbMPGGRNUXXyAcYqLgPR0bk8yUo,6100
@@ -9,7 +9,7 @@ viztracer/decorator.py,sha256=XhNMxdMM1qs5W22B5sBIbOWtZahHYTLOc0TKtTZhxH4,5699
 viztracer/snaptrace.pyi,sha256=a7z19WnUa2MPFdEZOpd_fUHL7IizrP4P0uk8QFMuVhA,1793
 viztracer/attach.py,sha256=VsEE3G_mpqMkiGwWiV0kNPe7VMlG_cjiOr8opDRiUAI,1991
 viztracer/viztracer.py,sha256=RcZ59vkeOwYHcjuaum8R7eObH1dTT8WYrDGEImVi558,17851
-viztracer/__init__.py,sha256=Apna-Ca9rtndiVHRUy54Qc8UWigwbGbE9HMhl9dzti8,507
+viztracer/__init__.py,sha256=fdwl94zpBuVrw1_ztf57BOru3Le3Gg_LLF045A564O4,507
 viztracer/vizcounter.py,sha256=aqTTg0FxYE82fn__TXF_Mk3nU9Sbn5N3P1AdlSwV7bg,793
 viztracer/functree.py,sha256=KKkGrqf5L4L4c2PC_zX6QVqwlF8SxRID4UOZaI5wOaI,5188
 viztracer/vizlogging.py,sha256=cyrkW2sAWvG5z1n-OGI_eZMgS1N9ADFkqapHu6XWamM,668
@@ -17,8 +17,8 @@ viztracer/vizobject.py,sha256=vBoPBIzCIYb6yCrLTzsSdPNAtN3fprD1Yd7a36Tlccs,1240
 viztracer/vcompressor.pyi,sha256=6F6W8rREivfR1G4WI20xd50mSoayX9DkEf2Wi43-k-A,320
 viztracer/cellmagic.py,sha256=SLibORC1-Rjk2a66IIMMuV-Q6Rotg-4FaawzGH0oWIM,3365
 viztracer/event_base.py,sha256=4embca1Ixlb_baUZD3MfFTvuKoy7y_zhHI9uq3eomL8,3130
-viztracer/main.py,sha256=cY8wj0IDnWBC3af7TEqHz9d2CQCt7YV1VUFWYKyBrtI,31837
-viztracer/vcompressor.cpython-312-darwin.so,sha256=RnPclTyQLPBKLbX0dDzyMaM2e2sZMtsjE1Af5W7OPXc,73656
+viztracer/main.py,sha256=NMhSVY0SvslqEiChFs1ZmMIHNdYqCxGUn3K2suVXPgY,32007
+viztracer/vcompressor.cpython-312-darwin.so,sha256=6us7cWY_nYSnGe7ReW59xrSsE2B2GwR0mZTd3CElSjM,36792
 viztracer/__main__.py,sha256=-hhsTtiwE9cbgOj7DJGOaTJC8URmcRdicHRMSp4SnPo,226
 viztracer/code_monkey.py,sha256=jJrV4OWgYaCZyMe6Jk4jP8S6ck6pG5gBDZ_dxeXscGU,15279
 viztracer/web_dist/index.html,sha256=JwShy-t2ZMoxepYiNy6P9zlqnDznzmKjTHqL4R4RHGw,5790
@@ -83,13 +83,13 @@ viztracer/attach_process/linux_and_mac/lldb_prepare.py,sha256=h8dcRDjmohAxrWPpq5
 viztracer/html/flamegraph.html,sha256=BxbaPmmacZHdFXXD3LwTu0-IlE5ui3Y6s7fQbaPTY1k,1663
 viztracer/html/trace_viewer_embedder.html,sha256=hkY7VyVe6pq5syldbwRsqEW2wm9wgdnrHembwm7mC-o,6734
 viztracer/html/trace_viewer_full.html,sha256=h1TCFm2omE61KE0n-lovJcVVmxvGAIFxjJbI6yiJGjM,2598333
-viztracer/modules/snaptrace.c,sha256=kNZ2uBGHYgs8owQpT7jSbyCH6puZHtr6dSlCPA51ErM,69705
+viztracer/modules/snaptrace.c,sha256=J-mcQSg5q_050iByxiY1IZKB9uNx7JKhUiXkDADhKOQ,69751
 viztracer/modules/util.c,sha256=rFvn_9azPipunH4yEX0ALRluljvS2pg-E0oWfva2rps,1303
 viztracer/modules/eventnode.h,sha256=hQ9AoNjHLZqYlI3R2BoCITjFVw16XSrbFefx1lV0ews,1541
 viztracer/modules/pythoncapi_compat.h,sha256=9Fk7ww3THeLnoOmpEUnh0C86ZHD2Tfe4kDA7oDlahOc,46185
 viztracer/modules/quicktime.c,sha256=iT3AkQw2pxPOzVhY1OlWEStavSsY_j_kDwR3SyDCTSk,4102
 viztracer/modules/snaptrace.h,sha256=Vf21wB9-rhfgKzB-lDwvGlxYyT0hDuJ7lTtlg0X0NoA,4016
-viztracer/modules/eventnode.c,sha256=hIdE7fCW_BqwDE6zulfPC_dXpxLPSxkxH1-PXph-mYU,5711
+viztracer/modules/eventnode.c,sha256=zacMDr3kR9n3qYOgT-HdY3w7eO2xLR7oMPGCx2gA7Gg,6210
 viztracer/modules/snaptrace_member.c,sha256=mkAN7xibIplRshyEXubKYT3ZFd2Leg93OEeVxtfhDp0,13757
 viztracer/modules/util.h,sha256=x5V0b71CjSD2HLjW513mMM1Hf1pxzBiD-0G2i477cFM,586
 viztracer/modules/quicktime.h,sha256=Aooi4aRl3awX9BoU99FHnvF0MMc7YxGulbMacsIbzg0,2354
@@ -97,10 +97,10 @@ viztracer/modules/vcompressor/vcompressor.h,sha256=ZV6Z1N_C8gsmXH7J6vIZm6EXcsrwd
 viztracer/modules/vcompressor/vc_dump.c,sha256=K9_MVaWp9VORCjKcEKEpmDPvrWeo6a8iWjahGDZI5RA,39337
 viztracer/modules/vcompressor/vcompressor.c,sha256=6KgwtIhcSQwQei7nmwvdxbcxKn7z_7hqlT7ORGScKhw,12664
 viztracer/modules/vcompressor/vc_dump.h,sha256=gxFuI3Pm3DPXHvO00XF5QJ8AT9PeAtHo7ojwUgvO8Lk,1421
-viztracer-1.1.0.dist-info/RECORD,,
-viztracer-1.1.0.dist-info/WHEEL,sha256=MdxgY50Fc-IVjK3WCRGjXjKIDI1Of7db2_McocDAT3I,137
-viztracer-1.1.0.dist-info/entry_points.txt,sha256=-MeyWMzg_Ci74E8Y3dr_pjzZlYWq3nQYTA41OQcrZ48,91
-viztracer-1.1.0.dist-info/top_level.txt,sha256=xph8Zo9F28dycmRhR2cr64PLrKzc_ulOQ_lzqT3Di28,10
-viztracer-1.1.0.dist-info/METADATA,sha256=J8n31MotIp1WyrHEB3pMNHI_RguMH1yWmIMCUgVNS5w,13112
-viztracer-1.1.0.dist-info/licenses/LICENSE,sha256=pvJg_eSaSgdfGDmIDrCrtrz-nU5svUG8y1u1AafTWt8,11935
-viztracer-1.1.0.dist-info/licenses/NOTICE.txt,sha256=0TrouYbPH7Zwnrq2k2H6PfRyQJWO4aUew3685aqraZQ,1031
+viztracer-1.1.1.dist-info/RECORD,,
+viztracer-1.1.1.dist-info/WHEEL,sha256=MdxgY50Fc-IVjK3WCRGjXjKIDI1Of7db2_McocDAT3I,137
+viztracer-1.1.1.dist-info/entry_points.txt,sha256=-MeyWMzg_Ci74E8Y3dr_pjzZlYWq3nQYTA41OQcrZ48,91
+viztracer-1.1.1.dist-info/top_level.txt,sha256=xph8Zo9F28dycmRhR2cr64PLrKzc_ulOQ_lzqT3Di28,10
+viztracer-1.1.1.dist-info/METADATA,sha256=LV4HSB8SKh2OUYJlK4pAVG7eZaGlGTzxYcHLH9HY_N4,13552
+viztracer-1.1.1.dist-info/licenses/LICENSE,sha256=pvJg_eSaSgdfGDmIDrCrtrz-nU5svUG8y1u1AafTWt8,11935
+viztracer-1.1.1.dist-info/licenses/NOTICE.txt,sha256=0TrouYbPH7Zwnrq2k2H6PfRyQJWO4aUew3685aqraZQ,1031