Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-10 09:35:34 +00:00
Code Releases Activity

Style changes in vcr/config.py

Browse Source
This commit is contained in:
Ivan Malison
2015-08-22 18:48:04 -07:00
parent 29d37e410a
commit be3bf39161
1 changed files with 25 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
vcr/config.py
View File

@@ -124,7 +124,9 @@ class VCR(object):
if not path.startswith(cassette_library_dir): if not path.startswith(cassette_library_dir):
return os.path.join(cassette_library_dir, path) return os.path.join(cassette_library_dir, path)
return path return path
path_transformer = compose(add_cassette_library_dir, path_transformer) path_transformer = compose(
add_cassette_library_dir, path_transformer
)
elif not func_path_generator: elif not func_path_generator:
# If we don't have a library dir, use the functions # If we don't have a library dir, use the functions
# location to build a full path for cassettes. # location to build a full path for cassettes.
@@ -137,9 +139,7 @@ class VCR(object):
), ),
'record_mode': kwargs.get('record_mode', self.record_mode), 'record_mode': kwargs.get('record_mode', self.record_mode),
'before_record_request': self._build_before_record_request(kwargs), 'before_record_request': self._build_before_record_request(kwargs),
'before_record_response': self._build_before_record_response( 'before_record_response': self._build_before_record_response(kwargs),
kwargs
),
'custom_patches': self._custom_patches + kwargs.get( 'custom_patches': self._custom_patches + kwargs.get(
'custom_patches', () 'custom_patches', ()
), ),
@@ -162,6 +162,7 @@ class VCR(object):
before_record_response = (before_record_response,) before_record_response = (before_record_response,)
for function in before_record_response: for function in before_record_response:
filter_functions.append(function) filter_functions.append(function)
def before_record_response(response): def before_record_response(response):
for function in filter_functions: for function in filter_functions:
if response is None: if response is None:
@@ -182,7 +183,8 @@ class VCR(object):
'filter_post_data_parameters', self.filter_post_data_parameters 'filter_post_data_parameters', self.filter_post_data_parameters
) )
before_record_request = options.get( before_record_request = options.get(
"before_record_request", options.get("before_record", self.before_record_request) "before_record_request",
options.get("before_record", self.before_record_request)
) )
ignore_hosts = options.get( ignore_hosts = options.get(
'ignore_hosts', self.ignore_hosts 'ignore_hosts', self.ignore_hosts
@@ -191,14 +193,24 @@ class VCR(object):
'ignore_localhost', self.ignore_localhost 'ignore_localhost', self.ignore_localhost
) )
if filter_headers: if filter_headers:
filter_functions.append(functools.partial(filters.remove_headers, filter_functions.append(
headers_to_remove=filter_headers)) functools.partial(
filters.remove_headers,
headers_to_remove=filter_headers
)
)
if filter_query_parameters: if filter_query_parameters:
filter_functions.append(functools.partial(filters.remove_query_parameters, filter_functions.append(functools.partial(
query_parameters_to_remove=filter_query_parameters)) filters.remove_query_parameters,
query_parameters_to_remove=filter_query_parameters
))
if filter_post_data_parameters: if filter_post_data_parameters:
filter_functions.append(functools.partial(filters.remove_post_data_parameters, filter_functions.append(
post_data_parameters_to_remove=filter_post_data_parameters)) functools.partial(
filters.remove_post_data_parameters,
post_data_parameters_to_remove=filter_post_data_parameters
)
)
hosts_to_ignore = list(ignore_hosts) hosts_to_ignore = list(ignore_hosts)
if ignore_localhost: if ignore_localhost:
@@ -213,6 +225,7 @@ class VCR(object):
before_record_request = (before_record_request,) before_record_request = (before_record_request,)
for function in before_record_request: for function in before_record_request:
filter_functions.append(function) filter_functions.append(function)
def before_record_request(request): def before_record_request(request):
request = copy.copy(request) request = copy.copy(request)
for function in filter_functions: for function in filter_functions:
@@ -234,7 +247,7 @@ class VCR(object):
@staticmethod @staticmethod
def _build_path_from_func_using_module(function): def _build_path_from_func_using_module(function):
return os.path.join(os.path.dirname(inspect.getfile(function)), return os.path.join(os.path.dirname(inspect.getfile(function)),
function.__name__) function.__name__)
def register_serializer(self, name, serializer): def register_serializer(self, name, serializer):
self.serializers[name] = serializer self.serializers[name] = serializer