feat: add PostgreSQL availability checks and environment variables for dual-write and sync services

Author: AnujChhikara

.github/workflows/test.yml

@@ -14,6 +14,8 @@ jobs:
       ALLOWED_HOSTS: "localhost,127.0.0.1"
       MONGODB_URI: mongodb://db:27017
       DB_NAME: todo-app
+      POSTGRES_SYNC_ENABLED: "False"
+      DUAL_WRITE_ENABLED: "False"
       GOOGLE_OAUTH_CLIENT_ID: "test-client-id"
       GOOGLE_OAUTH_CLIENT_SECRET: "test-client-secret"
       GOOGLE_OAUTH_REDIRECT_URI: "http://localhost:8000/v1/auth/google/callback"

todo/services/dual_write_service.py

@@ -44,6 +44,13 @@ class DualWriteService:
 
     def __init__(self):
         self.sync_failures = []
+        try:
+            from django.db import connection
+
+            connection.ensure_connection()
+            self.postgres_available = True
+        except Exception:
+            self.postgres_available = False
 
     def create_document(self, collection_name: str, data: Dict[str, Any], mongo_id: str) -> bool:
         """
@@ -58,6 +65,11 @@ def create_document(self, collection_name: str, data: Dict[str, Any], mongo_id:
             bool: True if both writes succeeded, False otherwise
         """
         try:
+            # Check if PostgreSQL is available
+            if not self.postgres_available:
+                logger.debug("PostgreSQL not available, skipping Postgres sync")
+                return True
+
             # First, write to MongoDB (this should already be done by the calling code)
             # Then, write to Postgres
             postgres_model = self._get_postgres_model(collection_name)

todo/services/postgres_sync_service.py

@@ -30,6 +30,13 @@ def sync_all_tables(self) -> bool:
         if not self.enabled:
             logger.info("PostgreSQL sync is disabled, skipping")
             return True
+        try:
+            from django.db import connection
+
+            connection.ensure_connection()
+        except Exception as e:
+            logger.warning(f"PostgreSQL not available, skipping sync: {str(e)}")
+            return True
 
         logger.info("Starting PostgreSQL table synchronization for labels and roles")
         logger.info(f"PostgreSQL sync enabled: {self.enabled}")
@@ -151,7 +158,11 @@ def _sync_labels_table(self) -> bool:
             for label in labels:
                 try:
                     # Check if label already exists in PostgreSQL
-                    from todo.models.postgres.label import PostgresLabel
+                    try:
+                        from todo.models.postgres.label import PostgresLabel
+                    except Exception:
+                        logger.warning("PostgreSQL models not available, skipping sync")
+                        return False
 
                     existing = PostgresLabel.objects.filter(mongo_id=str(label["_id"])).first()
                     if existing:
@@ -203,7 +214,11 @@ def _sync_roles_table(self) -> bool:
             synced_count = 0
             for role in roles:
                 try:
-                    from todo.models.postgres.role import PostgresRole
+                    try:
+                        from todo.models.postgres.role import PostgresRole
+                    except Exception:
+                        logger.warning("PostgreSQL models not available, skipping sync")
+                        return False
 
                     existing = PostgresRole.objects.filter(mongo_id=str(role["_id"])).first()
                     if existing:

todo_project/settings/test.py

@@ -1,10 +1,12 @@
 from .base import *
+import os
 
 DUAL_WRITE_ENABLED = False
+POSTGRES_SYNC_ENABLED = False
 
 # Remove PostgreSQL database configuration for tests
-# This prevents Django from trying to connect to PostgreSQL
 DATABASES = {}
 
-# Use MongoDB only for tests
-# The tests will use testcontainers to spin up their own MongoDB instance
+
+os.environ.setdefault("POSTGRES_SYNC_ENABLED", "False")
+os.environ.setdefault("DUAL_WRITE_ENABLED", "False")