fix: optimize message push settings

Author: ihmily

app/core/record_manager.py

@@ -272,7 +272,9 @@ async def check_if_live(self, recording: Recording):
 
                 msg_manager = MessagePusher(self.settings)
                 user_config = self.settings.user_config
-                if MessagePusher.should_push_message(self.settings, recording, message_type='start'):
+                
+                if (MessagePusher.should_push_message(self.settings, recording, message_type='start')
+                        and not recording.notified_live_start):
                     push_content = self._["push_content"]
                     begin_push_message_text = user_config.get("custom_stream_start_content")
                     if begin_push_message_text:
@@ -286,20 +288,28 @@ async def check_if_live(self, recording: Recording):
                     msg_title = msg_title or self._["status_notify"]
 
                     self.app.page.run_task(msg_manager.push_messages, msg_title, push_content)
-
-                if not user_config.get("only_notify_no_record"):
+                    recording.notified_live_start = True
+  
+                if not recording.only_notify_no_record:
                     recording.loop_time_seconds = self.loop_time_seconds
                     self.start_update(recording)
                     self.app.page.run_task(recorder.start_recording, stream_info)
                 else:
-                    notify_loop_time = user_config.get("notify_loop_time")
-                    recording.loop_time_seconds = int(notify_loop_time or 3600)
+                    if recording.notified_live_start:
+                        notify_loop_time = user_config.get("notify_loop_time")
+                        recording.loop_time_seconds = int(notify_loop_time or 3600)
+                    else:
+                        recording.loop_time_seconds = self.loop_time_seconds
                     recording.status_info = RecordingStatus.NOT_RECORDING
                     recording.is_checking = False
 
                 self.app.page.run_task(self.app.record_card_manager.update_card, recording)
                 self.app.page.pubsub.send_others_on_topic("update", recording)
             else:
+                if not recording.is_live and recording.notified_live_start:
+                    recording.notified_live_start = False
+                    recording.notified_live_end = False
+                
                 recording.is_checking = False
                 recording.status_info = RecordingStatus.MONITORING
                 title = f"{stream_info.anchor_name or recording.streamer_name} - {self._[recording.quality]}"

app/core/stream_manager.py

@@ -292,7 +292,7 @@ async def start_ffmpeg(
                     user_config = self.settings.user_config
 
                     if self.app.recording_enabled and MessagePusher.should_push_message(
-                            self.settings, self.recording, check_manually_stopped=True, message_type='end'):
+                            self.settings, self.recording, check_manually_stopped=True, message_type='end') and not self.recording.notified_live_end:
                         push_content = self._["push_content_end"]
                         end_push_message_text = user_config.get("custom_stream_end_content")
                         if end_push_message_text:
@@ -306,6 +306,7 @@ async def start_ffmpeg(
                         msg_title = msg_title or self._["status_notify"]
 
                         self.app.page.run_task(msg_manager.push_messages, msg_title, push_content)
+                        self.recording.notified_live_end = True
 
                     self.recording.is_recording = False
                 try:

app/models/recording_model.py

@@ -16,7 +16,8 @@ def __init__(
         scheduled_start_time,
         monitor_hours,
         recording_dir,
-        enabled_message_push
+        enabled_message_push,
+        only_notify_no_record
     ):
         """
         Initialize a recording object.
@@ -34,6 +35,7 @@ def __init__(
         :param monitor_hours: Number of hours to monitor from the scheduled recording start time, e.g., 3.
         :param recording_dir: Directory path where the recorded files will be saved.
         :param enabled_message_push: Whether to enable message push.
+        :param only_notify_no_record: Whether to only notify when no record is made.
         """
 
         self.rec_id = rec_id
@@ -49,6 +51,7 @@ def __init__(
         self.monitor_hours = monitor_hours
         self.recording_dir = recording_dir
         self.enabled_message_push = enabled_message_push
+        self.only_notify_no_record = only_notify_no_record
         self.scheduled_time_range = None
         self.title = f"{streamer_name} - {self.quality}"
         self.speed = "X KB/s"
@@ -89,6 +92,7 @@ def to_dict(self):
             "enabled_message_push": self.enabled_message_push,
             "platform": self.platform,
             "platform_key": self.platform_key,
+            "only_notify_no_record": self.only_notify_no_record
         }
 
     @classmethod
@@ -108,6 +112,7 @@ def from_dict(cls, data):
             data.get("monitor_hours"),
             data.get("recording_dir"),
             data.get("enabled_message_push"),
+            data.get("only_notify_no_record")
         )
         recording.title = data.get("title", recording.title)
         recording.display_title = data.get("display_title", recording.title)

app/ui/components/card_dialog.py

@@ -18,7 +18,7 @@ def __init__(self, app, recording):
 
     def load(self):
         language = self.app.language_manager.language
-        for key in ("recording_card", "recording_manager", "base", "video_quality"):
+        for key in ("recording_card", "recording_manager", "base", "video_quality", "recording_dialog"):
             self._.update(language.get(key, {}))
 
     def get_content(self, recording):
