class RawImage:
""" class for handling raw images
- extract thumbnails
- copy them in the repository
"""
def __init__(self, strRawFile):
"""
Contructor of the class
@param strRawFile: path to the RawImage
@type strRawFile: string
"""
self.strRawFile = strRawFile
self.exif = None
self.strJepgFile = None
logger.info("Importing [Raw|Jpeg] image %s" % strRawFile)
def getJpegPath(self):
if self.exif is None:
self.exif = Exif(self.strRawFile)
self.exif.read()
if self.strJepgFile is None:
self.strJepgFile = unicode2ascii("%s-%s.jpg" % (
self.exif.interpretedExifValue("Exif.Photo.DateTimeOriginal").replace(" ", os.sep).replace(":", "-", 2).replace(":", "h", 1).replace(":", "m", 1),
self.exif.interpretedExifValue("Exif.Image.Model").strip().split(",")[-1].replace("/", "").replace(" ", "_")
))
while op.isfile(op.join(config.DefaultRepository, self.strJepgFile)):
number = ""
idx = None
listChar = list(self.strJepgFile[:-4])
listChar.reverse()
for val in listChar:
if val.isdigit():
number = val + number
elif val == "-":
idx = int(number)
break
else:
break
if idx is None:
self.strJepgFile = self.strJepgFile[:-4] + "-1.jpg"
else:
self.strJepgFile = self.strJepgFile[:-5 - len(number)] + "-%i.jpg" % (idx + 1)
dirname = op.dirname(op.join(config.DefaultRepository, self.strJepgFile))
if not op.isdir(dirname):
makedir(dirname)
return self.strJepgFile
def extractJPEG(self):
"""
extract the raw image to its right place
"""
extension = op.splitext(self.strRawFile)[1].lower()
strJpegFullPath = op.join(config.DefaultRepository, self.getJpegPath())
if extension in config.RawExtensions:
process = subprocess.Popen("%s %s" % (config.Dcraw, self.strRawFile), shell=True, stdout=subprocess.PIPE)
ret = process.wait()
if ret != 0:
logger.error("'%s %s' ended with error %s" % (config.Dcraw, self.strRawFile, ret))
data = process.stdout.readlines()
img = Image.fromstring("RGB", tuple([int(i) for i in data[1].split()]), "".join(tuple(data[3:])))
img.save(strJpegFullPath, format='JPEG')
#Copy all metadata useful for us.
exifJpeg = Exif(strJpegFullPath)
exifJpeg.read()
exifJpeg['Exif.Image.Orientation'] = 1
exifJpeg["Exif.Photo.UserComment"] = self.strRawFile
for metadata in [ 'Exif.Image.Make', 'Exif.Image.Model', 'Exif.Photo.DateTimeOriginal', 'Exif.Photo.ExposureTime', 'Exif.Photo.FNumber', 'Exif.Photo.ExposureBiasValue', 'Exif.Photo.Flash', 'Exif.Photo.FocalLength', 'Exif.Photo.ISOSpeedRatings']:
try:
exifJpeg[metadata] = self.exif[metadata]
except:
logger.error("Unable to copying metadata %s in file %s, value: %s" % (metadata, self.strRawFile, self.exif[metadata]))
#self.exif.copyMetadataTo(self.strJepgFile)
exifJpeg.write()
else: #in config.Extensions, i.e. a JPEG file
shutil.copy(self.strRawFile, strJpegFullPath)
Exiftran.autorotate(strJpegFullPath)
os.chmod(strJpegFullPath, config.DefaultFileMode)
