austin-python 2.1.1__tar.gz → 2.1.2__tar.gz

{austin_python-2.1.1 → austin_python-2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: austin-python
-Version: 2.1.1
+Version: 2.1.2
 Summary: Python wrapper for Austin, the CPython frame stack sampler.
 Project-URL: homepage, https://github.com/P403n1x87/austin-python
 Project-URL: documentation, https://austin-python.readthedocs.io

{austin_python-2.1.1 → austin_python-2.1.2}/austin/events.py

@@ -1,10 +1,9 @@
 import typing as t
-from collections import namedtuple
 from dataclasses import dataclass
 
 
-ThreadName = namedtuple("ThreadName", ["thread", "iid"])
 ProcessId = int
+ThreadName = str
 InterpreterId = int
 MicroSeconds = int
 Bytes = int

{austin_python-2.1.1 → austin_python-2.1.2}/austin/format/mojo.py

@@ -277,7 +277,12 @@ def str_reader() -> t.Generator[t.Optional[str], bytes, str]:
             (b,) = yield bytes(buffer[:i]).decode(errors="replace")
             i = 0
         else:
-            buffer[i] = b
+            try:
+                buffer[i] = b
+            except IndexError:
+                # Expand the buffer if needed
+                buffer += bytearray(i - len(buffer) + 1)
+                buffer[i] = b
             (b,) = yield None
             i += 1
 
@@ -353,6 +358,14 @@ class BaseMojoStreamReader(AustinEventIterator):
 
         return metadata
 
+    @property
+    def mode(self) -> t.Optional[str]:
+        return self.metadata.get("mode")
+
+    @property
+    def gc(self) -> t.Optional[str]:
+        return self.metadata.get("gc")
+
     def _finalize_sample(self) -> AustinSample:
         """Finalize the current sample."""
         assert self._running_sample is not None, self._running_sample
@@ -395,7 +408,13 @@ class BaseMojoStreamReader(AustinEventIterator):
         if self._running_sample is not None:
             self._finalize_sample()
 
-        self._running_sample = _RunningSample(pid=pid, iid=iid, thread=thread)
+        self._running_sample = _RunningSample(
+            pid=pid,
+            iid=iid,
+            thread=thread,
+            idle=False if self.mode == "full" else None,
+            gc=False if self.gc is not None else None,
+        )
 
         return MojoStack(pid, iid if iid is not None else -1, thread)
 

{austin_python-2.1.1 → austin_python-2.1.2}/austin/stats.py

@@ -21,6 +21,7 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+from collections import namedtuple
 from copy import deepcopy
 from dataclasses import dataclass
 from dataclasses import field
@@ -39,10 +40,12 @@ from austin.events import AustinMetadata
 from austin.events import AustinMetrics
 from austin.events import AustinSample
 from austin.events import ProcessId
-from austin.events import ThreadName
 from austin.format.collapsed_stack import AustinEventCollapsedStackFormatter
 
 
+ThreadInfo = namedtuple("ThreadInfo", ["thread", "iid"])
+
+
 COLLAPSED_STACK_FORMATTER = AustinEventCollapsedStackFormatter()
 
 
@@ -136,12 +139,12 @@ class FrameStats(HierarchicalStats):
 class ThreadStats(HierarchicalStats):
     """Thread statistics."""
 
-    label: ThreadName
+    label: ThreadInfo
     children: Dict[AustinFrame, FrameStats] = field(default_factory=dict)  # type: ignore[assignment]
 
     def collapse(
         self, stats_type: AustinStatsType
-    ) -> Generator[Tuple[ThreadName, List[AustinFrame], AustinMetrics], None, None]:
+    ) -> Generator[Tuple[ThreadInfo, List[AustinFrame], AustinMetrics], None, None]:
         """Collapse the hierarchical statistics."""
 
         if self.own:
@@ -166,7 +169,7 @@ class ProcessStats:
     """Process statistics."""
 
     pid: ProcessId
