xevion/phototag
1
0
Fork 0
mirror of https://github.com/Xevion/phototag.git synced 2025-12-15 18:12:35 -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

27
package/app.py
View File

@@ -15,43 +15,38 @@ from .process import FileProcessor
from . import INPUT_PATH, TEMP_PATH, OUTPUT_PATH, PROCESSING_PATH
from . import RAW_EXTS, LOSSY_EXTS
# Process a single file in these steps:
# 1) Create a temporary file
# 2) Send it to GoogleAPI
# 3) Read XMP, then write new tags to it
# 4) Delete temporary file, move NEF/JPEG and XMP
log = logging.getLogger('app')
def run():
client = vision.ImageAnnotatorClient()
# Find files we want to process based on if they have a corresponding .XMP
logging.info('Locating processable files...')
log.info('Locating processable files...')
files = os.listdir(INPUT_PATH)
select = [file for file in files if os.path.splitext(file)[1] != '.xmp']
logging.info(f'Found {len(files)} valid files')
log.info(f'Found {len(files)} valid files')
# Create the 'temp' directory
logging.info('Creating temporary processing directory')
log.info('Creating temporary processing directory')
os.makedirs(TEMP_PATH)
os.makedirs(OUTPUT_PATH)
try:
# Process files
for index, file in progressbar.progressbar(list(enumerate(select)), redirect_stdout=True, term_width=110):
_, ext = os.path.splitext(file)
ext = ext[1:]
ext = ext[1:].lower()
if ext in LOSSY_EXTS or ext in RAW_EXTS:
process = FileProcessor(file)
logging.info(f"Processing file '{file}'...")
log.info(f"Processing file '{file}'...")
process.run(client)
except Exception as error:
logging.error(str(error))
logging.warning(
log.error(str(error))
log.warning(
'Removing temporary directory before raising exception.')
os.rmdir(TEMP_PATH)
raise
# Remove the directory, we are done here
logging.info('Removing temporary directory.')
log.info('Removing temporary directory.')
os.rmdir(TEMP_PATH)