graceful shutdown

foodie_automator_google.py

@@ -27,6 +27,20 @@ from foodie_utils import (
 )
 from foodie_hooks import get_dynamic_hook, select_best_cta
 
+# Flag to indicate if we're in the middle of posting
+is_posting = False
+
+def signal_handler(sig, frame):
+    logging.info("Received termination signal, checking if safe to exit...")
+    if is_posting:
+        logging.info("Currently posting, will exit after completion.")
+    else:
+        logging.info("Safe to exit immediately.")
+        sys.exit(0)
+
+signal.signal(signal.SIGTERM, signal_handler)
+signal.signal(signal.SIGINT, signal_handler)
+
 logger = logging.getLogger()
 logger.setLevel(logging.INFO)
 file_handler = logging.FileHandler('/tmp/foodie_automator_google_trends.log', mode='a')
@@ -231,23 +245,10 @@ def curate_from_google_trends(geo_list=['US']):
             cta = select_best_cta(post_data["title"], final_summary, post_url=None)
             post_data["content"] = f"{final_summary}\n\n{cta}"
 
-            post_id, post_url = post_to_wp(
-                post_data=post_data,
-                category=category,
-                link=link,
-                author=author,
-                image_url=image_url,
-                original_source=original_source,
-                image_source=image_source,
-                uploader=uploader,
-                pixabay_url=pixabay_url,
-                interest_score=interest_score
-            )
-
-            if post_id:
-                cta = select_best_cta(post_data["title"], final_summary, post_url=post_url)
-                post_data["content"] = f"{final_summary}\n\n{cta}"
-                post_to_wp(
+            global is_posting
+            is_posting = True
+            try:
+                post_id, post_url = post_to_wp(
                     post_data=post_data,
                     category=category,
                     link=link,
@@ -257,9 +258,31 @@ def curate_from_google_trends(geo_list=['US']):
                     image_source=image_source,
                     uploader=uploader,
                     pixabay_url=pixabay_url,
-                    interest_score=interest_score,
-                    post_id=post_id
+                    interest_score=interest_score
                 )
+            finally:
+                is_posting = False
+
+            if post_id:
+                cta = select_best_cta(post_data["title"], final_summary, post_url=post_url)
+                post_data["content"] = f"{final_summary}\n\n{cta}"
+                is_posting = True
+                try:
+                    post_to_wp(
+                        post_data=post_data,
+                        category=category,
+                        link=link,
+                        author=author,
+                        image_url=image_url,
+                        original_source=original_source,
+                        image_source=image_source,
+                        uploader=uploader,
+                        pixabay_url=pixabay_url,
+                        interest_score=interest_score,
+                        post_id=post_id
+                    )
+                finally:
+                    is_posting = False
 
                 timestamp = datetime.now(timezone.utc).isoformat()
                 save_json_file(POSTED_TITLES_FILE, title, timestamp)