@@ -40,6 +40,7 @@ def get_content(self, recording):
         from ...messages.message_pusher import MessagePusher
         should_push_message = MessagePusher.should_push_message(self.app.settings, recording)
         message_push = self._["enabled"] if should_push_message else self._["disabled"]
+        only_notify_no_record = self._["enabled"] if recording.only_notify_no_record else self._["disabled"]
 
         dialog_content = ft.Column(
             [
@@ -56,6 +57,7 @@ def get_content(self, recording):
                 ft.Text(f"{self._['scheduled_recording']}: {scheduled_recording_status}", size=14),
                 ft.Text(f"{self._['scheduled_time_range']}: {scheduled_time_range}", size=14),
                 ft.Text(f"{self._['message_push']}: {message_push}", size=14),
+                ft.Text(f"{self._['only_notify_no_record']}: {only_notify_no_record}", size=14),
                 ft.Text(f"{self._['save_path']}: {save_path}", size=14, selectable=True),
                 ft.Text(f"{self._['recording_status']}: {recording_status_info}", size=14),
             ],

app/ui/components/recording_dialog.py

@@ -35,6 +35,7 @@ async def show_dialog(self):
         video_segment_time = user_config.get('video_segment_time', 1800)
         segment_record = initial_values.get("segment_record", segmented_recording_enabled)
         segment_time = initial_values.get("segment_time", video_segment_time)
+        only_notify_no_record = user_config.get("only_notify_no_record", False)
 
         async def on_url_change(_):
             """Enable or disable the submit button based on whether the URL field is filled."""
@@ -221,6 +222,18 @@ async def handle_change(_):
             width=500,
         )
 
+        no_record_dropdown = ft.Dropdown(
+            label=self._["only_notify_no_record"],
+            options=[
+                ft.dropdown.Option("true", self._["yes"]),
+                ft.dropdown.Option("false", self._["no"]),
+            ],
+            border_radius=5,
+            filled=False,
+            value="true" if only_notify_no_record else "false",
+            width=500,
+        )
+
         hint_text_dict = {
             "en": "Example:\n0，https://v.douyin.com/AbcdE，nickname1\n0，https://v.douyin.com/EfghI，nickname2\n\nPS: "
             "0=original image or Blu ray, 1=ultra clear, 2=high-definition, 3=standard definition, 4=smooth\n",
@@ -268,7 +281,8 @@ async def handle_change(_):
                                 scheduled_setting_dropdown,
                                 schedule_and_monitor_row,
                                 monitor_hours_input,
-                                message_push_dropdown
+                                message_push_dropdown,
+                                no_record_dropdown
                             ],
                             tight=True,
                             spacing=10,
@@ -324,7 +338,8 @@ async def on_confirm(e):
                         "scheduled_start_time": str(scheduled_start_time_input.value),
                         "monitor_hours": monitor_hours_input.value,
                         "recording_dir": recording_dir_field.value,
-                        "enabled_message_push": message_push_dropdown.value == "true"
+                        "enabled_message_push": message_push_dropdown.value == "true",
+                        "only_notify_no_record": no_record_dropdown.value == "true"
                     }
                 ]
                 await self.on_confirm_callback(recordings_info)

app/ui/views/home_view.py

@@ -491,7 +491,8 @@ async def add_recording(self, recordings_info):
                     scheduled_start_time=recording_info["scheduled_start_time"],
                     monitor_hours=recording_info["monitor_hours"],
                     recording_dir=recording_info["recording_dir"],
-                    enabled_message_push=recording_info["enabled_message_push"]
+                    enabled_message_push=recording_info["enabled_message_push"],
+                    only_notify_no_record=recording_info["only_notify_no_record"]
                 )
             else:
                 recording = Recording(
@@ -507,7 +508,8 @@ async def add_recording(self, recordings_info):
                     scheduled_start_time=user_config.get("scheduled_start_time"),
                     monitor_hours=user_config.get("monitor_hours"),
                     recording_dir=None,
-                    enabled_message_push=False
+                    enabled_message_push=False,
+                    only_notify_no_record=user_config.get("only_notify_no_record"),
                 )
 
             platform, platform_key = get_platform_info(recording.url)

locales/en.json

@@ -56,6 +56,7 @@
     "scheduled_start_time": "Daily Monitoring Start Time",
     "monitor_hours": "Daily Monitoring Hours",
     "enable_message_push": "Enable Message Push",
+    "only_notify_no_record": "Only notify without recording",
     "batch_input_tip": "Batch Input (one record per line)",
     "single_input": "Single Input",
     "batch_input": "Batch Input",

locales/zh_CN.json

@@ -56,6 +56,7 @@
     "scheduled_start_time": "每日监听开始时间",
     "monitor_hours": "每日监听小时数",
     "enable_message_push": "是否开启消息推送",
+    "only_notify_no_record": "仅通知不录制",
     "batch_input_tip": "批量录入（每行一条记录）",
     "single_input": "单个录入",
     "batch_input": "批量录入",