-    threads: Dict[ThreadName, ThreadStats] = field(default_factory=dict)
+    threads: Dict[ThreadInfo, ThreadStats] = field(default_factory=dict)
 
     def collapse(
         self, stats_type: AustinStatsType
@@ -175,22 +178,22 @@ class ProcessStats:
         yield from (
             AustinSample(
                 pid=self.pid,
-                iid=thread_name.iid,
-                thread=thread_name.thread,
+                iid=thread_info.iid,
+                thread=thread_info.thread,
                 frames=tuple(frames),
                 metrics=metric,
             )
             for thread in self.threads.values()
-            for thread_name, frames, metric in thread.collapse(stats_type)
+            for thread_info, frames, metric in thread.collapse(stats_type)
         )
 
-    def get_thread(self, thread_name: ThreadName) -> Optional[ThreadStats]:
+    def get_thread(self, thread_info: ThreadInfo) -> Optional[ThreadStats]:
         """Get thread statistics from this process by name.
 
         If the given thread name is not registered with this current process
         statistics, then ``None`` is returned.
         """
-        return self.threads.get(thread_name)
+        return self.threads.get(thread_info)
 
 
 @dataclass
@@ -286,8 +289,8 @@ class AustinStats:
             return
 
         pid = sample.pid
-        thread_name = ThreadName(sample.thread, sample.iid or 0)
-        thread_stats = ThreadStats(thread_name, own=0, total=metric)
+        thread_info = ThreadInfo(sample.thread, sample.iid or 0)
+        thread_stats = ThreadStats(thread_info, own=0, total=metric)
 
         # Convert the list of frames into a nested FrameStats instance
         stats: HierarchicalStats = thread_stats
@@ -300,15 +303,15 @@ class AustinStats:
 
         with self._lock:
             if pid not in self.processes:
-                self.processes[pid] = ProcessStats(pid, {thread_name: thread_stats})
+                self.processes[pid] = ProcessStats(pid, {thread_info: thread_stats})
                 return
 
             process = self.processes[pid]
-            if thread_name not in process.threads:
-                process.threads[thread_name] = thread_stats
+            if thread_info not in process.threads:
+                process.threads[thread_info] = thread_stats
                 return
 
-            process.threads[thread_name] << thread_stats
+            process.threads[thread_info] << thread_stats
 
     def flatten(self) -> Generator[AustinSample, None, None]:
         for process in self.processes.values():

{austin_python-2.1.1 → austin_python-2.1.2}/pyproject.toml

@@ -127,10 +127,10 @@ source = ["austin"]
 [tool.coverage.report]
 show_missing = true
 
-[tool.isort]
-force_single_line = true
-lines_after_imports = 2
-profile = "black"
+[tool.ruff.lint.isort]
+force-single-line = true
+lines-after-imports = 2
+force-sort-within-sections = true
 
 [tool.mypy]
 ignore_missing_imports = true

{austin_python-2.1.1 → austin_python-2.1.2}/test/stats/test_austin_stats.py

@@ -21,19 +21,19 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-import io
 from copy import deepcopy
+import io
 
 from austin.events import AustinFrame
 from austin.events import AustinMetrics
 from austin.events import AustinSample
-from austin.events import ThreadName
 from austin.format.collapsed_stack import AustinFileReader
 from austin.format.collapsed_stack import parse_collapsed_stack
 from austin.stats import AustinStats
 from austin.stats import AustinStatsType
 from austin.stats import FrameStats
 from austin.stats import ProcessStats
+from austin.stats import ThreadInfo
 from austin.stats import ThreadStats
 
 
@@ -54,8 +54,8 @@ def test_austin_stats_single_process():
             42: ProcessStats(
                 pid=42,
                 threads={
-                    ThreadName("0x7f45645646", 0): ThreadStats(
-                        label=ThreadName("0x7f45645646", 0),
+                    ThreadInfo("0x7f45645646", 0): ThreadStats(
+                        label=ThreadInfo("0x7f45645646", 0),
                         own=0,
                         total=152,
                         children={
@@ -82,8 +82,8 @@ def test_austin_stats_single_process():
             42: ProcessStats(
                 pid=42,
                 threads={
-                    ThreadName("0x7f45645646", 0): ThreadStats(
-                        label=ThreadName("0x7f45645646", 0),
+                    ThreadInfo("0x7f45645646", 0): ThreadStats(
+                        label=ThreadInfo("0x7f45645646", 0),
                         total=300,
                         own=148,
                         children={
@@ -110,8 +110,8 @@ def test_austin_stats_single_process():
             42: ProcessStats(
                 pid=42,
                 threads={
-                    ThreadName("0x7f45645646", 0): ThreadStats(
-                        label=ThreadName("0x7f45645646", 0),
+                    ThreadInfo("0x7f45645646", 0): ThreadStats(
+                        label=ThreadInfo("0x7f45645646", 0),
                         total=400,
                         own=148,
                         children={
@@ -138,8 +138,8 @@ def test_austin_stats_single_process():
             42: ProcessStats(
                 pid=42,
                 threads={
-                    ThreadName("0x7f45645646", 0): ThreadStats(
-                        label=ThreadName("0x7f45645646", 0),
+                    ThreadInfo("0x7f45645646", 0): ThreadStats(
+                        label=ThreadInfo("0x7f45645646", 0),
                         total=800,
                         own=148,
                         children={
@@ -175,8 +175,8 @@ def test_austin_stats_single_process():
             42: ProcessStats(
                 pid=42,
                 threads={
-                    ThreadName("0x7f45645664", 0): ThreadStats(
-                        label=ThreadName("0x7f45645664", 0),
+                    ThreadInfo("0x7f45645664", 0): ThreadStats(
+                        label=ThreadInfo("0x7f45645664", 0),
                         own=0,
                         total=152,
                         children={
@@ -191,8 +191,8 @@ def test_austin_stats_single_process():
                             )
                         },
                     ),
-                    ThreadName("0x7f45645646", 0): ThreadStats(
-                        label=ThreadName("0x7f45645646", 0),
+                    ThreadInfo("0x7f45645646", 0): ThreadStats(
+                        label=ThreadInfo("0x7f45645646", 0),
                         total=800,
                         own=148,
                         children={
@@ -302,8 +302,8 @@ def test_load():
             42: ProcessStats(
                 pid=42,
                 threads={
-                    ThreadName("0x7f45645646", 0): ThreadStats(
-                        label=ThreadName("0x7f45645646", 0),
+                    ThreadInfo("0x7f45645646", 0): ThreadStats(
+                        label=ThreadInfo("0x7f45645646", 0),
                         own=0,
                         total=1300,
                         children={

{austin_python-2.1.1 → austin_python-2.1.2}/austin/format/collapsed_stack.py

@@ -1,6 +1,6 @@
-import typing as t
 from functools import singledispatchmethod
 from pathlib import Path
+import typing as t
 
 from austin.events import AustinEvent
 from austin.events import AustinEventIterator