Skip to content

Instantly share code, notes, and snippets.

@lambdafu
Created November 7, 2016 14:23
Show Gist options
  • Save lambdafu/553737b6b0f0c8de7d830880ec8ea1c5 to your computer and use it in GitHub Desktop.
Save lambdafu/553737b6b0f0c8de7d830880ec8ea1c5 to your computer and use it in GitHub Desktop.
from PIL import Image, TiffImagePlugin
import shutil
import os
from base64 import decodestring
TEST_TIF = "test-delete.tif"
test_tif = decodestring(
"""SUkqAAgAAAAMAAABBAABAAAACgAAAAEBBAABAAAACgAAAAIBAwABAAAACAAAAAMBAwABAAAAAQAA
AAYBAwABAAAAAwAAABEBBAABAAAArgYAABYBBAABAAAACgAAABcBBAABAAAAZAAAABoBBQABAAAA
ngAAABsBBQABAAAApgAAABwBAwABAAAAAQAAAEABAwAAAwAArgAAAAAAAAAAAAAAAQAAAAAAAAAB
AAAAAAAAAAACAAMABAAFAAYABwAIAAkACgALAAwADQAOAA8AEAARABIAEwAUABUAFgAXABgAGQAa
ABsAHAAdAB4AHwAgACEAIgAjACQAJQAmACcAKAApACoAKwAsAC0ALgAvADAAMQAyADMANAA1ADYA
NwA4ADkAOgA7ADwAPQA+AD8AQABBAEIAQwBEAEUARgBHAEgASQBKAEsATABNAE4ATwBQAFEAUgBT
AFQAVQBWAFcAWABZAFoAWwBcAF0AXgBfAGAAYQBiAGMAZABlAGYAZwBoAGkAagBrAGwAbQBuAG8A
cABxAHIAcwB0AHUAdgB3AHgAeQB6AHsAfAB9AH4AfwCAAIEAggCDAIQAhQCGAIcAiACJAIoAiwCM
AI0AjgCPAJAAkQCSAJMAlACVAJYAlwCYAJkAmgCbAJwAnQCeAJ8AoAChAKIAowCkAKUApgCnAKgA
qQCqAKsArACtAK4ArwCwALEAsgCzALQAtQC2ALcAuAC5ALoAuwC8AL0AvgC/AMAAwQDCAMMAxADF
AMYAxwDIAMkAygDLAMwAzQDOAM8A0ADRANIA0wDUANUA1gDXANgA2QDaANsA3ADdAN4A3wDgAOEA
4gDjAOQA5QDmAOcA6ADpAOoA6wDsAO0A7gDvAPAA8QDyAPMA9AD1APYA9wD4APkA+gD7APwA/QD+
AP8AgAAAAAIAAwAEAAUABgAHAAgACQAKAAsADAANAA4ADwAQABEAEgATABQAFQAWABcAGAAZABoA
GwAcAB0AHgAfACAAIQAiACMAJAAlACYAJwAoACkAKgArACwALQAuAC8AMAAxADIAMwA0ADUANgA3
ADgAOQA6ADsAPAA9AD4APwBAAEEAQgBDAEQARQBGAEcASABJAEoASwBMAE0ATgBPAFAAUQBSAFMA
VABVAFYAVwBYAFkAWgBbAFwAXQBeAF8AYABhAGIAYwBkAGUAZgBnAGgAaQBqAGsAbABtAG4AbwBw
AHEAcgBzAHQAdQB2AHcAeAB5AHoAewB8AH0AfgB/AIAAgQCCAIMAhACFAIYAhwCIAIkAigCLAIwA
jQCOAI8AkACRAJIAkwCUAJUAlgCXAJgAmQCaAJsAnACdAJ4AnwCgAKEAogCjAKQApQCmAKcAqACp
AKoAqwCsAK0ArgCvALAAsQCyALMAtAC1ALYAtwC4ALkAugC7ALwAvQC+AL8AwADBAMIAwwDEAMUA
xgDHAMgAyQDKAMsAzADNAM4AzwDQANEA0gDTANQA1QDWANcA2ADZANoA2wDcAN0A3gDfAOAA4QDi
AOMA5ADlAOYA5wDoAOkA6gDrAOwA7QDuAO8A8ADxAPIA8wD0APUA9gD3APgA+QD6APsA/AD9AP4A
/wAAAAAAAgADAAQABQAGAAcACAAJAAoACwAMAA0ADgAPABAAEQASABMAFAAVABYAFwAYABkAGgAb
ABwAHQAeAB8AIAAhACIAIwAkACUAJgAnACgAKQAqACsALAAtAC4ALwAwADEAMgAzADQANQA2ADcA
OAA5ADoAOwA8AD0APgA/AEAAQQBCAEMARABFAEYARwBIAEkASgBLAEwATQBOAE8AUABRAFIAUwBU
AFUAVgBXAFgAWQBaAFsAXABdAF4AXwBgAGEAYgBjAGQAZQBmAGcAaABpAGoAawBsAG0AbgBvAHAA
cQByAHMAdAB1AHYAdwB4AHkAegB7AHwAfQB+AH8AgACBAIIAgwCEAIUAhgCHAIgAiQCKAIsAjACN
AI4AjwCQAJEAkgCTAJQAlQCWAJcAmACZAJoAmwCcAJ0AngCfAKAAoQCiAKMApAClAKYApwCoAKkA
qgCrAKwArQCuAK8AsACxALIAswC0ALUAtgC3ALgAuQC6ALsAvAC9AL4AvwDAAMEAwgDDAMQAxQDG
AMcAyADJAMoAywDMAM0AzgDPANAA0QDSANMA1ADVANYA1wDYANkA2gDbANwA3QDeAN8A4ADhAOIA
4wDkAOUA5gDnAOgA6QDqAOsA7ADtAO4A7wDwAPEA8gDzAPQA9QD2APcA+AD5APoA+wD8AP0A/gD/
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA==
""")
def rewrite_test_tif():
if os.path.exists(TEST_TIF):
os.remove(TEST_TIF)
with open(TEST_TIF, "wb") as fh:
fh.write(test_tif)
def test_run(expect=True):
# setup
rewrite_test_tif()
im = Image.open(TEST_TIF)
im.load()
im.close()
success = False
try:
rewrite_test_tif()
success = True
except Exception as e:
print "ERROR: %r" % e
if expect != success:
print "ERROR: Result %r is not expected result %r" % (success, expect)
def patch_pillow():
TiffImageFile = TiffImagePlugin.TiffImageFile
def close(self):
# Name is mangled!
if self._TiffImageFile__fp:
self._TiffImageFile__fp = None
super(TiffImageFile, self).close()
TiffImageFile.close = close
print "Test run with original Pillow"
test_run(False)
patch_pillow()
print "Test run with patched Pillow"
test_run(True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment