Skip to content

Commit 911ff88

Browse files
committed
fix: check if Linkedin url is not None after googlesearch
1 parent 51941dc commit 911ff88

File tree

1 file changed

+24
-21
lines changed

1 file changed

+24
-21
lines changed

src/core/modules/opendata/workflows/LinkedInOrganizationsWorkflows.py

+24-21
Original file line numberDiff line numberDiff line change
@@ -79,33 +79,36 @@ def get_organization_info(self, parameters: LinkedInOrganizationParameters) -> d
7979
linkedin_url = self.__google_search_integration.search_linkedin_organization_url(parameters.organization_name)
8080
logger.debug(f"LinkedIn URL: {linkedin_url}")
8181

82-
logger.info(f"Extracting organization data from LinkedInIntegration for '{parameters.organization_name}'")
83-
df = self.__linkedin_integration.get_organization_details(linkedin_url)
84-
if len(df) > 0:
85-
data = {
86-
"organization_name": parameters.organization_name,
87-
"organization_id": create_id_from_string(parameters.organization_name),
88-
"linkedin_url": linkedin_url,
89-
"data": df.to_dict('records'),
90-
"output_dir": str(output_dir),
91-
"date_extracted": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
92-
}
93-
logger.info(f"Data successfully extracted: {data}")
94-
else:
95-
logger.error(f"No data found for '{parameters.organization_name}'")
96-
97-
# Save data to storage
98-
services.storage_service.put_object(
99-
prefix=output_dir,
100-
key=file_name,
101-
content=json.dumps(data, indent=4, ensure_ascii=False).encode("utf-8")
102-
)
82+
if linkedin_url is not None:
83+
logger.info(f"Extracting organization data from LinkedInIntegration for '{parameters.organization_name}'")
84+
df = self.__linkedin_integration.get_organization_details(linkedin_url)
85+
if len(df) > 0:
86+
data = {
87+
"organization_name": parameters.organization_name,
88+
"organization_id": create_id_from_string(parameters.organization_name),
89+
"linkedin_url": linkedin_url,
90+
"data": df.to_dict('records'),
91+
"output_dir": str(output_dir),
92+
"date_extracted": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
93+
}
94+
logger.info(f"Data successfully extracted: {data}")
95+
else:
96+
logger.error(f"No data found for '{parameters.organization_name}'")
97+
98+
# Save data to storage
99+
services.storage_service.put_object(
100+
prefix=output_dir,
101+
key=file_name,
102+
content=json.dumps(data, indent=4, ensure_ascii=False).encode("utf-8")
103+
)
103104
return data
104105

105106
def get_organization_logo_url(self, parameters: LinkedInOrganizationParameters) -> str:
106107
"""Get organization logo URL from the OpenData store."""
107108
# Initialize storage
108109
data = self.get_organization_info(parameters)
110+
if data is None:
111+
return None
109112
logo_url = data.get("data", [{}])[0].get("LOGO_URL")
110113
output_dir = data.get("output_dir")
111114
organization_id = data.get("organization_id")

0 commit comments

Comments
 (0)