From ce6d806c8992fdab3e087048e92d93ad9bb2b4de Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lukas=20L=C3=B6sche?= <lukas@some.engineering>
Date: Thu, 19 Sep 2024 21:41:53 +0200
Subject: [PATCH] Add more failsafes

---
 fixattiosync/attiodata.py | 4 ++++
 fixattiosync/fixdata.py   | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/fixattiosync/attiodata.py b/fixattiosync/attiodata.py
index 17b66e0..3bde845 100644
--- a/fixattiosync/attiodata.py
+++ b/fixattiosync/attiodata.py
@@ -1,3 +1,4 @@
+import sys
 import os
 import requests
 from uuid import UUID
@@ -118,6 +119,9 @@ def hydrate(self) -> None:
         self.__people = self.__marshal(self._records("people"), AttioPerson)  # type: ignore
         self.__users = self.__marshal(self._records("users"), AttioUser)  # type: ignore
         self.__connect()
+        if len(self.__workspaces) == 0 or len(self.__people) == 0 or len(self.__users) == 0:
+            log.fatal("No data found in Attio")
+            sys.exit(3)
         self.hydrated = True
 
     def __connect(self) -> None:
diff --git a/fixattiosync/fixdata.py b/fixattiosync/fixdata.py
index 85e01cd..fd9dbc0 100644
--- a/fixattiosync/fixdata.py
+++ b/fixattiosync/fixdata.py
@@ -82,6 +82,9 @@ def hydrate(self) -> None:
                 self.close()
             log.debug(f"Found {len(self.__workspaces)} workspaces in database")
             log.debug(f"Found {len(self.__users)} users in database")
+            if len(self.__users) == 0 or len(self.__workspaces) == 0:
+                log.fatal("No data found in Fix database")
+                sys.exit(2)
             self.hydrated = True
 
     def close(self) -> None: