Update linkedin_follower_stats.py

linkedin_follower_stats.py

@@ -224,12 +224,17 @@ def _parse_urn_to_id(urn_string):
 
 # --- Follower Data Fetching Functions ---
 
-def fetch_monthly_follower_gains(session, org_urn, api_rest_base_url):
+def fetch_monthly_follower_gains(comm_client_id, token_dict, org_urn, api_rest_base_url):
+    """
+    Fetches monthly follower gains using a dedicated OAuth2Session for REST API calls.
+    """
     now_utc = datetime.now(timezone.utc)
+    # Calculate start of the month, approximately 12 months ago
     start_of_reporting_period = (now_utc - timedelta(days=365)).replace(day=1, hour=0, minute=0, second=0, microsecond=0)
     start_ms = int(start_of_reporting_period.timestamp() * 1000)
 
     base_url = f"{api_rest_base_url}/organizationalEntityFollowerStatistics"
+    # Using the same timeIntervals format as in the original failing log, as it's standard for REST
     time_intervals_value = f"(timeRange:(start:{start_ms}),timeGranularityType:MONTH)"
 
     api_params = {
@@ -238,30 +243,45 @@ def fetch_monthly_follower_gains(session, org_urn, api_rest_base_url):
         "timeIntervals": time_intervals_value
     }
     
-    logging.info(f"Preparing to fetch monthly follower gains for {org_urn}.")
+    logging.info(f"Preparing to fetch monthly follower gains for {org_urn} using a dedicated session.")
     logging.debug(f"API Parameters for monthly gains: {json.dumps(api_params)}")
+    logging.debug(f"Received client_id for dedicated session: {comm_client_id}")
+    logging.debug(f"Received token_dict for dedicated session (access_token type): {type(token_dict.get('access_token')) if token_dict else 'N/A'}")
+
 
     results = []
     request_url_for_logging = "Not constructed" 
     response_obj = None 
 
     try:
-        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]}...")
+        # Create a new session specifically for this REST API call
+        # This session will not have X-Restli-Protocol-Version header by default from OAuth2Session
+        # It will also not inherit any potentially problematic headers from a shared session.
+        rest_session = OAuth2Session(client_id=comm_client_id, token=token_dict)
+        rest_session.headers.update({
+            "LinkedIn-Version": LINKEDIN_API_VERSION,
+            "Accept-Language": "en_US" 
+            # DO NOT ADD "X-Restli-Protocol-Version": "2.0.0" here for /rest/ endpoints
+        })
+        logging.info(f"Dedicated REST session created for monthly gains (id: {id(rest_session)}).")
+        logging.debug(f"Dedicated REST session token (type): {type(rest_session.token)}, Authorized: {rest_session.authorized}")
+        if rest_session.token and 'access_token' in rest_session.token:
+             logging.debug(f"Dedicated REST session access token (partial): {str(rest_session.token['access_token'])[:20]}...")
         else:
-            logging.warning("fetch_monthly_follower_gains: session.token is None or 'access_token' key is missing before prepare_request.")
+            logging.warning("Dedicated REST session: token is None or 'access_token' key is missing.")
+
 
         req = requests.Request('GET', base_url, params=api_params)
-        prepared_req = session.prepare_request(req)
+        prepared_req = rest_session.prepare_request(req) # Use the dedicated session
         request_url_for_logging = prepared_req.url
         
         logging.info(f"Requesting monthly follower gains from URL: {request_url_for_logging}")
-        logging.debug(f"Request Headers for monthly gains: {json.dumps(dict(prepared_req.headers), indent=2)}")
+        # Log only essential headers, avoid logging full token if present in headers directly
+        headers_to_log = {k: v for k, v in prepared_req.headers.items() if k.lower() != 'authorization'}
+        logging.debug(f"Request Headers for monthly gains (excluding Authorization): {json.dumps(headers_to_log, indent=2)}")
 
-        response_obj = session.send(prepared_req, timeout=30) 
+
+        response_obj = rest_session.send(prepared_req, timeout=30) # Use the dedicated session
         response_obj.raise_for_status() 
         data = response_obj.json()
         
@@ -513,7 +533,7 @@ def get_linkedin_follower_stats(comm_client_id, community_token, org_urn):
 
     all_follower_data = []
 
-    monthly_gains = fetch_monthly_follower_gains(session, org_urn, API_REST_BASE) # Pass session
+    monthly_gains = fetch_monthly_follower_gains(comm_client_id, token_dict, org_urn, API_REST_BASE)
     all_follower_data.extend(monthly_gains)
 
     demographics = fetch_follower_demographics(session, org_urn, functions_map, seniorities_map) # Pass session