Mirrors/PySnooper
1
0
Fork 0
mirror of https://github.com/cool-RR/PySnooper.git synced 2026-01-23 10:15:11 +00:00
Code Issues Projects Releases Wiki Activity

Test for open attribute on output, to support Paths

Browse source
This commit is contained in:
Alex Hall 2019-05-03 20:00:22 +02:00 • committed by Ram Rachum
parent 2732922e1d
commit 5a82a65bde
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
pysnooper/pysnooper.py
View file

@ -30,11 +30,18 @@ def get_write_and_truncate_functions(output):
elif callable(output):
write = output
truncate = None
else:
assert isinstance(output, utils.WritableStream)
elif isinstance(output, utils.WritableStream):
def write(s):
output.write(s)
truncate = None
elif hasattr(output, 'open'):
def write(s):
with output.open('a') as output_file:
output_file.write(s)
def truncate():
with output.open('w'):
pass
return (write, truncate)