from datetime import datetime import pytest from pathlib import Path import re import shutil import tempfile from ordigi import constants from ordigi.media import Media from ordigi.images import Images from ordigi.exiftool import ExifTool, ExifToolCaching ORDIGI_PATH = Path(__file__).parent.parent CACHING = True class TestMetadata: @pytest.fixture(autouse=True) def setup_class(cls, sample_files_paths): cls.src_paths, cls.file_paths = sample_files_paths cls.ignore_tags = ('EXIF:CreateDate', 'File:FileModifyDate', 'File:FileAccessDate', 'EXIF:Make', 'Composite:LightValue') def get_media(self): for file_path in self.file_paths: self.exif_data = ExifTool(str(file_path)).asdict() yield Media(str(file_path), self.ignore_tags) def test_get_metadata(self): for media in self.get_media(): result = media.get_metadata() assert result assert isinstance(media.metadata, dict), media.metadata #check if all tags key are present for tags_key, tags in media.tags_keys.items(): assert tags_key in media.metadata for tag in tags: for tag_regex in self.ignore_tags: assert not re.match(tag_regex, tag) # Check for valid type for key, value in media.metadata.items(): if value or value == '': if 'date' in key: assert isinstance(value, datetime) elif key in ('latitude', 'longitude'): assert isinstance(value, float) else: assert isinstance(value, str) else: assert value is None # Check if has_exif_data() is True if 'date_original' key is # present, else check if it's false has_exif_data = False for tag in media.tags_keys['date_original']: if tag in media.exif_metadata: if media.get_date_format(media.exif_metadata[tag]): has_exif_data = True assert media.has_exif_data() break if has_exif_data == False: assert not media.has_exif_data() # Will be changed to get_metadata # check if metatadata type are correct # if(isinstance(self.metadata, dict) and update_cache is False): # return self.metadata # Album for folder implemented other place