Skip to content

Commit

Permalink
Updated report writer
Browse files Browse the repository at this point in the history
  • Loading branch information
RhetTbull committed Apr 21, 2024
1 parent 2714fc0 commit 5417731
Showing 1 changed file with 27 additions and 4 deletions.
31 changes: 27 additions & 4 deletions osxphotos/cli/report_writer.py
Expand Up @@ -109,6 +109,8 @@ def __init__(
"user_written",
"user_skipped",
"user_error",
"aae_written",
"aae_skipped",
]

mode = "a" if append else "w"
Expand Down Expand Up @@ -208,9 +210,9 @@ def write(self, export_results: ExportResults):
cursor = self._conn.cursor()
cursor.execute(
"INSERT INTO report "
"(datetime, filename, exported, new, updated, skipped, exif_updated, touched, converted_to_jpeg, sidecar_xmp, sidecar_json, sidecar_exiftool, missing, error, exiftool_warning, exiftool_error, extended_attributes_written, extended_attributes_skipped, cleanup_deleted_file, cleanup_deleted_directory, exported_album, report_id, sidecar_user, sidecar_user_error, user_written, user_skipped, user_error) " # noqa
"(datetime, filename, exported, new, updated, skipped, exif_updated, touched, converted_to_jpeg, sidecar_xmp, sidecar_json, sidecar_exiftool, missing, error, exiftool_warning, exiftool_error, extended_attributes_written, extended_attributes_skipped, cleanup_deleted_file, cleanup_deleted_directory, exported_album, report_id, sidecar_user, sidecar_user_error, user_written, user_skipped, user_error, aae_written, aae_skipped) " # noqa
"VALUES "
"(:datetime, :filename, :exported, :new, :updated, :skipped, :exif_updated, :touched, :converted_to_jpeg, :sidecar_xmp, :sidecar_json, :sidecar_exiftool, :missing, :error, :exiftool_warning, :exiftool_error, :extended_attributes_written, :extended_attributes_skipped, :cleanup_deleted_file, :cleanup_deleted_directory, :exported_album, :report_id, :sidecar_user, :sidecar_user_error, :user_written, :user_skipped, :user_error);", # noqa
"(:datetime, :filename, :exported, :new, :updated, :skipped, :exif_updated, :touched, :converted_to_jpeg, :sidecar_xmp, :sidecar_json, :sidecar_exiftool, :missing, :error, :exiftool_warning, :exiftool_error, :extended_attributes_written, :extended_attributes_skipped, :cleanup_deleted_file, :cleanup_deleted_directory, :exported_album, :report_id, :sidecar_user, :sidecar_user_error, :user_written, :user_skipped, :user_error, :aae_written, :aae_skipped);", # noqa
data,
)
self._conn.commit()
Expand Down Expand Up @@ -306,6 +308,19 @@ def _create_tables(self):
)
self._conn.commit()

# migrate report table for aae_written and aae_skipped
if "aae_written" not in sqlite_columns(self._conn, "report"):
self._conn.cursor().execute(
"ALTER TABLE report ADD COLUMN aae_written TEXT;"
)
self._conn.commit()

if "aae_skipped" not in sqlite_columns(self._conn, "report"):
self._conn.cursor().execute(
"ALTER TABLE report ADD COLUMN aae_skipped TEXT;"
)
self._conn.commit()

# create report_summary view
c.execute(
"""
Expand Down Expand Up @@ -396,6 +411,8 @@ def prepare_export_results_for_writing(
"user_written": false,
"user_skipped": false,
"user_error": "",
"aae_written": false,
"aae_skipped": false,
}

for result in export_results.exported:
Expand Down Expand Up @@ -492,6 +509,12 @@ def prepare_export_results_for_writing(
for result in export_results.user_error:
all_results[str(result[0])]["user_error"] = result[1]

for result in export_results.aae_written:
all_results[str(result)]["aae_written"] = true

for result in export_results.aae_skipped:
all_results[str(result)]["aae_skipped"] = true

return all_results


Expand Down Expand Up @@ -644,7 +667,7 @@ def _create_tables(self):
c.execute(
"""
CREATE TABLE IF NOT EXISTS report (
report_id TEXT,
report_id TEXT,
uuid TEXT,
filename TEXT,
fingerprint TEXT,
Expand Down Expand Up @@ -1002,7 +1025,7 @@ def _create_tables(self):
c.execute(
"""
CREATE TABLE IF NOT EXISTS report (
report_id TEXT,
report_id TEXT,
uuid TEXT,
original_filename TEXT,
filename TEXT,
Expand Down

0 comments on commit 5417731

Please sign in to comment.