import threading import os import re import importlib from .runtime import runtime_info if runtime_info.GEVENT: import gevent class FrameCache(object): MAX_CACHE_SIZE = 2500 def __init__(self, agent): self.agent = agent self.agent_frame_cache = None self.system_frame_cache = None self.include_agent_frames = None self.include_system_frames = None self.agent_dir = os.path.dirname(os.path.realpath(__file__)) self.system_dir = os.path.dirname(os.path.realpath(threading.__file__)) if runtime_info.GEVENT: self.gevent_dir = os.path.dirname(os.path.realpath(gevent.__file__)) def start(self): self.agent_frame_cache = dict() self.system_frame_cache = dict() self.include_agent_frames = self.agent.get_option('include_agent_frames') self.include_system_frames = self.agent.get_option('include_system_frames') def stop(self): pass def is_agent_frame(self, filename): if filename in self.agent_frame_cache: return self.agent_frame_cache[filename] agent_frame = False if not self.include_agent_frames: if filename.startswith(self.agent_dir): agent_frame = True if len(self.agent_frame_cache) < self.MAX_CACHE_SIZE: self.agent_frame_cache[filename] = agent_frame return agent_frame def is_system_frame(self, filename): if filename in self.system_frame_cache: return self.system_frame_cache[filename] system_frame = False if not self.include_system_frames: if (filename.startswith(self.system_dir) or (runtime_info.GEVENT and filename.startswith(self.gevent_dir))): system_frame = True if len(self.system_frame_cache) < self.MAX_CACHE_SIZE: self.system_frame_cache[filename] = system_frame return system_frame