Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 08:11:00 +00:00
Code Issues Projects Releases Wiki Activity

[web] Update OptManager.save() to suit the need of mitmweb.

Browse Source
This commit is contained in:
Matthew Shao 2017-07-24 20:30:14 +08:00
parent 93cd1562de
commit 6c4dbcc7a7
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mitmproxy/optmanager.py
View File

@ -512,13 +512,15 @@ def serialize(opts, text, defaults=False):
return ruamel.yaml.round_trip_dump(data) return ruamel.yaml.round_trip_dump(data)
def save(opts, path, defaults=False): def save(opts, path=None, defaults=False):
""" """
Save to path. If the destination file exists, modify it in-place. If the path is given, save to path, otherwise return the serialized data.
If the destination file exists, modify it in-place.
Raises OptionsError if the existing data is corrupt. Raises OptionsError if the existing data is corrupt.
""" """
if os.path.exists(path) and os.path.isfile(path): if path and os.path.exists(path) and os.path.isfile(path):
with open(path, "rt", encoding="utf8") as f: with open(path, "rt", encoding="utf8") as f:
try: try:
data = f.read() data = f.read()
@ -529,5 +531,9 @@ def save(opts, path, defaults=False):
else: else:
data = "" data = ""
data = serialize(opts, data, defaults) data = serialize(opts, data, defaults)
if path:
with open(path, "wt", encoding="utf8") as f: with open(path, "wt", encoding="utf8") as f:
f.write(data) f.write(data)
else:
return data