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

replaced debug print statements with logging.debug

Browse Source
This commit is contained in:
Rhet Turnbull
2019-11-17 16:45:46 -08:00
parent 10f0cf1092
commit e0b1113870
1 changed files with 73 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

139
osxphotos/__init__.py
View File

@@ -1,7 +1,7 @@
import json import json
import logging
import os.path import os.path
import platform import platform
import pprint
import sqlite3 import sqlite3
import sys import sys
import tempfile import tempfile
@@ -9,11 +9,13 @@ import urllib.parse
from datetime import datetime, timedelta, timezone from datetime import datetime, timedelta, timezone
from pathlib import Path from pathlib import Path
from plistlib import load as plistload from plistlib import load as plistload
from pprint import pformat
from shutil import copyfile from shutil import copyfile
import yaml
import CoreFoundation import CoreFoundation
import objc import objc
import yaml
from Foundation import * from Foundation import *
from . import _applescript from . import _applescript
@@ -42,7 +44,13 @@ _PHOTOS_5_VERSION = "6000"
_TESTED_OS_VERSIONS = ["12", "13", "14"] _TESTED_OS_VERSIONS = ["12", "13", "14"]
_debug = True _debug = True
logging.basicConfig(
level=logging.DEBUG,
format="%(asctime)s - %(levelname)s - %(filename)s - %(lineno)d - %(message)s",
)
if not _debug:
logging.disable()
def _get_os_version(): def _get_os_version():
# returns tuple containing OS version # returns tuple containing OS version
@@ -83,7 +91,7 @@ class PhotosDB:
"WARNING: This module has only been tested with MacOS 10." "WARNING: This module has only been tested with MacOS 10."
+ f"[{', '.join(_TESTED_OS_VERSIONS)}]: " + f"[{', '.join(_TESTED_OS_VERSIONS)}]: "
+ f"you have {system}, OS version: {major}", + f"you have {system}, OS version: {major}",
file=sys.stderr, file=sys.stderr
) )
# configure AppleScripts used to manipulate Photos # configure AppleScripts used to manipulate Photos
@@ -111,7 +119,7 @@ class PhotosDB:
# list of temporary files created so we can clean them up later # list of temporary files created so we can clean them up later
self._tmp_files = [] self._tmp_files = []
print(f"DEBUG: dbfile = {dbfile}") logging.debug(f"dbfile = {dbfile}")
if dbfile is None: if dbfile is None:
library_path = self.get_photos_library_path() library_path = self.get_photos_library_path()
# logger.debug("library_path: " + library_path) # logger.debug("library_path: " + library_path)
@@ -129,10 +137,10 @@ class PhotosDB:
self._tmp_db = self._copy_db_file(self._dbfile) self._tmp_db = self._copy_db_file(self._dbfile)
self._db_version = self._get_db_version() self._db_version = self._get_db_version()
if int(self._db_version) >= int(_PHOTOS_5_VERSION): if int(self._db_version) >= int(_PHOTOS_5_VERSION):
print(f"DEBUG: version is {self._db_version}") logging.debug(f"version is {self._db_version}")
dbpath = Path(self._dbfile).parent dbpath = Path(self._dbfile).parent
dbfile = dbpath / "Photos.sqlite" dbfile = dbpath / "Photos.sqlite"
print(f"DEBUG: dbfile = {dbfile}") logging.debug(f"dbfile = {dbfile}")
if not _check_file_exists(dbfile): if not _check_file_exists(dbfile):
sys.exit(f"dbfile {dbfile} does not exist") sys.exit(f"dbfile {dbfile} does not exist")
else: else:
@@ -152,7 +160,7 @@ class PhotosDB:
masters_path = os.path.join(library_path, "originals") masters_path = os.path.join(library_path, "originals")
self._masters_path = masters_path self._masters_path = masters_path
print(f"DEBUG: library = {library_path}, masters = {masters_path}") logging.debug(f"library = {library_path}, masters = {masters_path}")
# logger.info(f"database filename = {dbfile}") # logger.info(f"database filename = {dbfile}")
@@ -164,9 +172,9 @@ class PhotosDB:
def _cleanup_tmp_files(self): def _cleanup_tmp_files(self):
""" removes all temporary files whose names are stored in self.tmp_files """ """ removes all temporary files whose names are stored in self.tmp_files """
""" does not raise exception if file cannot be deleted (e.g. it was already cleaned up) """ """ does not raise exception if file cannot be deleted (e.g. it was already cleaned up) """
print(f"DEBUG: tmp files ={self._tmp_files}") logging.debug(f"tmp files ={self._tmp_files}")
for f in self._tmp_files: for f in self._tmp_files:
print(f"DEBUG: cleaning up {f}") logging.debug(f"cleaning up {f}")
try: try:
os.remove(f) os.remove(f)
except Exception as e: except Exception as e:
@@ -304,7 +312,7 @@ class PhotosDB:
else: else:
print( print(
"Could not extract photos URL String from IPXDefaultLibraryURLBookmark", "Could not extract photos URL String from IPXDefaultLibraryURLBookmark",
file=sys.stderr, file=sys.stderr
) )
return None return None
@@ -337,7 +345,7 @@ class PhotosDB:
conn = sqlite3.connect(f"{fname}") conn = sqlite3.connect(f"{fname}")
c = conn.cursor() c = conn.cursor()
except sqlite3.Error as e: except sqlite3.Error as e:
print(f"An error occurred: {e.args[0]} {fname}") print(f"An error occurred: {e.args[0]} {fname}",file=sys.stderr)
sys.exit(3) sys.exit(3)
# logger.debug("SQLite database is open") # logger.debug("SQLite database is open")
return (conn, c) return (conn, c)
@@ -345,7 +353,6 @@ class PhotosDB:
def _get_db_version(self): def _get_db_version(self):
""" gets the Photos DB version from LiGlobals table """ """ gets the Photos DB version from LiGlobals table """
""" returns the version as str""" """ returns the version as str"""
global _debug
version = None version = None
(conn, c) = self._open_sql_file(self._tmp_db) (conn, c) = self._open_sql_file(self._tmp_db)
@@ -373,7 +380,6 @@ class PhotosDB:
def _process_database4(self): def _process_database4(self):
""" process the Photos database to extract info """ """ process the Photos database to extract info """
""" works on Photos version <= 4.0 """ """ works on Photos version <= 4.0 """
global _debug
# Epoch is Jan 1, 2001 # Epoch is Jan 1, 2001
td = (datetime(2001, 1, 1, 0, 0) - datetime(1970, 1, 1, 0, 0)).total_seconds() td = (datetime(2001, 1, 1, 0, 0) - datetime(1970, 1, 1, 0, 0)).total_seconds()
@@ -523,8 +529,7 @@ class PhotosDB:
# set_pbar_status(i) # set_pbar_status(i)
i = i + 1 i = i + 1
uuid = row[0] uuid = row[0]
if _debug: logging.debug(f"i = {i:d}, uuid = '{uuid}, master = '{row[2]}")
print(f"i = {i:d}, uuid = '{uuid}, master = '{row[2]}")
self._dbphotos[uuid] = {} self._dbphotos[uuid] = {}
self._dbphotos[uuid]["modelID"] = row[1] self._dbphotos[uuid]["modelID"] = row[1]
self._dbphotos[uuid]["masterUuid"] = row[2] self._dbphotos[uuid]["masterUuid"] = row[2]
@@ -599,27 +604,26 @@ class PhotosDB:
self._cleanup_tmp_files() self._cleanup_tmp_files()
if _debug: if _debug:
pp = pprint.PrettyPrinter(indent=4) logging.debug("Faces:")
print("Faces:") logging.debug(pformat(self._dbfaces_uuid))
pp.pprint(self._dbfaces_uuid)
print("Keywords by uuid:") logging.debug("Keywords by uuid:")
pp.pprint(self._dbkeywords_uuid) logging.debug(pformat(self._dbkeywords_uuid))
print("Keywords by keyword:") logging.debug("Keywords by keyword:")
pp.pprint(self._dbkeywords_keyword) logging.debug(pformat(self._dbkeywords_keyword))
print("Albums by uuid:") logging.debug("Albums by uuid:")
pp.pprint(self._dbalbums_uuid) logging.debug(pformat(self._dbalbums_uuid))
print("Albums by album:") logging.debug("Albums by album:")
pp.pprint(self._dbalbums_album) logging.debug(pformat(self._dbalbums_album))
print("Volumes:") logging.debug("Volumes:")
pp.pprint(self._dbvolumes) logging.debug(pformat(self._dbvolumes))
print("Photos:") logging.debug("Photos:")
pp.pprint(self._dbphotos) logging.debug(pformat(self._dbphotos))
# logger.debug(f"processed {len(self._dbphotos)} photos") # logger.debug(f"processed {len(self._dbphotos)} photos")
@@ -627,10 +631,7 @@ class PhotosDB:
""" process the Photos database to extract info """ """ process the Photos database to extract info """
""" works on Photos version >= 5.0 """ """ works on Photos version >= 5.0 """
global _debug logging.debug(f"_process_database5")
if _debug:
print(f"DEBUG: _process_database5")
# Epoch is Jan 1, 2001 # Epoch is Jan 1, 2001
td = (datetime(2001, 1, 1, 0, 0) - datetime(1970, 1, 1, 0, 0)).total_seconds() td = (datetime(2001, 1, 1, 0, 0) - datetime(1970, 1, 1, 0, 0)).total_seconds()
@@ -656,7 +657,7 @@ class PhotosDB:
# (conn, c) = self._open_sql_file(tmp_db) # (conn, c) = self._open_sql_file(tmp_db)
# Look for all combinations of persons and pictures # Look for all combinations of persons and pictures
print(f"DEBUG: Getting information about persons") logging.debug(f"Getting information about persons")
i = 0 i = 0
c.execute( c.execute(
@@ -685,10 +686,9 @@ class PhotosDB:
self._dbfaces_person[person[0]].append(person[1]) self._dbfaces_person[person[0]].append(person[1])
# set_pbar_status(i) # set_pbar_status(i)
i = i + 1 i = i + 1
print(f"DEBUG: Finished walking through persons") logging.debug(f"Finished walking through persons")
pp = pprint.PrettyPrinter(indent=4) logging.debug(pformat(self._dbfaces_person))
print(f"DEBUG: {pp.pprint(self._dbfaces_person)}") logging.debug(self._dbfaces_uuid)
print(f"DEBUG: {pp.pprint(self._dbfaces_uuid)}")
# close_pbar_status() # close_pbar_status()
@@ -721,9 +721,9 @@ class PhotosDB:
# logger.debug(f"{album[1]} {album[0]}") # logger.debug(f"{album[1]} {album[0]}")
# set_pbar_status(i) # set_pbar_status(i)
i = i + 1 i = i + 1
print(f"DEBUG: Finished walking through albums") logging.debug(f"Finished walking through albums")
print(f"DEBUG: {pp.pprint(self._dbalbums_album)}") logging.debug(pformat(self._dbalbums_album))
print(f"DEBUG: {pp.pprint(self._dbalbums_uuid)}") logging.debug(pformat(self._dbalbums_uuid))
# close_pbar_status() # close_pbar_status()
# logger.debug("Getting information about keywords") # logger.debug("Getting information about keywords")
@@ -755,10 +755,10 @@ class PhotosDB:
# logger.debug(f"{keyword[1]} {keyword[0]}") # logger.debug(f"{keyword[1]} {keyword[0]}")
# set_pbar_status(i) # set_pbar_status(i)
i = i + 1 i = i + 1
print(f"DEBUG: Finished walking through keywords") logging.debug(f"Finished walking through keywords")
# close_pbar_status() # close_pbar_status()
pp.pprint(self._dbkeywords_keyword) logging.debug(pformat(self._dbkeywords_keyword))
pp.pprint(self._dbkeywords_uuid) logging.debug(pformat(self._dbkeywords_uuid))
# logger.debug("Getting information about volumes") # logger.debug("Getting information about volumes")
c.execute("SELECT COUNT(*) FROM ZFILESYSTEMVOLUME") c.execute("SELECT COUNT(*) FROM ZFILESYSTEMVOLUME")
@@ -770,11 +770,11 @@ class PhotosDB:
# logger.debug(f"{vol[0]} {vol[1]}") # logger.debug(f"{vol[0]} {vol[1]}")
# set_pbar_status(i) # set_pbar_status(i)
i = i + 1 i = i + 1
print(f"DEBUG: Finished walking through volumes") logging.debug(f"Finished walking through volumes")
pp.pprint(self._dbvolumes) logging.debug(self._dbvolumes)
# close_pbar_status() # close_pbar_status()
print(f"DEBUG: Getting information about photos") logging.debug(f"Getting information about photos")
# TODO: Since I don't use progress bars now, can probably remove the count # TODO: Since I don't use progress bars now, can probably remove the count
c.execute( c.execute(
"SELECT COUNT(*) " "SELECT COUNT(*) "
@@ -830,7 +830,7 @@ class PhotosDB:
i = i + 1 i = i + 1
uuid = row[0] uuid = row[0]
if _debug: if _debug:
print(f"i = {i:d}, uuid = '{uuid}") logging.debug(f"i = {i:d}, uuid = '{uuid}")
self._dbphotos[uuid] = {} self._dbphotos[uuid] = {}
self._dbphotos[uuid]["modelID"] = None self._dbphotos[uuid]["modelID"] = None
self._dbphotos[uuid]["masterUuid"] = None self._dbphotos[uuid]["masterUuid"] = None
@@ -901,8 +901,8 @@ class PhotosDB:
if uuid in self._dbphotos: if uuid in self._dbphotos:
self._dbphotos[uuid]["extendedDescription"] = row[1] self._dbphotos[uuid]["extendedDescription"] = row[1]
else: else:
print( logging.debug(
f"DEBUG WARNING: found description {row[1]} but no photo for {uuid}" f"WARNING: found description {row[1]} but no photo for {uuid}"
) )
# get information on local/remote availability # get information on local/remote availability
@@ -917,7 +917,7 @@ class PhotosDB:
i = 0 i = 0
for row in c: for row in c:
i = i+1 i = i + 1
uuid = row[0] uuid = row[0]
if uuid in self._dbphotos: if uuid in self._dbphotos:
self._dbphotos[uuid]["localAvailability"] = row[1] self._dbphotos[uuid]["localAvailability"] = row[1]
@@ -928,7 +928,7 @@ class PhotosDB:
self._dbphotos[uuid]["isMissing"] = 0 self._dbphotos[uuid]["isMissing"] = 0
if _debug: if _debug:
pp.pprint(self._dbphotos) logging.debug(pformat(self._dbphotos))
# add faces and keywords to photo data # add faces and keywords to photo data
for uuid in self._dbphotos: for uuid in self._dbphotos:
@@ -966,27 +966,26 @@ class PhotosDB:
self._cleanup_tmp_files() self._cleanup_tmp_files()
if _debug: if _debug:
pp = pprint.PrettyPrinter(indent=4) logging.debug("Faces:")
print("Faces:") logging.debug(pformat(self._dbfaces_uuid))
pp.pprint(self._dbfaces_uuid)
print("Keywords by uuid:") logging.debug("Keywords by uuid:")
pp.pprint(self._dbkeywords_uuid) logging.debug(pformat(self._dbkeywords_uuid))
print("Keywords by keyword:") logging.debug("Keywords by keyword:")
pp.pprint(self._dbkeywords_keyword) logging.debug(pformat(self._dbkeywords_keyword))
print("Albums by uuid:") logging.debug("Albums by uuid:")
pp.pprint(self._dbalbums_uuid) logging.debug(pformat(self._dbalbums_uuid))
print("Albums by album:") logging.debug("Albums by album:")
pp.pprint(self._dbalbums_album) logging.debug(pformat(self._dbalbums_album))
print("Volumes:") logging.debug("Volumes:")
pp.pprint(self._dbvolumes) logging.debug(pformat(self._dbvolumes))
print("Photos:") logging.debug("Photos:")
pp.pprint(self._dbphotos) logging.debug(pformat(self._dbphotos))
# logger.debug(f"processed {len(self._dbphotos)} photos") # logger.debug(f"processed {len(self._dbphotos)} photos")
@@ -1096,7 +1095,9 @@ class PhotoInfo:
if vol is not None: if vol is not None:
photopath = os.path.join("/Volumes", vol, self.__info["imagePath"]) photopath = os.path.join("/Volumes", vol, self.__info["imagePath"])
else: else:
photopath = os.path.join(self.__db._masters_path, self.__info["imagePath"]) photopath = os.path.join(
self.__db._masters_path, self.__info["imagePath"]
)
if self.__info["isMissing"] == 1: if self.__info["isMissing"] == 1:
# logger.warning( # logger.warning(