From 484ab4921a5f5ac9778d4aaa2eb837240bc26a02 Mon Sep 17 00:00:00 2001 From: srlee056 Date: Thu, 29 Feb 2024 22:54:57 +0900 Subject: [PATCH] [hot-fix]: change follower dag --- dags/streaming/follower.py | 21 +++++++++------------ 1 file changed, 9 insertions(+), 12 deletions(-) diff --git a/dags/streaming/follower.py b/dags/streaming/follower.py index b16278a..f7d7e55 100644 --- a/dags/streaming/follower.py +++ b/dags/streaming/follower.py @@ -95,18 +95,15 @@ def get_broad_info(bjid, headers): broad_res = get_broad_info(bjid, headers) try: - broad_info = broad_res.get("broad") - - if broad_info: - combined_res = {"broad_info": broad_res} - - stream_data = { - "streamer_id": s_id, - "afc_s_id": bjid, - "current_time": current_time, - **combined_res, - } - live_stream_data.append(stream_data) + combined_res = {"broad_info": broad_res} + + stream_data = { + "streamer_id": s_id, + "afc_s_id": bjid, + "current_time": current_time, + **combined_res, + } + live_stream_data.append(stream_data) except (AttributeError, TypeError) as e: error_msg = f"Error occurred: {str(e)}"