github/DouyinLiveRecorder
1
0
Fork 0
mirror of https://github.com/ihmily/DouyinLiveRecorder.git synced 2025-12-26 05:48:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: ensure thread safety in file update operation

Browse Source
This commit is contained in:
ihmily 2024-04-23 22:05:09 +08:00
parent a7e0a5779f
commit 795063fc06
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
main.py
View File

@ -4,7 +4,7 @@
Author: Hmily
GitHub: https://github.com/ihmily
Date: 2023-07-17 23:52:05
Update: 2024-04-23 21:34:55
Update: 2024-04-23 22:02:49
Copyright (c) 2023-2024 by Hmily, All Rights Reserved.
Function: Record live stream video.
"""
@ -158,19 +158,20 @@ def display_info():
logger.error(f"错误信息: {e} 发生错误的行数: {e.__traceback__.tb_lineno}")
def update_file(file_path: str, old_str: str, new_str: str, start_str: str = None):
# TODO: 更新文件操作
file_data = ""
with open(file_path, "r", encoding="utf-8-sig") as f:
for text_line in f:
if old_str in text_line:
text_line = text_line.replace(old_str, new_str)
if start_str:
text_line = f'{start_str}{text_line}'
file_data += text_line
with open(file_path, "w", encoding="utf-8-sig") as f:
f.write(file_data)
def update_file(file_path: str, old_str: str, new_str: str, start_str: str = None, lock: threading.Lock = None):
if lock is None:
lock = threading.Lock()
with lock:
file_data = ""
with open(file_path, "r", encoding="utf-8-sig") as f:
for text_line in f:
if old_str in text_line:
text_line = text_line.replace(old_str, new_str)
if start_str:
text_line = f'{start_str}{text_line}'
file_data += text_line
with open(file_path, "w", encoding="utf-8-sig") as f:
f.write(file_data)
def converts_mp4(address: str):
@ -1847,4 +1848,4 @@ while True:
first_run = False
time.sleep(3)
time.sleep(3)