This commit is contained in:
Luis Aleixo 2024-02-12 12:10:33 +01:00
parent 00dc79f492
commit f9d2fc39ec

View file

@ -9,7 +9,7 @@ import typing
from datetime import datetime from datetime import datetime
from enum import Enum from enum import Enum
from functools import wraps from functools import wraps
from pstats import f8, func_std_string from pstats import f8, func_std_string # type: ignore
from pyinstrument import Profiler as PyInstrumentProfiler from pyinstrument import Profiler as PyInstrumentProfiler
@ -134,12 +134,12 @@ class Profiler:
class CaimiraProfiler: class CaimiraProfiler:
_cache_dirpath: str
_cache_filepath: str
CACHE_DIR: str = os.environ.get("CAIMIRA_PROFILER_CACHE_DIR", "/tmp") CACHE_DIR: str = os.environ.get("CAIMIRA_PROFILER_CACHE_DIR", "/tmp")
ROOT_URL: str = "/profiler" ROOT_URL: str = "/profiler"
_cache_dirpath: str = None
_cache_filepath: str = None
def __init__(self, cache_dir: typing.Optional[str] = None): def __init__(self):
_cache_dir = os.path.join(self.CACHE_DIR, "caimira-profiler") _cache_dir = os.path.join(self.CACHE_DIR, "caimira-profiler")
self._cache_dirpath = _cache_dir self._cache_dirpath = _cache_dir
self._cache_filepath = os.path.join(_cache_dir, "active") self._cache_filepath = os.path.join(_cache_dir, "active")
@ -209,7 +209,7 @@ class CaimiraProfiler:
def start_session( def start_session(
self, self,
name: typing.Optional[str] = None, name: typing.Optional[str] = None,
profiler_type: typing.Optional[Profilers] = Profilers.PYINSTRUMENT, profiler_type: Profilers = Profilers.PYINSTRUMENT,
): ):
"""Start a new session, assigning the given name.""" """Start a new session, assigning the given name."""
now = datetime.utcnow().isoformat() now = datetime.utcnow().isoformat()