initial commit

src/sync.py

@@ -0,0 +1,94 @@
+#!/usr/bin/env python3
+"""Einstiegspunkt für einen Sync-Lauf über alle konfigurierten Accounts.
+Für jeden Account wird ein Delta-Sync per CardDAV sync-collection (RFC 6578)
+durchgeführt. Beim allerersten Lauf eines Accounts (kein gespeicherter
+sync-token) sowie nach einem vom Server abgelehnten Token erfolgt ein
+vollständiger Re-Sync."""
+import logging
+import sys
+
+from config import Config
+from carddav_client import CardDAVClient, ICLOUD_BASE_URL, SyncTokenInvalid
+from vcard_parser import parse_vcard
+import db
+
+logging.basicConfig(level=Config.LOG_LEVEL, format="%(asctime)s [%(levelname)s] %(message)s")
+logger = logging.getLogger("sync")
+
+
+def sync_account(conn, account, href_to_uid_cache: dict):
+    client = CardDAVClient(ICLOUD_BASE_URL, account.apple_email, account.apple_app_password)
+    collection_url = client.discover_collection()
+
+    stored_token = db.get_sync_token(conn, account.name)
+    sync_type = "delta" if stored_token else "initial"
+    run_id = db.start_sync_run(conn, account.name, sync_type)
+
+    try:
+        if not stored_token:
+            logger.info("[%s] Kein sync-token vorhanden, führe initialen Full-Sync aus", account.name)
+            raw_vcards = client.fetch_all_vcards(collection_url)
+            contacts = [c for c in (parse_vcard(v, account.name) for v in raw_vcards) if c]
+            db.replace_all_contacts_for_account(conn, account.name, contacts, run_id)
+            _, _, new_token = client.sync_collection(collection_url, None)
+            if new_token:
+                db.save_sync_token(conn, account.name, new_token)
+            db.finish_sync_run(conn, run_id, "success", upserted=len(contacts), deleted=0)
+            logger.info("[%s] Initialer Sync abgeschlossen: %d Kontakte", account.name, len(contacts))
+            return
+
+        try:
+            changed_vcards, deleted_hrefs, new_token = client.sync_collection(collection_url, stored_token)
+        except SyncTokenInvalid:
+            logger.warning("[%s] sync-token vom Server abgelehnt, führe vollen Re-Sync aus", account.name)
+            db.clear_sync_token(conn, account.name)
+            raw_vcards = client.fetch_all_vcards(collection_url)
+            contacts = [c for c in (parse_vcard(v, account.name) for v in raw_vcards) if c]
+            db.replace_all_contacts_for_account(conn, account.name, contacts, run_id)
+            _, _, new_token = client.sync_collection(collection_url, None)
+            if new_token:
+                db.save_sync_token(conn, account.name, new_token)
+            db.finish_sync_run(conn, run_id, "success", upserted=len(contacts), deleted=0)
+            logger.info("[%s] Re-Sync abgeschlossen: %d Kontakte", account.name, len(contacts))
+            return
+
+        contacts = [c for c in (parse_vcard(v, account.name) for v in changed_vcards) if c]
+        db.upsert_contacts(conn, contacts, run_id)
+
+        deleted_uids = [href.rstrip("/").rsplit("/", 1)[-1].replace(".vcf", "") for href in deleted_hrefs]
+        db.delete_contacts_by_href_uids(conn, account.name, deleted_uids)
+
+        if new_token:
+            db.save_sync_token(conn, account.name, new_token)
+
+        db.finish_sync_run(conn, run_id, "success", upserted=len(contacts), deleted=len(deleted_uids))
+        logger.info(
+            "[%s] Delta-Sync abgeschlossen: %d geändert/neu, %d gelöscht",
+            account.name, len(contacts), len(deleted_uids),
+        )
+    except Exception as exc:
+        logger.exception("[%s] Sync-Lauf %s fehlgeschlagen", account.name, run_id)
+        db.finish_sync_run(conn, run_id, "failed", error_message=str(exc))
+        raise
+
+
+def main() -> int:
+    try:
+        Config.validate_db()
+        accounts = Config.load_accounts()
+    except RuntimeError as exc:
+        logger.error(str(exc))
+        return 1
+
+    exit_code = 0
+    with db.get_connection() as conn:
+        for account in accounts:
+            try:
+                sync_account(conn, account, {})
+            except Exception:
+                exit_code = 1
+    return exit_code
+
+
+if __name__ == "__main__":
+    sys.exit(main())