1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
# -*- coding: utf-8 -*-
# ******************************************************************************
#
# Copyright (C) 2010 Olivier Tilloy <olivier@tilloy.net>
#
# This file is part of the pyexiv2 distribution.
#
# pyexiv2 is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# pyexiv2 is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with pyexiv2; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, 5th Floor, Boston, MA 02110-1301 USA.
#
# Author: Olivier Tilloy <olivier@tilloy.net>
#
# ******************************************************************************
import unittest
import os.path
import hashlib
from datetime import datetime
from pyexiv2.metadata import ImageMetadata
import testutils
class TestBuffer(unittest.TestCase):
def setUp(self):
self.filename = os.path.join('data', 'smiley1.jpg')
self.md5sum = 'c066958457c685853293058f9bf129c1'
self.assert_(testutils.CheckFileSum(self.filename, self.md5sum))
def _metadata_from_buffer(self):
fd = open(self.filename, 'rb')
data = fd.read()
fd.close()
return ImageMetadata.from_buffer(data)
def test_from_file_and_from_buffer(self):
# from file
m1 = ImageMetadata(self.filename)
m1.read()
self.assertEqual(hashlib.md5(m1.buffer).hexdigest(), self.md5sum)
# from buffer
m2 = self._metadata_from_buffer()
self.assertEqual(hashlib.md5(m2.buffer).hexdigest(), self.md5sum)
def test_buffer_not_updated_until_write_called(self):
m = self._metadata_from_buffer()
m.read()
self.assertEqual(hashlib.md5(m.buffer).hexdigest(), self.md5sum)
# Modify the value of an EXIF tag
m['Exif.Image.DateTime'].value = datetime.today()
# Check that the buffer is unchanged until write() is called
self.assertEqual(hashlib.md5(m.buffer).hexdigest(), self.md5sum)
# Write back the changes
m.write()
# Check that the buffer has changed
self.failIfEqual(hashlib.md5(m.buffer).hexdigest(), self.md5sum)
def test_from_original_buffer(self):
m1 = self._metadata_from_buffer()
m2 = ImageMetadata.from_buffer(m1.buffer)
self.assertEqual(hashlib.md5(m2.buffer).hexdigest(), self.md5sum)
def test_from_modified_buffer(self):
m1 = self._metadata_from_buffer()
m1.read()
key = 'Exif.Image.ImageDescription'
value = 'my kingdom for a semiquaver'
m1[key] = value
m1.write()
m2 = ImageMetadata.from_buffer(m1.buffer)
m2.read()
self.assertEqual(m2[key].value, value)
|
