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

change to named loggers

Browse Source
This commit is contained in:
Xevion
2019-11-01 23:36:15 -05:00
parent 3d7179be34
commit 7e3cb12513
4 changed files with 26 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
package/process.py
View File

@@ -12,6 +12,7 @@ from google.cloud import vision
from . import TEMP_PATH, INPUT_PATH, OUTPUT_PATH, RAW_EXTS, LOSSY_EXTS
from .xmp import XMPParser
log = logging.getLogger('process')
class FileProcessor(object):
def __init__(self, file_name: str):
@@ -63,21 +64,21 @@ class FileProcessor(object):
# Performs label detection on the image file
response = client.label_detection(image=image)
labels = [label.description for label in response.label_annotations]
logging.info('Keywords Identified: {}'.format(', '.join(labels)))
log.info('Keywords Identified: {}'.format(', '.join(labels)))
# XMP sidecar file specified, write to it using XML module
if self.xmp:
logging.info('Writing {} tags to output XMP.'.format(len(labels)))
log.info('Writing {} tags to output XMP.'.format(len(labels)))
parser = XMPParser(self.input_xmp)
parser.add_keywords(labels)
# Save the new XMP file
logging.debug('Saving to new XMP file.')
log.debug('Saving to new XMP file.')
parser.save(self.output_xmp)
logging.debug('Removing old XMP file.')
log.debug('Removing old XMP file.')
os.remove(self.input_xmp)
# No XMP file is specified, using IPTC tagging
else:
logging.info('Writing {} tags to image IPTC'.format(len(labels)))
log.info('Writing {} tags to image IPTC'.format(len(labels)))
info = iptcinfo3.IPTCInfo(os.path.join(INPUT_PATH, self.file_name))
info['keywords'].extend(labels)
info.save()