from hashlib import sha1 import os from caliendo import config if config.should_use_caliendo(): from caliendo.db.flatfiles import insert_test, select_test __counters = { } def get_from_trace_for_ev(trace): if os.environ.get('CALIENDO_DISABLE_EV_COUNTER', False) == 'True': return 0 return get_from_trace(trace) def get_from_trace_for_cache(trace): if os.environ.get('CALIENDO_DISABLE_CACHE_COUNTER', False) == 'True': return 0 return get_from_trace(trace) def get_from_trace(trace): key = sha1( trace ).hexdigest() if key in __counters: t = __counters[ key ] __counters[ key ] = t + 1 return t else: t = __get_seed_from_trace( trace ) if not t: t = __set_seed_by_trace( trace )
import os import random as random_r import time from caliendo.config import should_use_caliendo from caliendo import util from caliendo.db import flatfiles as ff USE_CALIENDO = should_use_caliendo() if USE_CALIENDO: util.create_tables( ) def seq(): return util.seq() def random(): return util.random() else: counter = int( time.time() * 10000 ) def seq(): global counter c = counter counter = counter + 1 return c def random(*args): return random_r.random() def recache(*args, **kwargs): return util.recache(*args, **kwargs) class UNDEFINED(object):
import cPickle as pickle import sys import copy_reg import types import inspect from hashlib import sha1 import caliendo from caliendo import config from caliendo import pickling from caliendo import util from caliendo import prompt from caliendo import counter if config.should_use_caliendo(): from caliendo.db.flatfiles import select_expected_value, delete_expected_value, insert_expected_value def get_or_store(observed_value): caller = inspect.stack()[2][3] trace_string = util.get_stack(caller) counter_value = counter.get_from_trace_for_ev(trace_string) call_hash = sha1("%s.%s" % (trace_string, counter_value)).hexdigest() ev = fetch(call_hash) if not ev or (config.CALIENDO_PROMPT and prompt.should_modify_expected_value(caller)): ev = ExpectedValue(call_hash=call_hash, expected_value=prompt.modify_expected_value( observed_value, caller)) ev.save()
import os import sys import inspect import importlib import types from contextlib import contextmanager import caliendo from caliendo import util from caliendo import config from caliendo import call_descriptor from caliendo import counter from caliendo import prompt from caliendo.__init__ import UNDEFINED from caliendo.hooks import Hook USE_CALIENDO = config.should_use_caliendo() if USE_CALIENDO: from caliendo.db.flatfiles import delete_io def should_exclude(type_or_instance, exclusion_list): """ Tests whether an object should be simply returned when being wrapped """ if type_or_instance in exclusion_list: # Check class definition return True if type(type_or_instance) in exclusion_list: # Check instance type return True try: if type_or_instance.__class__ in exclusion_list: # Check instance class
import os import sys import inspect import importlib import types from contextlib import contextmanager import caliendo from caliendo import util from caliendo import config from caliendo import call_descriptor from caliendo import counter from caliendo import prompt from caliendo import UNDEFINED from caliendo.hooks import Hook USE_CALIENDO = config.should_use_caliendo() if USE_CALIENDO: from caliendo.db.flatfiles import delete_io def should_exclude(type_or_instance, exclusion_list): """ Tests whether an object should be simply returned when being wrapped """ if type_or_instance in exclusion_list: # Check class definition return True if type(type_or_instance) in exclusion_list: # Check instance type return True try: if type_or_instance.__class__ in exclusion_list: # Check instance class