def placebo_session(function):
 @functools.wraps(function)
 def wrapper(*args, **kwargs):
  session_kwargs = {'region_name': os.environ.get('AWS_DEFAULT_REGION', 'us-east-1')}
  profile_name = os.environ.get('PLACEBO_PROFILE', None)
  if profile_name:
   session_kwargs['profile_name'] = profile_name
  session = boto3.Session(**session_kwargs)
  self = args[0]
  prefix = ((self.__class__.__name__ + '.') + function.__name__)
  record_dir = os.path.join(PLACEBO_DIR, prefix)
  if (not os.path.exists(record_dir)):
   os.makedirs(record_dir)
  pill = placebo.attach(session, data_path=record_dir)
  if (os.environ.get('PLACEBO_MODE') == 'record'):
   pill.record()
  else:
   pill.playback()
  kwargs['session'] = session
  return function(*args, **kwargs)
 return wrapper

def placebo_session ( function ) : @ functools . wraps ( function ) def wrapper ( * args , ** kwargs ) : session_kwargs = { 'region_name' : os . environ . get ( 'AWS_DEFAULT_REGION' , 'us-east-1' ) } profile_name = os . environ . get ( 'PLACEBO_PROFILE' , None ) if profile_name : session_kwargs [ 'profile_name' ] = profile_name session = boto3 . Session ( ** session_kwargs ) self = args [ 0 ] prefix = ( ( self . __class__ . __name__ + '.' ) + function . __name__ ) record_dir = os . path . join ( PLACEBO_DIR , prefix ) if ( not os . path . exists ( record_dir ) ) : os . makedirs ( record_dir ) pill = placebo . attach ( session , data_path = record_dir ) if ( os . environ . get ( 'PLACEBO_MODE' ) == 'record' ) : pill . record ( ) else : pill . playback ( ) kwargs [ 'session' ] = session return function ( * args , ** kwargs ) return wrapper
decorator to help do testing with placebo .