-
Notifications
You must be signed in to change notification settings - Fork 39
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #130 from LeoHsiao1/dev
Ready to release v2.11.0Dev
- Loading branch information
Showing
12 changed files
with
522 additions
and
226 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,138 @@ | ||
| import re | ||
|
|
||
| from .lib import exiv2api | ||
|
|
||
|
|
||
| # These tags are used by Windows and encoded in UCS2-LE. | ||
| # pyexiv2 will automatically convert encoding formats when reading and writing them. | ||
| EXIF_TAGS_ENCODED_IN_UCS2 = [ | ||
| 'Exif.Image.XPTitle', | ||
| 'Exif.Image.XPComment', | ||
| 'Exif.Image.XPAuthor', | ||
| 'Exif.Image.XPKeywords', | ||
| 'Exif.Image.XPSubject', | ||
| ] | ||
|
|
||
| # These tags can be written repeatedly, so there may be multiple values. | ||
| # pyexiv2 will convert their values to a list of strings. | ||
| IPTC_TAGS_REPEATABLE = [ | ||
| 'Iptc.Envelope.Destination', | ||
| 'Iptc.Envelope.ProductId', | ||
| 'Iptc.Application2.ObjectAttribute', | ||
| 'Iptc.Application2.Subject', | ||
| 'Iptc.Application2.SuppCategory', | ||
| 'Iptc.Application2.Keywords', | ||
| 'Iptc.Application2.LocationCode', | ||
| 'Iptc.Application2.LocationName', | ||
| 'Iptc.Application2.ReferenceService', | ||
| 'Iptc.Application2.ReferenceDate', | ||
| 'Iptc.Application2.ReferenceNumber', | ||
| 'Iptc.Application2.Byline', | ||
| 'Iptc.Application2.BylineTitle', | ||
| 'Iptc.Application2.Contact', | ||
| 'Iptc.Application2.Writer', | ||
| ] | ||
|
|
||
|
|
||
| def _parse(table: list, encoding='utf-8') -> dict: | ||
| """ Parse the metadata from a text table into a dict. """ | ||
| data = {} | ||
| for line in table: | ||
| tag, value, typeName = [field.decode(encoding) for field in line] | ||
| if typeName in ['XmpBag', 'XmpSeq']: | ||
| value = value.split(', ') | ||
| elif typeName in ['XmpText']: | ||
| # Handle nested array structures in XML. Refer to https://exiv2.org/manpage.html#set_xmp_struct | ||
| if value in ['type="Bag"', 'type="Seq"']: | ||
| value = [''] | ||
| elif typeName in ['LangAlt']: | ||
| # Refer to https://exiv2.org/manpage.html#langalt_values | ||
| if 'lang=' in value: | ||
| fields = re.split(r', (lang="\S+") ', ', ' + value)[1:] | ||
| value = {language: content for language, content in zip(fields[0::2], fields[1::2])} | ||
|
|
||
| # Convert the values to a list of strings if the tag has multiple values | ||
| pre_value = data.get(tag) | ||
| if pre_value == None: | ||
| data[tag] = value | ||
| elif isinstance(pre_value, str): | ||
| data[tag] = [pre_value, value] | ||
| elif isinstance(pre_value, list): | ||
| data[tag].append(value) | ||
|
|
||
| return data | ||
|
|
||
|
|
||
| def _dumps(data: dict) -> list: | ||
| """ Convert the metadata from a dict into a text table. """ | ||
| table = [] | ||
| for tag, value in data.items(): | ||
| tag = str(tag) | ||
| if value == None: | ||
| typeName = '_delete' | ||
| value = '' | ||
| elif isinstance(value, (list, tuple)): | ||
| typeName = 'array' | ||
| value = list(value) | ||
| elif isinstance(value, dict): | ||
| typeName = 'string' | ||
| value = ', '.join(['{} {}'.format(k,v) for k,v in value.items()]) | ||
| else: | ||
| typeName = 'string' | ||
| value = str(value) | ||
| line = [tag, value, typeName] | ||
| table.append(line) | ||
| return table | ||
|
|
||
|
|
||
| def decode_ucs2(text: str) -> str: | ||
| """ | ||
| Convert text from UCS2 encoding to UTF8 encoding. | ||
| For example: | ||
| >>> decode_ucs2('116 0 101 0 115 0 116 0') | ||
| 'test' | ||
| """ | ||
| hex_str = ''.join(['{:02x}'.format(int(i)) for i in text.split()]) | ||
| return bytes.fromhex(hex_str).decode('utf-16le') | ||
|
|
||
|
|
||
| def encode_ucs2(text: str) -> str: | ||
| """ | ||
| Convert text from UTF8 encoding to UCS2 encoding. | ||
| For example: | ||
| >>> encode_ucs2('test') | ||
| '116 0 101 0 115 0 116 0' | ||
| """ | ||
| hex_str = text.encode('utf-16le').hex() | ||
| int_list = [int(''.join(i), base=16) for i in zip(*[iter(hex_str)] * 2)] | ||
| return ' '.join([str(i) for i in int_list]) | ||
|
|
||
|
|
||
| def convert_exif_to_xmp(data: dict, encoding='utf-8') -> dict: | ||
| """ Input EXIF metadata, convert to XMP metadata and return. It works like executing modify_exif() then read_xmp(). """ | ||
| data = data.copy() | ||
| for tag in EXIF_TAGS_ENCODED_IN_UCS2: | ||
| value = data.get(tag) | ||
| if value: | ||
| data[tag] = encode_ucs2(value) | ||
| converted_data = exiv2api.convert_exif_to_xmp(_dumps(data), encoding) | ||
| return _parse(converted_data, encoding) | ||
|
|
||
|
|
||
| def convert_iptc_to_xmp(data: dict, encoding='utf-8') -> dict: | ||
| """ Input IPTC metadata, convert to XMP metadata and return. It works like executing modify_iptc() then read_xmp(). """ | ||
| converted_data = exiv2api.convert_iptc_to_xmp(_dumps(data), encoding) | ||
| return _parse(converted_data, encoding) | ||
|
|
||
|
|
||
| def convert_xmp_to_exif(data: dict, encoding='utf-8') -> dict: | ||
| """ Input XMP metadata, convert to EXIF metadata and return. It works like executing modify_xmp() then read_exif(). """ | ||
| converted_data = exiv2api.convert_xmp_to_exif(_dumps(data), encoding) | ||
| return _parse(converted_data, encoding) | ||
|
|
||
|
|
||
| def convert_xmp_to_iptc(data: dict, encoding='utf-8') -> dict: | ||
| """ Input XMP metadata, convert to IPTC metadata and return. It works like executing modify_xmp() then read_iptc(). """ | ||
| converted_data = exiv2api.convert_xmp_to_iptc(_dumps(data), encoding) | ||
| return _parse(converted_data, encoding) | ||
|
|
Oops, something went wrong.