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: