diff --git a/part-07-database/app/backend_pre_start.py b/part-07-database/app/backend_pre_start.py index 3363a41..0bbd681 100644 --- a/part-07-database/app/backend_pre_start.py +++ b/part-07-database/app/backend_pre_start.py @@ -20,8 +20,11 @@ def init() -> None: try: db = SessionLocal() - # Try to create session to check if DB is awake - db.execute("SELECT 1") + is_awake = db.is_active + if is_awake: + print("Session is awake") + else: + print("Session not awake") except Exception as e: logger.error(e) raise e diff --git a/part-08-structure-and-versioning/app/backend_pre_start.py b/part-08-structure-and-versioning/app/backend_pre_start.py index 3363a41..0bbd681 100644 --- a/part-08-structure-and-versioning/app/backend_pre_start.py +++ b/part-08-structure-and-versioning/app/backend_pre_start.py @@ -20,8 +20,11 @@ def init() -> None: try: db = SessionLocal() - # Try to create session to check if DB is awake - db.execute("SELECT 1") + is_awake = db.is_active + if is_awake: + print("Session is awake") + else: + print("Session not awake") except Exception as e: logger.error(e) raise e diff --git a/part-09-async/app/backend_pre_start.py b/part-09-async/app/backend_pre_start.py index 3363a41..0bbd681 100644 --- a/part-09-async/app/backend_pre_start.py +++ b/part-09-async/app/backend_pre_start.py @@ -20,8 +20,11 @@ def init() -> None: try: db = SessionLocal() - # Try to create session to check if DB is awake - db.execute("SELECT 1") + is_awake = db.is_active + if is_awake: + print("Session is awake") + else: + print("Session not awake") except Exception as e: logger.error(e) raise e diff --git a/part-10-jwt-auth/app/backend_pre_start.py b/part-10-jwt-auth/app/backend_pre_start.py index 3363a41..0bbd681 100644 --- a/part-10-jwt-auth/app/backend_pre_start.py +++ b/part-10-jwt-auth/app/backend_pre_start.py @@ -20,8 +20,11 @@ def init() -> None: try: db = SessionLocal() - # Try to create session to check if DB is awake - db.execute("SELECT 1") + is_awake = db.is_active + if is_awake: + print("Session is awake") + else: + print("Session not awake") except Exception as e: logger.error(e) raise e diff --git a/part-11-dependency-injection/app/backend_pre_start.py b/part-11-dependency-injection/app/backend_pre_start.py index 3363a41..0bbd681 100644 --- a/part-11-dependency-injection/app/backend_pre_start.py +++ b/part-11-dependency-injection/app/backend_pre_start.py @@ -20,8 +20,11 @@ def init() -> None: try: db = SessionLocal() - # Try to create session to check if DB is awake - db.execute("SELECT 1") + is_awake = db.is_active + if is_awake: + print("Session is awake") + else: + print("Session not awake") except Exception as e: logger.error(e) raise e diff --git a/part-12-react-frontend/backend/app/backend_pre_start.py b/part-12-react-frontend/backend/app/backend_pre_start.py index 3363a41..0bbd681 100644 --- a/part-12-react-frontend/backend/app/backend_pre_start.py +++ b/part-12-react-frontend/backend/app/backend_pre_start.py @@ -20,8 +20,11 @@ def init() -> None: try: db = SessionLocal() - # Try to create session to check if DB is awake - db.execute("SELECT 1") + is_awake = db.is_active + if is_awake: + print("Session is awake") + else: + print("Session not awake") except Exception as e: logger.error(e) raise e diff --git a/part-13-docker-deployment/backend/app/app/backend_pre_start.py b/part-13-docker-deployment/backend/app/app/backend_pre_start.py index 3363a41..0bbd681 100644 --- a/part-13-docker-deployment/backend/app/app/backend_pre_start.py +++ b/part-13-docker-deployment/backend/app/app/backend_pre_start.py @@ -20,8 +20,11 @@ def init() -> None: try: db = SessionLocal() - # Try to create session to check if DB is awake - db.execute("SELECT 1") + is_awake = db.is_active + if is_awake: + print("Session is awake") + else: + print("Session not awake") except Exception as e: logger.error(e) raise e diff --git a/part-14-send-email-in-background/backend/app/app/backend_pre_start.py b/part-14-send-email-in-background/backend/app/app/backend_pre_start.py index 3363a41..0bbd681 100644 --- a/part-14-send-email-in-background/backend/app/app/backend_pre_start.py +++ b/part-14-send-email-in-background/backend/app/app/backend_pre_start.py @@ -20,8 +20,11 @@ def init() -> None: try: db = SessionLocal() - # Try to create session to check if DB is awake - db.execute("SELECT 1") + is_awake = db.is_active + if is_awake: + print("Session is awake") + else: + print("Session not awake") except Exception as e: logger.error(e) raise e