Files
kg-scr/kg_ocr/ocr/extractor.py

18 lines
563 B
Python

import platform
from pathlib import Path
from typing import Optional
from .constants import def_paths, sc_pathpatterns
def get_screenshots(path: Optional[str | Path] = None) -> list[str]:
"""Find screenshot files for the current OS."""
if path is None:
path = def_paths.get(platform.system(), Path.home())
path = Path(path)
patterns = sc_pathpatterns.get(platform.system(), ["SCR*.png"])
results = []
for pattern in patterns:
results.extend(str(f.absolute()) for f in path.glob(pattern))
return sorted(set(results))