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

XMP restructuring, mostly ready for production usage

Browse Source
This commit is contained in:
Xevion
2019-08-08 16:01:52 -05:00
parent ec3e7c0fd4
commit 62e5f08deb
4 changed files with 30 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
package/xmp.py
View File

@@ -1,57 +1,33 @@
# from libxmp import file_to_dict
import os, sys, re
basepath = "E:\\Photography\\Colorado 2019\\"
ext_filter = lambda file : file.endswith('.NEF')
mainPattern = r'<dc:subject>\n\s+<rdf:Bag>\n(?: <rdf:li>.+<\/rdf:li>\n)*\s+<\/rdf:Bag>\n\s+<\/dc:subject>\n'
subPattern = r'<rdf:li>(.*)</rdf:li>\n'
subFormatPattern = ' <rdf:li>{}</rdf:li>\n'
def getXMP(file, basepath=None):
xmpFile = file.rfind('.')
xmpFile = file[:xmpFile] + '.xmp'
if basepath:
xmpFile = os.path.join(basepath, xmpFile)
return xmpFile
def writeTags(tags, filename):
fullpath = os.path.join(basepath, getXMP(filename))
if not os.path.exists(fullpath):
raise FileNotFoundError(f'No XMP file found for {filename}.')
data = open(fullpath, 'r').read()
def writeTags(inputPath, outputPath, tags):
data = open(inputPath, 'r').read()
# Detect and find the <dc:Subject: part
match = re.search(mainPattern, data)
# If it's not found, we just add the tags area right below the creator tag
if not match:
addition = """\n <dc:subject>\n <rdf:Bag>\n </rdf:Bag>\n </dc:subject>"""
look = data.find('</dc:creator>') + 13
data = data[:look] + addition + data[look:]
match = re.search(mainPattern, data)
print(os.path.join(basepath, filename))
print(fullpath)
print('. . .')
# Get last matching tag
submatch = None
for submatch in re.finditer(subPattern, match.group(0)):
pass
# Calculate very end
spanend = match.span()[0] + (submatch.span()[1] if submatch else 0)
# add tags to end
data = data[:spanend] + ''.join(subFormatPattern.format(tag) for tag in tags) + data[spanend:]
# Write file to disk
open(fullpath, 'w').write( data)
files = os.listdir(basepath)
print(f'Total Files: {len(files)}')
files = list(filter(ext_filter, files))
print(f'Total Files after .NEF filter: {len(files)}')
xmps = [file for file in files if os.path.exists(getXMP(file, basepath=basepath))]
print(f'Total .NEF files with .XMP pair: {len(xmps)}')
print(f'{len(xmps)} x 2 == {len(xmps) * 2}')
select = files[43]
writeTags(['helpgod', 'ohfuck'], select)
print(select)
open(outputPath, 'w+').write( data)