xevion/phototag
1
0
Fork 0
mirror of https://github.com/Xevion/phototag.git synced 2025-12-06 05:15:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

feature to copy old XMP sidecar file stats to new XMP sidecar file

Browse Source
This commit is contained in:
Xevion
2019-11-03 22:48:27 -06:00
parent 1d328aeffe
commit b66163e58e
2 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
phototag/app.py
View File

@@ -35,9 +35,6 @@ def run():
if not os.path.exists(TEMP_PATH):
log.info('Creating temporary processing directory')
os.makedirs(TEMP_PATH)
if not os.path.exists(OUTPUT_PATH):
log.info('Creating output processing directory')
os.makedirs(OUTPUT_PATH)
try:
# Process files

19
phototag/process.py
View File

@@ -5,6 +5,9 @@ import imageio
import io
import iptcinfo3
import logging
import string
import random
import shutil
from PIL import Image
from google.cloud.vision import types
from google.cloud import vision
@@ -26,7 +29,6 @@ class FileProcessor(object):
if self.ext.lower() in RAW_EXTS:
self.xmp = self.base + '.xmp'
self.input_xmp = os.path.join(INPUT_PATH, self.xmp)
self.output_xmp = os.path.join(OUTPUT_PATH, self.xmp)
if not os.path.exists(self.input_xmp):
raise Exception('Sidecar file for \'{}\' does not exist.'.format(self.xmp))
@@ -72,10 +74,15 @@ class FileProcessor(object):
parser = XMPParser(self.input_xmp)
parser.add_keywords(labels)
# Save the new XMP file
log.debug('Saving to new XMP file.')
parser.save(self.output_xmp)
log.debug('Removing old XMP file.')
os.remove(self.input_xmp)
log.debug('Moving old XMP to temp XMP')
temp_name = self.input_xmp + ''.join(random.choices(list(string.ascii_letters), k=10))
os.rename(self.input_xmp, temp_name)
log.debug('Saving new XMP')
parser.save(self.input_xmp)
log.debug('Copying old stats to new XMP')
shutil.copystat(temp_name, self.input_xmp)
log.debug('Removing temp file')
os.remove(temp_name)
# No XMP file is specified, using IPTC tagging
else:
log.info('Writing {} tags to image IPTC'.format(len(labels)))
@@ -87,7 +94,7 @@ class FileProcessor(object):
# Copy dry-run
# shutil.copy2(os.path.join(INPUT_PATH, self.file_name), os.path.join(OUTPUT_PATH, self.file_name))
os.rename(os.path.join(INPUT_PATH, self.file_name), os.path.join(OUTPUT_PATH, self.file_name))
# os.rename(os.path.join(INPUT_PATH, self.file_name), os.path.join(OUTPUT_PATH, self.file_name))
except:
self._cleanup()
raise