102 lines
2.6 KiB
Python
102 lines
2.6 KiB
Python
|
#!/usr/bin/env python
|
||
|
"""
|
||
|
REPL-friendly python mono-module for importing and execution singleton
|
||
|
"""
|
||
|
|
||
|
import os
|
||
|
from dataclasses import dataclass
|
||
|
from itertools import groupby
|
||
|
from typing import Iterable
|
||
|
|
||
|
mac_photos = os.getenv("MAC_PHOTOS_PATH", default="mac_photos")
|
||
|
|
||
|
|
||
|
@dataclass(slots=True, frozen=True)
|
||
|
class PhotosKExt:
|
||
|
name: str
|
||
|
parent: tuple[str, ...]
|
||
|
exts_lowered: tuple[str, ...]
|
||
|
|
||
|
@classmethod
|
||
|
def from_root_file(cls, root: str, filename: str):
|
||
|
ext_elems = filename.split('.')
|
||
|
return cls(
|
||
|
name=ext_elems[0],
|
||
|
parent=os.path.split(root),
|
||
|
exts_lowered=tuple(
|
||
|
ext.lower() for ext in ext_elems[1:]
|
||
|
)
|
||
|
)
|
||
|
|
||
|
|
||
|
@dataclass(slots=True, frozen=True)
|
||
|
class PhotosKey:
|
||
|
filename: str
|
||
|
parent: tuple[str, ...]
|
||
|
|
||
|
@classmethod
|
||
|
def from_key_ext(cls, ext: PhotosKExt):
|
||
|
return cls(filename=ext.name, parent=ext.parent)
|
||
|
|
||
|
|
||
|
@dataclass(slots=True)
|
||
|
class PhotosValue:
|
||
|
exts: dict[str, list[str]]
|
||
|
|
||
|
def heic_paths(self) -> list[str]:
|
||
|
return self.exts.get("heic", list())
|
||
|
|
||
|
def mov_paths(self) -> list[str]:
|
||
|
return self.exts.get("mov", list())
|
||
|
|
||
|
|
||
|
def candidate_exts(file_exts: Iterable[PhotosKExt]):
|
||
|
"""
|
||
|
>>> {ext: len(files)for ext, files in cexts.items()}
|
||
|
stdout> {... 'mov': 2769, 'heic': 2354 ...}
|
||
|
|
||
|
We could make a bold guess that iPhone 11 stores both `heic` and `mov`
|
||
|
for materialized best capture and live movie for live capture.
|
||
|
"""
|
||
|
exts_rv: dict[str, list[PhotosKExt]] = dict()
|
||
|
|
||
|
for file in file_exts:
|
||
|
for ext in file.exts_lowered:
|
||
|
exts_rv.setdefault(ext, list()).append(file)
|
||
|
|
||
|
return exts_rv
|
||
|
|
||
|
|
||
|
def file_exts(photos_dir: str = mac_photos):
|
||
|
return {
|
||
|
PhotosKExt.from_root_file(filename=file, root=root)
|
||
|
for root, _, files in os.walk(photos_dir, followlinks=True)
|
||
|
for file in files
|
||
|
}
|
||
|
|
||
|
|
||
|
def photos_exts(files_with_ext: Iterable[PhotosKExt]):
|
||
|
return {
|
||
|
k: list(v)
|
||
|
for k, v in groupby(files_with_ext, key=lambda x: (x.name, x.parent))
|
||
|
}
|
||
|
|
||
|
|
||
|
def main():
|
||
|
fexts = file_exts()
|
||
|
cexts = candidate_exts(fexts)
|
||
|
print(f"{photos_exts(fexts)=}")
|
||
|
print(f"{cexts=}")
|
||
|
movs = cexts['mov']
|
||
|
heics = cexts['heic']
|
||
|
heic_name_set = {heic.name.lower() for heic in heics}
|
||
|
mov_name_set = {mov.name.lower() for mov in movs}
|
||
|
print(len(heic_name_set - mov_name_set) - len(heic_name_set))
|
||
|
# stdout> (not 0)
|
||
|
# Hence, we are not able to match `.heic` with its `.mov` with just names
|
||
|
# Could it be that the `.heic` also contains the high-quality live?
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
main()
|