Skip to content

Commit 0cf76de

Browse files
committed
fix: Make notice daily fetcher being hard failing
1 parent fa46d7e commit 0cf76de

2 files changed

Lines changed: 6 additions & 7 deletions

File tree

dags/pipelines/notice_fetcher_pipelines.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ def notice_fetcher_by_date_pipeline(date_wild_card: str = None) -> List[str]:
1111
from ted_sws.supra_notice_manager.services.daily_supra_notice_manager import \
1212
create_and_store_in_mongo_db_daily_supra_notice
1313
from ted_sws.event_manager.services.log import log_error
14-
notice_ids = None
1514
try:
1615
date_wild_card = date_wild_card if date_wild_card else (datetime.now() - timedelta(days=1)).strftime("%Y%m%d*")
1716
notice_publication_date = datetime.strptime(date_wild_card, "%Y%m%d*").date()
@@ -24,6 +23,7 @@ def notice_fetcher_by_date_pipeline(date_wild_card: str = None) -> List[str]:
2423
ted_publication_date=notice_publication_date)
2524
except Exception as error:
2625
log_error(message=str(error))
26+
raise error
2727

2828
return notice_ids
2929

@@ -35,7 +35,6 @@ def notice_fetcher_by_query_pipeline(query: str = None) -> List[str]:
3535
from ted_sws.notice_fetcher.adapters.ted_api import TedAPIAdapter, TedRequestAPI
3636
from ted_sws.notice_fetcher.services.notice_fetcher import NoticeFetcher
3737
from ted_sws.event_manager.services.log import log_error
38-
notice_ids = None
3938
try:
4039
ted_api_query = {"query": query}
4140
mongodb_client = MongoClient(config.MONGO_DB_AUTH_URL)
@@ -44,4 +43,5 @@ def notice_fetcher_by_query_pipeline(query: str = None) -> List[str]:
4443
request_api=TedRequestAPI())).fetch_notices_by_query(query=ted_api_query)
4544
except Exception as error:
4645
log_error(message=str(error))
46+
raise error
4747
return notice_ids

ted_sws/notice_fetcher/adapters/ted_api.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -72,11 +72,10 @@ def __call__(self, api_url: str, api_query: dict) -> dict:
7272
headers = get_configured_custom_headers(CUSTOM_HEADER)
7373
response = execute_request_with_retries(
7474
request_lambda=lambda: requests.post(api_url, json=api_query, headers=headers))
75-
if response.ok:
76-
response_content = json.loads(response.text)
77-
return response_content
78-
else:
79-
raise Exception(f"The TED-API call failed with: {response}")
75+
response.raise_for_status()
76+
response_content = json.loads(response.text)
77+
return response_content
78+
8079

8180

8281
class TedAPIAdapter(TedAPIAdapterABC):

0 commit comments

Comments
 (0)