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

Make migration python 2.6 compatible

Browse Source
This commit is contained in:
Max Shytikov
2014-04-22 23:42:12 +02:00
parent 424c658da4
commit 9d8426e668
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
vcr/migration.py
View File

@@ -37,7 +37,7 @@ def migrate_json(in_fp, out_fp):
data = json.load(in_fp) data = json.load(in_fp)
for item in data: for item in data:
req = item['request'] req = item['request']
uri = {k: req.pop(k) for k in PARTS} uri = dict((k, req.pop(k)) for k in PARTS)
req['uri'] = build_uri(**uri) req['uri'] = build_uri(**uri)
json.dump(data, out_fp, indent=4) json.dump(data, out_fp, indent=4)
@@ -47,7 +47,7 @@ def migrate_yml(in_fp, out_fp):
uri = dict.fromkeys(PARTS, None) uri = dict.fromkeys(PARTS, None)
for line in in_fp: for line in in_fp:
for part in uri: for part in uri:
match = re.match('\s+{}:\s(.*)'.format(part), line) match = re.match('\s+{0}:\s(.*)'.format(part), line)
if match: if match:
uri[part] = match.group(1) uri[part] = match.group(1)
break break
@@ -55,7 +55,7 @@ def migrate_yml(in_fp, out_fp):
out_fp.write(line) out_fp.write(line)
if None not in uri.values(): # if all uri parts are collected if None not in uri.values(): # if all uri parts are collected
out_fp.write(" uri: {}\n".format(build_uri(**uri))) out_fp.write(" uri: {0}\n".format(build_uri(**uri)))
uri = dict.fromkeys(PARTS, None) # reset dict uri = dict.fromkeys(PARTS, None) # reset dict
migrated = True migrated = True
if not migrated: if not migrated:
@@ -100,7 +100,7 @@ def main():
for file_path in files: for file_path in files:
migrated = try_migrate(file_path) migrated = try_migrate(file_path)
status = 'OK' if migrated else 'FAIL' status = 'OK' if migrated else 'FAIL'
sys.stderr.write("[{}] {}\n".format(status, file_path)) sys.stderr.write("[{0}] {1}\n".format(status, file_path))
sys.stderr.write("Done.\n") sys.stderr.write("Done.\n")
if __name__ == '__main__': if __name__ == '__main__':