Update state_manager.py

state_manager.py

@@ -19,6 +19,7 @@ from Bubble_API_Calls import (
 from config import (
     DEFAULT_INITIAL_FETCH_COUNT, DEFAULT_POSTS_UPDATE_FETCH_COUNT,
     BUBBLE_POST_DATE_COLUMN_NAME, BUBBLE_POSTS_TABLE_NAME,
+    BUBBLE_POST_STATS_TABLE_NAME,
     BUBBLE_MENTIONS_TABLE_NAME, BUBBLE_MENTIONS_DATE_COLUMN_NAME,
     BUBBLE_FOLLOWER_STATS_TABLE_NAME, FOLLOWER_STATS_TYPE_COLUMN, FOLLOWER_STATS_CATEGORY_COLUMN,
     LINKEDIN_CLIENT_ID_ENV_VAR,
@@ -130,7 +131,8 @@ def process_and_store_bubble_token(url_user_token, org_urn, token_state):
             "bubble_posts_df": BUBBLE_POSTS_TABLE_NAME,
             "bubble_mentions_df": BUBBLE_MENTIONS_TABLE_NAME,
             "bubble_follower_stats_df": BUBBLE_FOLLOWER_STATS_TABLE_NAME,
-            "bubble_operations_log_df": BUBBLE_OPERATIONS_LOG_TABLE_NAME # NEW
+            "bubble_operations_log_df": BUBBLE_OPERATIONS_LOG_TABLE_NAME, # NEW
+            "bubble_post_stats_df": BUBBLE_POST_STATS_TABLE_NAME
         }
         for state_key, table_name in data_tables_to_fetch.items():
             logging.info(f"Attempting to fetch {table_name} from Bubble for org_urn: {current_org_urn}")