Update linkedin_follower_stats.py

linkedin_follower_stats.py

@@ -4,7 +4,7 @@ import requests
 import logging
 from datetime import datetime, timezone, timedelta
 from urllib.parse import quote, urlencode
-# from dateutil.relativedelta import relativedelta # For more precise month arithmetic if needed
+import requests_oauthlib # For version logging
 
 # Assuming you have a sessions.py with create_session
 # If sessions.py or create_session is not found, it will raise an ImportError,
@@ -15,7 +15,7 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 
 API_V2_BASE = 'https://api.linkedin.com/v2'
 API_REST_BASE = "https://api.linkedin.com/rest"
-LINKEDIN_API_VERSION = "202502" # As per user's example for follower stats
+LINKEDIN_API_VERSION = "202502"
 
 # --- ID to Name Mapping Helper Functions ---
 
@@ -27,7 +27,9 @@ def _fetch_linkedin_names(session, url, params, result_key_path, name_key_path,
     request_url_for_logging = url 
     response_obj = None 
     try:
-        logging.debug(f"_fetch_linkedin_names: About to prepare request. Session token: {session.token}, Session authorized: {session.authorized}")
+        logging.debug(f"_fetch_linkedin_names (id: {id(session)}): About to prepare request. Session token: {session.token}, Session authorized: {session.authorized}, Expires at: {session.token.get('expires_at') if session.token else 'N/A'}")
+        logging.debug(f"_fetch_linkedin_names (id: {id(session)}): Session auth object: type={type(session.auth)}, value={session.auth}")
+
         req = requests.Request('GET', url, params=params)
         prepared_req = session.prepare_request(req) 
         request_url_for_logging = prepared_req.url
@@ -133,7 +135,8 @@ def get_industries_map(session, industry_urns, version="DEFAULT"):
     response_obj = None
     logging.info(f"Fetching all industries (to filter {len(unique_ids)} IDs)")
     try:
-        logging.debug(f"get_industries_map: About to prepare request. Session token: {session.token}, Session authorized: {session.authorized}")
+        logging.debug(f"get_industries_map (id: {id(session)}): About to prepare request. Session token: {session.token}, Session authorized: {session.authorized}, Expires at: {session.token.get('expires_at') if session.token else 'N/A'}")
+        logging.debug(f"get_industries_map (id: {id(session)}): Session auth object: type={type(session.auth)}, value={session.auth}")
         req = requests.Request('GET', url, params=params)
         prepared_req = session.prepare_request(req)
         request_url_for_logging = prepared_req.url
@@ -243,7 +246,9 @@ def fetch_monthly_follower_gains(session, org_urn, api_rest_base_url):
     response_obj = None 
 
     try:
-        logging.debug(f"fetch_monthly_follower_gains: About to prepare request. Session token: {session.token}, Session authorized: {session.authorized}")
+        logging.debug(f"fetch_monthly_follower_gains (id: {id(session)}): About to prepare request. Session token: {session.token}, Session authorized: {session.authorized}, Expires at: {session.token.get('expires_at') if session.token else 'N/A'}")
+        logging.debug(f"fetch_monthly_follower_gains (id: {id(session)}): Session auth object: type={type(session.auth)}, value={session.auth}")
+        logging.debug(f"fetch_monthly_follower_gains (id: {id(session)}): Auto-refresh URL: {session.auto_refresh_url}, Auto-refresh kwargs: {session.auto_refresh_kwargs}")
         if session.token and 'access_token' in session.token:
             logging.debug(f"fetch_monthly_follower_gains: Access token (partial): {str(session.token['access_token'])[:20]}...")
         else:
@@ -331,7 +336,9 @@ def fetch_follower_demographics(session, org_urn, functions_map, seniorities_map
     response_obj = None
 
     try:
-        logging.debug(f"fetch_follower_demographics: About to prepare request. Session token: {session.token}, Session authorized: {session.authorized}")
+        logging.debug(f"fetch_follower_demographics (id: {id(session)}): About to prepare request. Session token: {session.token}, Session authorized: {session.authorized}, Expires at: {session.token.get('expires_at') if session.token else 'N/A'}")
+        logging.debug(f"fetch_follower_demographics (id: {id(session)}): Session auth object: type={type(session.auth)}, value={session.auth}")
+        logging.debug(f"fetch_follower_demographics (id: {id(session)}): Auto-refresh URL: {session.auto_refresh_url}, Auto-refresh kwargs: {session.auto_refresh_kwargs}")
         if session.token and 'access_token' in session.token:
             logging.debug(f"fetch_follower_demographics: Access token (partial): {str(session.token['access_token'])[:20]}...")
         else:
@@ -449,6 +456,7 @@ def fetch_follower_demographics(session, org_urn, functions_map, seniorities_map
 
 def get_linkedin_follower_stats(comm_client_id, community_token, org_urn):
     logging.info(f"--- Initiating get_linkedin_follower_stats for org: {org_urn} ---")
+    logging.info(f"requests-oauthlib version: {requests_oauthlib.__version__}") # Log version
     logging.debug(f"Received comm_client_id: {comm_client_id}")
     logging.debug(f"Received community_token - Type: {type(community_token)}, IsSet: {bool(community_token)}")
     if isinstance(community_token, str) and len(community_token) > 10:
@@ -480,8 +488,11 @@ def get_linkedin_follower_stats(comm_client_id, community_token, org_urn):
             "LinkedIn-Version": LINKEDIN_API_VERSION, 
             "Accept-Language": "en_US" 
         })
-        logging.info(f"Session created and headers updated for org {org_urn}.")
-        logging.debug(f"get_linkedin_follower_stats: Session token after creation and header update: {session.token}, Session authorized: {session.authorized}")
+        logging.info(f"Session (id: {id(session)}) created and headers updated for org {org_urn}.")
+        logging.debug(f"get_linkedin_follower_stats (id: {id(session)}): Session token after creation: {session.token}, Session authorized: {session.authorized}, Expires at: {session.token.get('expires_at') if session.token else 'N/A'}")
+        logging.debug(f"get_linkedin_follower_stats (id: {id(session)}): Session auth object: type={type(session.auth)}, value={session.auth}")
+        logging.debug(f"get_linkedin_follower_stats (id: {id(session)}): Auto-refresh URL: {session.auto_refresh_url}, Auto-refresh kwargs: {session.auto_refresh_kwargs}")
+
         if session.token and 'access_token' in session.token:
              logging.debug(f"get_linkedin_follower_stats: Access token in session (partial): {str(session.token['access_token'])[:20]}...")
         else:
@@ -492,20 +503,20 @@ def get_linkedin_follower_stats(comm_client_id, community_token, org_urn):
         logging.error(f"Failed to create session or update headers for org {org_urn}: {e}", exc_info=True)
         return []
 
-    logging.info(f"Starting follower stats retrieval for org: {org_urn}")
+    logging.info(f"Starting follower stats retrieval for org: {org_urn} using session (id: {id(session)})")
 
-    functions_map = get_functions_map(session)
-    seniorities_map = get_seniorities_map(session)
+    functions_map = get_functions_map(session) # Pass session
+    seniorities_map = get_seniorities_map(session) # Pass session
     
     if not functions_map: logging.warning(f"Functions map is empty for org {org_urn}.")
     if not seniorities_map: logging.warning(f"Seniorities map is empty for org {org_urn}.")
 
     all_follower_data = []
 
-    monthly_gains = fetch_monthly_follower_gains(session, org_urn, API_REST_BASE)
+    monthly_gains = fetch_monthly_follower_gains(session, org_urn, API_REST_BASE) # Pass session
     all_follower_data.extend(monthly_gains)
 
-    demographics = fetch_follower_demographics(session, org_urn, functions_map, seniorities_map)
+    demographics = fetch_follower_demographics(session, org_urn, functions_map, seniorities_map) # Pass session
     all_follower_data.extend(demographics)
 
     if not all_follower_data: