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

Updated migration to support default ports

Browse Source
This commit is contained in:
Max Shytikov
2014-05-04 02:05:55 +02:00
parent 0b1aeac25e
commit 5d10a38160
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
vcr/migration.py
View File

@@ -38,7 +38,11 @@ PARTS = [
def build_uri(**parts): def build_uri(**parts):
return "{protocol}://{host}:{port}{path}".format(**parts) port = parts['port']
scheme = parts['protocol']
default_port = {'https': 433, 'http': 80}[scheme]
parts['port'] = ':{0}'.format(port) if port != default_port else ''
return "{protocol}://{host}{port}{path}".format(**parts)
def migrate_json(in_fp, out_fp): def migrate_json(in_fp, out_fp):
@@ -84,11 +88,11 @@ def migrate_yml(in_fp, out_fp):
if not isinstance(req, request.Request): if not isinstance(req, request.Request):
raise Exception("already migrated") raise Exception("already migrated")
else: else:
req.uri = "{0}://{1}:{2}{3}".format( req.uri = build_uri(
req.__dict__['protocol'], protocol=req.__dict__['protocol'],
req.__dict__['host'], host=req.__dict__['host'],
req.__dict__['port'], port=req.__dict__['port'],
req.__dict__['path'], path=req.__dict__['path'],
) )
# convert headers to dict of lists # convert headers to dict of lists