forked/osxphotos
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated QR Code example

Browse Source
This commit is contained in:
Rhet Turnbull 2022-09-05 18:32:09 -07:00
parent 74a730e420
commit 83ce702a46
1 changed files with 30 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
examples/detect_qrcodes.py
View File

@ -3,21 +3,21 @@
Run with `osxphotos run detect_qrcodes.py` Run with `osxphotos run detect_qrcodes.py`
Run with `osxphotos run detect_qrcodes.py --help` for help Run with `osxphotos run detect_qrcodes.py --help` for help
You'll need to install opencv into your osxphotos environment. This can be done with:
`osxphotos install opencv-python`
All the other dependencies should be already installed with osxphotos (e.g. rich, click) All dependencies are already installed as part of a standard osxphotos install.
""" """
import datetime import datetime
import json import json
import os import os
import os.path import os.path
from typing import Optional from typing import List
import click import click
import cv2 import objc
import Quartz
from Cocoa import NSURL
from Foundation import NSDictionary
from photoscript import Photo, PhotosLibrary from photoscript import Photo, PhotosLibrary
from rich import print from rich import print
from rich.progress import Progress from rich.progress import Progress
@ -29,12 +29,28 @@ from osxphotos.sqlitekvstore import SQLiteKVStore
QRCODE_KEYWORD = "qrcode" QRCODE_KEYWORD = "qrcode"
def detect_qrcode_in_image_cv2(filename: str) -> Optional[str]: def detect_qrcodes_in_image(filepath: str) -> List[str]:
"""Detect QR Code in image file""" """Detect QR Codes in images using CIDetector and return text of the found QR Codes"""
image = cv2.imread(filename) with objc.autorelease_pool():
qr_detect = cv2.QRCodeDetector() context = Quartz.CIContext.contextWithOptions_(None)
decoded_text, points, qrcode = qr_detect.detectAndDecode(image) options = NSDictionary.dictionaryWithDictionary_(
return decoded_text or None {"CIDetectorAccuracy": Quartz.CIDetectorAccuracyHigh}
)
detector = Quartz.CIDetector.detectorOfType_context_options_(
Quartz.CIDetectorTypeQRCode, context, options
)
results = []
input_url = NSURL.fileURLWithPath_(filepath)
input_image = Quartz.CIImage.imageWithContentsOfURL_(input_url)
features = detector.featuresInImage_(input_image)
if not features:
return []
for idx in range(features.count()):
feature = features.objectAtIndex_(idx)
results.append(feature.messageString())
return results
@click.command() @click.command()
@ -150,7 +166,7 @@ def detect_qrcodes(keyword, description, verbose_mode, dry_run, selected, reset)
} }
verbose(f"Processing {photo.original_filename} ({photo.uuid})") verbose(f"Processing {photo.original_filename} ({photo.uuid})")
num_processed += 1 num_processed += 1
if qrcode_text := detect_qrcode_in_image_cv2(photo_path): if qrcode_text := detect_qrcodes_in_image(photo_path):
# add qrcode tag/keyword to photo # add qrcode tag/keyword to photo
# osxphotos PhotoInfo objects are read-only but you can get a photoscript Photo object # osxphotos PhotoInfo objects are read-only but you can get a photoscript Photo object
# that allows you to modify certain data about the Photo via the Photos app AppleScript interface # that allows you to modify certain data about the Photo via the Photos app AppleScript interface
@ -158,7 +174,7 @@ def detect_qrcodes(keyword, description, verbose_mode, dry_run, selected, reset)
if not dry_run: if not dry_run:
photo_.keywords = list(set(photo_.keywords + [keyword])) photo_.keywords = list(set(photo_.keywords + [keyword]))
if description: if description:
photo_.description = qrcode_text photo_.description = ", ".join(qrcode_text)
record["qrcode"] = qrcode_text record["qrcode"] = qrcode_text
verbose( verbose(
f"Added {keyword} to {photo.original_filename} ({photo.uuid}), detected QR Code: {qrcode_text}" f"Added {keyword} to {photo.original_filename} ({photo.uuid}), detected QR Code: {qrcode_text}"