文件操作系统调用_壹

本文深入讲解了Linux系统中文件操作的基础知识,包括open、read、write等核心函数的使用方法及注意事项,并提供了一个检索文件中特定字符串并打印所在行的示例程序。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

目录:

1:概述

2:open、read、write、lseek、close

3:在文件中检索字符串,打印字符串所在的行并且返回文件的大小


1:概述

linux体现了“一切都是文件”的设计思想,所有的文件操作都是通过操作文件描述符来进行的,文件描述符是一个整形的变量,系统默认最大是1024,可以使用ulimit -a命令查询或者修改,系统已经分配的三个标准文件描述符,标准输入(stdin)、标准输出(stdout)、标准错误(stderr),分别为0,1,2,文件描述符为负值,说明出现错误;


2:open、read、write、lseek、close

2.1:open()

函数声明:

int open(const char *pathname, int flags);

int open(const char *payhname,int flags, mode_t mode);

open()函数依据标志flags和权限mode在pathname下创建或者打开一个文件,成功返回文件描述符,错误返回-1;

flags:表示以什么方式打开文件,常用的是O_RDWR/O_APPEND/O_NOBLOCK,分别对应读写、追加、非阻塞,如果O_CREAT,需要设置mode参数;

mode:表示创建文件的权限和ID,使用5个数字来表示文件的各种权限,分别为:用户ID、组ID、用户权限、组权限、其他人权限;如:10777;

2.2:read()和write()

函数声明:

int read(int fd,const void *buf,size_t length);

int write(int fd,const void *buf,size_t length);

两个函数的返回值都是读取或者写入的字节数;

2.3:lseek()

函数声明:

int lseek(int fd,offset_t offset, int whence);

lseek函数将文件读写指针相对whence移动offset字节,offset为正值,文件读写指针向后移动,为负值,文件读写指针向前移动,函数返回文件指针相对于文件头的位置;

whence可以用如些宏代替:

SEEK_SET:相对文件头;

SEEK_CUR:相对文件读写指针的当前位置;

SEEK_END:相对文件末尾;

例如:

将文件读写指针移动到文件末尾:lseek(fd,0,SEEK_END),函数返回文件指针相对于头文件的位置,也就是文件的长度;

2.4:close()

函数声明:

int close(int fd);

所有文件的操作只有等到执行close后才能操作成功,所以,一定要调用close函数关闭文件;


3:在文件中检索字符串,打印字符串所在的行并且返回文件的大小

说明:代码简单的使用read函数,每次读取文件中的一个字节,然后判断是否是换行符(在linux系统下换行符为'\n'),如果是,将这行打印出来,最后返回文件的总大小;代码中每次是读取一字节的数据,所以本代码的执行效率很低,只为学习实践;

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define LENGTH 1
#define B  0
#define KB 1
#define MB 2
#define GB 3

char buff[150];
long long size = 0;
unsigned char flag = 0;

int main(int argc,char **argv)
{
	char index = 0;
	int fd = 0;

	if(argc != 4)
	{
		printf("usage: ./exe pathname string B/KB/MB/KB\r\n");
		exit(0);
	}
	fd = open(argv[1],O_RDWR);   /*文件已经存在,按照读写的方式打开文件*/
	if(fd < 0)
		perror("file is not found:\r\n"); /*打开文件不成功,返回系统错误*/

	while(read(fd,buff+index,LENGTH) > 0)  /*每次读取一个字节*/
	{
		if(buff[index++] == '\n')          /*当遇到换行符,对取出来出来的这行数据进行字符串比较*/
		{                                  
			buff[index] = '\0';
			if(strstr(buff,argv[2]) != NULL)  /*找到字符串,将所在的行进行打印*/
				printf("%s\r\n",buff);
			index = 0;

		}
		if(index > 149)  
		{
			printf("too large,no enough free\r\n");
			exit(0);
		}
		size++;
	}
	/*单位转换*/
	if(memcmp(argv[3],"B",1) == 0)
		flag = 0;
	else
	if(memcmp(argv[3],"KB",2) == 0)
		flag = 1;
	else
	if(memcmp(argv[3],"MB",2) == 0)
		flag = 2;
	else
	if(memcmp(argv[3],"GB",2) == 0)
		flag = 3;

	switch(flag)
	{
		case B:
			printf("file size is: %lld Bytes\r\n",size);
			break;
		case KB:
			printf("file size is: %.6f KB\r\n",size/1024.0);
			break;
		case MB:
			printf("file size is: %.6f MB\r\n",size/(1024.0*1024.0));
			break;
		case GB:
			printf("file size is: %.6f GB\r\n",size/(1024.0*1024.0*1024.0));
			break;
	}

	close(fd);
	return 0;
}



import json import requests import os import time import logging from datetime import datetime, timedelta import pytz from mutagen.mp3 import MP3 from mutagen.mp4 import MP4 import subprocess import platform # 配置日志 logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', handlers=[ logging.FileHandler("broadcast.log"), logging.StreamHandler() ] ) logger = logging.getLogger(__name__) def get_broadcast_data_with_token(tenant_access_token): """ 使用新的 token 获取飞书的数据 """ # 获取 Feishu Bitable 数据 url = 'https://open.feishu.cn/open-apis/bitable/v1/apps/QCPUbGid0aBjqLsZDWMcYetinRg/tables/tblMFWB9leKGDxnF/records/search' headers = { 'Content-Type': 'application/json', 'Authorization': f'Bearer {tenant_access_token}' # 使用新的 token } data = {} # 如果需要传递查询条件,可以在这里添加 try: logger.info(f"正在请求飞书数据,URL: {url}") response = requests.post(url, headers=headers, json=data, timeout=30) response.raise_for_status() # 如果响应失败,将抛出异常 response_dict = response.json() # 将返回的 JSON 数据转换为字典 items = response_dict.get("data", {}).get("items", []) logger.info(f"成功获取飞书数据,共 {len(items)} 条记录") data = [] for item in items: fields = item.get("fields", {}) data.append({ "播音日期": extract_broadcast_date(fields, '播音日期'), "时间段": extract_time_segment(fields, '时间段'), "开播音乐file_token": extract_file_token(fields, '开播音乐'), "开场白-播报file_token": extract_file_token(fields, '开场白-播报'), "需更新文案-播报file_token": extract_file_token(fields, '需更新文案-播报'), "首歌file_token": extract_file_token(fields, '首歌'), "需更新文案2-播报file_token": extract_file_token(fields, '需更新文案2-播报'), "贰首歌file_token": extract_file_token(fields, '贰首歌'), "结束语-播报file_token": extract_file_token(fields, '结束语-播报'), "结束音乐file_token": extract_file_token(fields, '结束音乐') }) return data except requests.exceptions.HTTPError as http_err: logger.error(f"HTTP 错误发生: {http_err}") except requests.exceptions.Timeout: logger.error("请求超时,服务器响应时间过长") except requests.exceptions.ConnectionError: logger.error("连接错误,无法连接到服务器") except Exception as err: logger.error(f"其他错误发生: {err}", exc_info=True) return [] def extract_file_token(fields, field_name): """提取 file_token""" field_data = fields.get(field_name, []) if isinstance(field_data, list) and len(field_data) > 0: value = field_data[0] if isinstance(value, dict): return value.get("file_token", "") return '' def extract_time_segment(fields, field_name): """提取时间段字段""" field_data = fields.get(field_name, []) if isinstance(field_data, list) and len(field_data) > 0: value = field_data[0] if isinstance(value, dict): return value.get("text", "") return None def extract_broadcast_date(fields, field_name): """提取播音日期字段""" field_data = fields.get(field_name, 0) if isinstance(field_data, int): try: timestamp = field_data / 1000 # 时间戳转化为秒 parsed_date = datetime.fromtimestamp(timestamp, tz=pytz.utc).astimezone(pytz.timezone('Asia/Shanghai')) return parsed_date.strftime("%Y-%m-%d") # 转换为 "YYYY-MM-DD" 格式 except (ValueError, OverflowError): pass return None def get_auth_token(): """获取认证 token""" url = "https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal" headers = {"Content-Type": "application/json; charset=utf-8"} payload = {"app_id": "cli_a882683e8779d00c", "app_secret": "3NKkALA7vyMRVnpKJinmrb1LJ7YuK4H0"} try: logger.info("正在获取认证token") response = requests.post(url, json=payload, headers=headers, timeout=30) response.raise_for_status() data = response.json() if data["code"] == 0: logger.info("成功获取认证token") return data["tenant_access_token"] else: logger.error(f"请求失败:{data['msg']}(错误码:{data['code']})") except requests.exceptions.HTTPError as http_err: logger.error(f"HTTP 错误发生: {http_err}") except requests.exceptions.Timeout: logger.error("获取token超时") except requests.exceptions.ConnectionError: logger.error("连接错误,无法获取token") except Exception as e: logger.error(f"获取token异常:{e}", exc_info=True) return None def create_folder(folder_name): """创建文件夹""" if not os.path.exists(folder_name): logger.info(f"创建文件夹: {folder_name}") os.makedirs(folder_name) def download_file(file_token, save_path, authorization): """下载文件""" url = f"https://open.feishu.cn/open-apis/drive/v1/medias/{file_token}/download" headers = {"Authorization": "Bearer " + authorization} try: logger.info(f"开始下载文件: {file_token}") response = requests.get(url, headers=headers, stream=True, timeout=60) if response.status_code == 200: with open(save_path, 'wb') as f: for chunk in response.iter_content(chunk_size=8192): f.write(chunk) logger.info(f"文件已成功下载到: {save_path}") return True else: logger.error(f"请求失败,状态码: {response.status_code}") logger.error(f"错误信息: {response.text}") except requests.exceptions.Timeout: logger.error(f"下载文件超时: {file_token}") except requests.exceptions.ConnectionError: logger.error(f"连接错误,无法下载文件: {file_token}") except Exception as e: logger.error(f"下载文件发生异常: {str(e)}", exc_info=True) return False def get_audio_duration(file_path): """获取音频时长""" try: if file_path.endswith(".mp3"): audio = MP3(file_path) elif file_path.endswith(".mp4"): audio = MP4(file_path) else: logger.error(f"支持的文件格式: {file_path}") return 0 return audio.info.length except Exception as e: logger.error(f"获取音频时长失败: {e}", exc_info=True) return 0 def kill_previous_players(): """清理之前残留的播放器进程""" system = platform.system() # 获取当前操作系统类型 try: logger.info(f"清理之前残留的播放器进程,操作系统: {system}") if system == "Windows": subprocess.run(['taskkill', '/F', '/IM', 'wmplayer.exe'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) subprocess.run(['taskkill', '/F', '/IM', 'vlc.exe'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) subprocess.run(['taskkill', '/F', '/IM', 'Music.UI.exe'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) elif system == "Darwin": # macOS subprocess.run(['killall', 'afplay'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) subprocess.run(['killall', 'Music'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) elif system == "Linux": subprocess.run(['pkill', 'mpg123'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) subprocess.run(['pkill', 'vlc'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) logger.info("已清理之前残留的播放器进程") except Exception as e: logger.error(f"清理播放器进程时发生错误: {e}", exc_info=True) def play_music_in_folder(folder_path): """播放文件夹中的音频文件,并在播放完成后关闭播放器""" audio_files = [f for f in os.listdir(folder_path) if f.endswith((".mp3", ".mp4"))] processes = [] # 用于存储播放器进程 for file_name in audio_files: full_file_path = os.path.join(folder_path, file_name) try: duration = get_audio_duration(full_file_path) if duration <= 0: logger.error(f"无法获取 {file_name} 的时长,跳过播放") continue logger.info(f"播放 {file_name},预计播放时长:{duration} 秒") if os.name == 'nt': # Windows 系统 process = subprocess.Popen(['start', '', full_file_path], shell=True) elif os.name == 'posix': # MacOS 或 Linux 系统 process = subprocess.Popen(['afplay', full_file_path]) # 对于 MacOS 使用 afplay else: logger.error(f"支持的操作系统类型: {os.name}") continue processes.append(process) # 保存进程对象 time.sleep(duration + 1) # 等待音频播放完成,额外增加1秒缓冲 except Exception as e: logger.error(f"无法播放 {full_file_path}: {e}", exc_info=True) # 关闭所有播放器进程 for process in processes: try: if process.poll() is None: # 检查进程是否仍在运行 process.kill() # 强制终止进程 logger.info("播放器已强制关闭") except Exception as e: logger.error(f"关闭播放器失败: {e}", exc_info=True) def wait_until(target_time_str, target_name="目标时间"): """等待直到指定时间""" # 获取当前时间(上海时区) tz = pytz.timezone('Asia/Shanghai') now = datetime.now(tz) # 正确解析目标时间并设置时区 time_part = datetime.strptime(target_time_str, "%H:%M").time() naive_target_time = datetime.combine(now.date(), time_part) target_time = tz.localize(naive_target_time) # 如果目标时间已经过去,则将其设置为明天的同一时间 if target_time <= now: target_time = tz.localize(naive_target_time + timedelta(days=1)) # 计算需要等待的总秒数 total_wait_seconds = (target_time - now).total_seconds() logger.info(f"当前时间: {now.strftime('%H:%M:%S')}, 等待 {target_name}: {target_time.strftime('%H:%M:%S')}, 预计等待 {total_wait_seconds/60:.1f} 分钟") # 等待策略优化 if total_wait_seconds > 600: # 超过10分钟 # 先等待大部分时间 initial_wait = total_wait_seconds - 300 # 预留5分钟精确检查 logger.info(f"先等待 {initial_wait/60:.1f} 分钟,然后进行精确检查") time.sleep(initial_wait) # 剩余时间每秒检查一次 remaining_wait = total_wait_seconds - initial_wait logger.info(f"剩余 {remaining_wait:.1f} 秒,每秒检查一次") while datetime.now(tz) < target_time: time.sleep(1) else: # 短时间等待,每秒检查一次 while datetime.now(tz) < target_time: current_time = datetime.now(tz) elapsed = (current_time - now).total_seconds() # 每分钟输出一次日志 if int(elapsed) % 60 == 0: logger.info(f"已等待 {elapsed/60:.1f} 分钟,当前时间: {current_time.strftime('%H:%M:%S')}, 目标时间: {target_time.strftime('%H:%M:%S')}") time.sleep(1) logger.info(f"已到达 {target_name}: {target_time_str}") def process_time_segment(segment, download_offset, play_offset, data, authorization, folder_name): """处理一个时间段的下载和播放""" try: logger.info(f"开始处理时间段: {segment}") target_data = next((entry for entry in data if entry["时间段"] == segment), None) if not target_data: logger.error(f"未找到时间段 {segment} 的文件数据!") return current_time = datetime.now(pytz.timezone('Asia/Shanghai')).strftime('%H:%M') segment_start_time = segment.split("-")[0] logger.info(f"当前时间: {current_time}, 时间段开始时间: {segment_start_time}") # 如果当前时间已经超过该时间段的开始时间,则跳过 if current_time > segment_start_time: logger.warning(f"当前时间已超过时间段 {segment} 的开始时间,跳过该时间段的处理") return # 计算下载时间 download_time = (datetime.strptime(segment_start_time, "%H:%M") - timedelta(minutes=download_offset)).strftime("%H:%M") # 等待到达下载时间 wait_until(download_time, "下载时间") # 在下载前3分钟,重新获取新的 token new_authorization = get_auth_token() if not new_authorization: logger.error("获取新的 token 失败,无法继续执行操作。") return logger.info(f"使用新的 token 开始获取 {segment} 的数据") # 使用新的 token 获取飞书数据 new_data = get_broadcast_data_with_token(new_authorization) if not new_data: logger.error("获取新的飞书数据失败!") return # 获取当前时间段的数据 target_data = next((entry for entry in new_data if entry["时间段"] == segment), None) if not target_data: logger.error(f"未找到时间段 {segment} 的文件数据!") return logger.info(f"开始下载 {segment} 的文件") files = [] file_tokens = [ ("开播音乐file_token", "mp3"), ("开场白-播报file_token", "mp4"), ("需更新文案-播报file_token", "mp4"), ("首歌file_token", "mp3"), ("需更新文案2-播报file_token", "mp4"), ("贰首歌file_token", "mp3"), ("结束语-播报file_token", "mp4"), ("结束音乐file_token", "mp3") ] for i, (key, file_format) in enumerate(file_tokens): token = target_data.get(key) if token: save_path = os.path.join(folder_name, f"file_{i+1}.{file_format}") if download_file(token, save_path, new_authorization): files.append(save_path) # 清理之前残留的播放器进程 kill_previous_players() # 计算播放时间 play_time = (datetime.strptime(segment_start_time, "%H:%M") - timedelta(minutes=play_offset)).strftime("%H:%M") # 等待到达播放时间 wait_until(play_time, "播放时间") logger.info(f"开始播放 {segment} 的文件") play_music_in_folder(folder_name) # 播放结束后再次清理播放器进程 kill_previous_players() # 删除下载的文件 for file in files: try: os.remove(file) logger.info(f"已删除文件: {file}") except Exception as e: logger.error(f"删除文件失败: {e}") logger.info(f"成功完成时间段: {segment} 的处理") except Exception as e: logger.error(f"处理时间段 {segment} 时发生异常: {e}", exc_info=True) def main(): """主函数""" logger.info("===== 广播自动化程序启动 =====") authorization = get_auth_token() if not authorization: logger.error("获取认证token失败,程序退出") return # Get the broadcast data using the authorization token data = get_broadcast_data_with_token(authorization) if not data: logger.error("未获取到有效的数据!程序退出") return folder_name = "bobao" create_folder(folder_name) segments = [ ("08:10-08:15", 10, 0), # 提前10分钟下载文件,0-准点播放 ("10:30-10:40", 10, 0), ("13:00-13:10", 10, 0), ("15:00-15:10", 10, 0) ] for segment, download_offset, play_offset in segments: logger.info(f"===== 准备处理时间段: {segment} =====") process_time_segment(segment, download_offset, play_offset, data, authorization, folder_name) logger.info(f"===== 完成处理时间段: {segment} =====") # 主程序结束后,再清理一次播放器进程 kill_previous_players() logger.info("===== 广播自动化程序结束 =====") if __name__ == "__main__": main() 为什么上述代码每个时间段播放完后,可以关闭调用的Windows播放器的窗口,而下面代码却能实现? import json import os import logging from datetime import datetime, timedelta import pytz import requests from urllib.parse import quote import time from selenium import webdriver from selenium.webdriver.edge.service import Service from selenium.webdriver.common.by import By from selenium.webdriver.edge.options import Options from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from bs4 import BeautifulSoup import re import shutil import glob import asyncio import edge_tts import threading from mutagen.mp3 import MP3 from mutagen.mp4 import MP4 import subprocess import platform # 配置日志 logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', handlers=[ logging.FileHandler("broadcast.log"), logging.StreamHandler() ] ) logger = logging.getLogger(__name__) # 下载目录配置 DOWNLOAD_DIR = os.path.expanduser("~/Downloads") # 浏览器默认下载目录 TARGET_DIR = "./bobao" # 目标保存目录 def get_broadcast_data_with_token(tenant_access_token): """使用token获取飞书的数据""" # 获取 Feishu Bitable 数据 url = 'https://open.feishu.cn/open-apis/bitable/v1/apps/E1zybPqiqa0TaesZjKKch5ZcnJd/tables/tblwFY4k3pmrV5WK/records/search' headers = { 'Content-Type': 'application/json', 'Authorization': f'Bearer {tenant_access_token}' } data = {} try: logger.info(f"正在请求飞书数据,URL: {url}") response = requests.post(url, headers=headers, json=data, timeout=30) response.raise_for_status() response_dict = response.json() items = response_dict.get("data", {}).get("items", []) logger.info(f"成功获取飞书数据,共 {len(items)} 条记录") data = [] for item in items: fields = item.get("fields", {}) time_segment = extract_time_segment(fields, '时间段') song1_text = extract_text_field(fields, '歌曲-歌手') song2_text = extract_text_field(fields, '贰歌曲-歌手') # 新增:从文本字段获取播报内容 song1_broadcast = extract_text_field(fields, '需更新文案-播报') song2_broadcast = extract_text_field(fields, '需更新文案2-播报') if time_segment == "08:10-08:15": song1_name = "" song1_artist = "" song2_name = "" song2_artist = "" else: song1_parts = song1_text.split(' ', 1) song1_name = song1_parts[0] if len(song1_parts) > 0 else "" song1_artist = song1_parts[1] if len(song1_parts) > 1 else "" song2_parts = song2_text.split(' ', 1) song2_name = song2_parts[0] if len(song2_parts) > 0 else "" song2_artist = song2_parts[1] if len(song2_parts) > 1 else "" data.append({ "播音日期": extract_broadcast_date(fields, '播音日期'), "时间段": time_segment, "开播音乐file_token": extract_file_token(fields, '开播音乐'), "开场白-播报file_token": extract_file_token(fields, '开场白-播报'), "需更新文案-播报": song1_broadcast, # 保留文本内容 "需更新文案-播报file_token": extract_file_token(fields, '需更新文案-播报'), "歌曲-歌手": song1_text, "首歌file_token": extract_file_token(fields, '首歌'), "需更新文案2-播报": song2_broadcast, # 保留文本内容 "需更新文案2-播报file_token": extract_file_token(fields, '需更新文案2-播报'), "贰歌曲-歌手": song2_text, "贰首歌file_token": extract_file_token(fields, '贰首歌'), "结束语-播报file_token": extract_file_token(fields, '结束语-播报'), "结束音乐file_token": extract_file_token(fields, '结束音乐'), "歌名": song1_name, "歌手": song1_artist, "贰歌名": song2_name, "贰歌手": song2_artist, }) return data except requests.exceptions.HTTPError as http_err: logger.error(f"HTTP 错误发生: {http_err}") except requests.exceptions.Timeout: logger.error("请求超时,服务器响应时间过长") except requests.exceptions.ConnectionError: logger.error("连接错误,无法连接到服务器") except Exception as err: logger.error(f"其他错误发生: {err}", exc_info=True) return [] def extract_file_token(fields, field_name): """提取 file_token""" field_data = fields.get(field_name, []) if isinstance(field_data, list) and len(field_data) > 0: value = field_data[0] if isinstance(value, dict): return value.get("file_token", "") return '' def extract_text_field(fields, field_name): """提取文本字段内容""" field_data = fields.get(field_name, []) if isinstance(field_data, list) and len(field_data) > 0: value = field_data[0] if isinstance(value, dict): # 尝试获取text字段,如果没有则获取content return value.get("text", value.get("content", "")) return '' def extract_time_segment(fields, field_name): """提取时间段字段""" field_data = fields.get(field_name, []) if isinstance(field_data, list) and len(field_data) > 0: value = field_data[0] if isinstance(value, dict): return value.get("text", "") return None def extract_broadcast_date(fields, field_name): """提取播音日期字段""" field_data = fields.get(field_name, 0) if isinstance(field_data, int): try: timestamp = field_data / 1000 parsed_date = datetime.fromtimestamp(timestamp, tz=pytz.utc).astimezone(pytz.timezone('Asia/Shanghai')) return parsed_date.strftime("%Y-%m-%d") except (ValueError, OverflowError): pass return None def get_auth_token(): """获取认证 token""" url = "https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal" headers = {"Content-Type": "application/json; charset=utf-8"} payload = {"app_id": "cli_a882683e8779d00c", "app_secret": "3NKkALA7vyMRVnpKJinmrb1LJ7YuK4H0"} try: logger.info("正在获取认证token") response = requests.post(url, json=payload, headers=headers, timeout=30) response.raise_for_status() data = response.json() if data["code"] == 0: logger.info("成功获取认证token") return data["tenant_access_token"] else: logger.error(f"请求失败:{data['msg']}(错误码:{data['code']})") except requests.exceptions.HTTPError as http_err: logger.error(f"HTTP 错误发生: {http_err}") except requests.exceptions.Timeout: logger.error("获取token超时") except requests.exceptions.ConnectionError: logger.error("连接错误,无法获取token") except Exception as e: logger.error(f"获取token异常:{e}", exc_info=True) return None def display_data(data): """展示获取的数据""" if not data: print("没有获取到任何数据") return print(f"共获取到 {len(data)} 条广播数据记录:") for i, item in enumerate(data, 1): print(f"\n--- 记录 {i} ---") for key, value in item.items(): if key.endswith('file_token') and value: print(f"{key}: {value[:10]}...") else: print(f"{key}: {value}") def format_song_info(record): """格式化并打印歌曲信息""" print("\n歌曲信息:") print(f"歌名: {record['歌名']}") print(f"歌手: {record['歌手']}") print(f"贰歌名: {record['贰歌名']}") print(f"贰歌手: {record['贰歌手']}") # 打印播报内容 print("\n播报内容:") print(f"需更新文案-播报: {record['需更新文案-播报']}") print(f"需更新文案2-播报: {record['需更新文案2-播报']}") def wait_for_download(initial_files, timeout=60): """等待下载完成并返回新文件路径""" start_time = time.time() while time.time() - start_time < timeout: current_files = os.listdir(DOWNLOAD_DIR) new_files = [f for f in current_files if f not in initial_files] # 过滤掉临时下载文件 valid_files = [f for f in new_files if not f.endswith('.crdownload')] if valid_files: # 返回第一个找到的有效文件 return os.path.join(DOWNLOAD_DIR, valid_files[0]) time.sleep(1) print(f"等待下载超时({timeout}秒)") return None def download_song(song_name, artist=None, save_path=TARGET_DIR, custom_filename=None): """ 搜索并下载指定歌曲,下载后重命名为指定名称 参数: song_name (str): 歌曲名称 artist (str, optional): 歌手名称,默认为None save_path (str, optional): 保存路径,默认为"./bobao" custom_filename (str, optional): 自定义文件名,默认为None """ # 创建保存目录(如果存在) if not os.path.exists(save_path): os.makedirs(save_path) # 构建搜索关键词 search_query = song_name if artist: search_query += f" {artist}" # 去除.html后缀 search_query = search_query.replace('.html', '') print(f"正在搜索歌曲: {search_query}") # 记录下载前的文件列表 initial_files = os.listdir(DOWNLOAD_DIR) # 设置Edge浏览器选项 edge_options = Options() edge_options.add_argument('--disable-gpu') edge_options.add_argument('--no-sandbox') edge_options.add_argument('--disable-dev-shm-usage') # 指定EdgeDriver的路径 edge_driver_path = r"C:\Users\shaopeng.qi\Downloads\edgedriver_win64 (1)\msedgedriver.exe" # 检查路径是否存在 if not os.path.exists(edge_driver_path): print(f"错误: EdgeDriver路径存在 - {edge_driver_path}") print("请下载与您Edge浏览器版本匹配的EdgeDriver,并更新路径") print("下载地址: https://developer.microsoft.com/en-us/microsoft-edge/tools/webdriver/") return # 初始化WebDriver service = Service(executable_path=edge_driver_path) service.log_path = "edgedriver.log" driver = webdriver.Edge(service=service, options=edge_options) try: # 直接访问搜索结果页面 - 修正:去掉.html后缀 search_url = f"https://www.gequbao.com/s/{quote(search_query)}" print(f"正在访问搜索结果页面: {search_url}") driver.get(search_url) # 等待页面加载完成 WebDriverWait(driver, 10).until( EC.presence_of_element_located((By.CSS_SELECTOR, ".card-body")) ) # 检查是否有搜索结果 try: no_results = WebDriverWait(driver, 5).until( EC.presence_of_element_located((By.XPATH, "//div[contains(text(), '没有找到相关歌曲')]")) ) print("未找到匹配的歌曲") return except: # 没有找到"没有找到相关歌曲"的提示,继续执行 pass # 滚动到页面底部,确保所有元素都加载 driver.execute_script("window.scrollTo(0, document.body.scrollHeight);") time.sleep(1) # 查找第一个歌曲的链接 print("正在查找歌曲链接...") song_links = WebDriverWait(driver, 10).until( EC.presence_of_all_elements_located((By.CSS_SELECTOR, ".music-link")) ) if not song_links: print("未找到歌曲链接") with open("error_page.html", "w", encoding="utf-8") as f: f.write(driver.page_source) print("已保存当前页面源码到error_page.html,可用于分析问题") return # 获取第一个歌曲链接的href属性 first_song_link = song_links[0].get_attribute("href") print(f"找到歌曲链接: {first_song_link}") # 提取歌曲ID match = re.search(r'/music/(\d+)', first_song_link) if not match: print("无法从链接中提取歌曲ID") with open("error_page.html", "w", encoding="utf-8") as f: f.write(driver.page_source) print("已保存当前页面源码到error_page.html,可用于分析问题") return song_id = match.group(1) print(f"提取的歌曲ID: {song_id}") # 直接构建并访问歌曲详情页URL song_detail_url = f"https://www.gequbao.com/music/{song_id}" print(f"正在访问歌曲详情页: {song_detail_url}") # 打开新窗口 driver.execute_script(f"window.open('{song_detail_url}');") # 等待新窗口打开 time.sleep(2) # 检查是否打开了新窗口 if len(driver.window_handles) > 1: # 切换到新窗口 for window in driver.window_handles: if window != driver.current_window_handle: driver.switch_to.window(window) print("已切换到歌曲详情页") break else: print("未成功打开歌曲详情页,尝试直接访问") # 直接访问歌曲详情页 driver.get(song_detail_url) print("已直接访问歌曲详情页") # 等待页面加载 print("等待页面加载完成...") time.sleep(5) # 检查是否在歌曲详情页 try: # 查找歌曲标题元素,确认在详情页 song_title = WebDriverWait(driver, 5).until( EC.presence_of_element_located((By.CSS_SELECTOR, ".song-title")) ) print(f"已进入歌曲详情页,歌曲标题: {song_title.text}") except: # 如果找到歌曲标题元素,尝试其他可能的元素 try: # 尝试查找下载按钮 download_btn = WebDriverWait(driver, 3).until( EC.presence_of_element_located((By.XPATH, "//a[contains(text(), '下载')]")) ) print("已进入歌曲详情页,找到下载按钮") except: print("无法确认是否在歌曲详情页") with open("error_page.html", "w", encoding="utf-8") as f: f.write(driver.page_source) print("已保存当前页面源码到error_page.html,可用于分析问题") return # 点击下载按钮 print("正在查找并点击下载按钮...") try: download_btn = WebDriverWait(driver, 10).until( EC.element_to_be_clickable((By.XPATH, "//*[@id=\"btn-download-mp3\"]")) ) print("找到下载按钮") # 点击下载按钮 download_btn.click() print("已点击下载按钮") # 等待下载对话框出现 print("等待下载对话框出现...") time.sleep(3) # 点击低品质MP3按钮 - 使用更灵活的XPath选择器 print("正在查找并点击低品质MP3按钮...") try: # 使用通配符匹配动态ID部分 low_quality_btn = WebDriverWait(driver, 10).until( EC.element_to_be_clickable((By.XPATH, "//*[starts-with(@id, 'jconfirm-box')]/div/ul/li[2]/a")) ) print("找到低品质MP3按钮") # 点击低品质MP3按钮 low_quality_btn.click() # 构建保存文件名 if custom_filename: target_filename = f"{custom_filename}.mp3" elif artist: target_filename = f"{artist} - {song_name}.mp3" else: target_filename = f"{song_name}.mp3" target_path = os.path.join(save_path, target_filename) print(f"已点击下载按钮,等待下载完成后将保存为: {target_path}") # 等待下载完成 print("等待下载开始...") downloaded_file = wait_for_download(initial_files, timeout=60) if downloaded_file: # 等待文件完全下载(避免文件正在写入时移动) time.sleep(5) # 移动并重命名文件 try: shutil.move(downloaded_file, target_path) print(f"歌曲已成功下载并重命名为: {target_path}") return True except Exception as move_err: print(f"移动文件时出错: {move_err}") print(f"下载的文件保存在: {downloaded_file}") else: print("未检测到新的下载文件") except Exception as e: print(f"点击低品质MP3按钮时出错: {e}") with open("error_page.html", "w", encoding="utf-8") as f: f.write(driver.page_source) print("已保存当前页面源码到error_page.html,可用于分析问题") return False except Exception as e: print(f"点击下载按钮时出错: {e}") with open("error_page.html", "w", encoding="utf-8") as f: f.write(driver.page_source) print("已保存当前页面源码到error_page.html,可用于分析问题") return False except Exception as e: print(f"操作过程中发生错误: {e}") with open("error_page.html", "w", encoding="utf-8") as f: f.write(driver.page_source) print("已保存当前页面源码到error_page.html,可用于分析问题") return False finally: # 关闭浏览器 try: driver.quit() print("浏览器已关闭") except Exception as e: print(f"关闭浏览器时出错: {e}") def download_required_songs(data, folder_name): """下载指定的两首歌曲并命名为file_4和file_6""" if not data or len(data) == 0: print("没有可下载的歌曲数据") return first_record = data[0] # 下载第一首歌 (file_4) song1_name = first_record.get("歌名") song1_artist = first_record.get("歌手") success1 = False if song1_name and song1_artist: print(f"\n准备下载第一首歌: {song1_name} - {song1_artist},保存为 file_4.mp3") try: success1 = download_song(song1_name, song1_artist, folder_name, "file_4") except Exception as e: print(f"下载失败: {song1_name} - {song1_artist}, 错误: {e}") else: print("第一首歌信息完整,无法下载") print(f"歌名: {song1_name}, 歌手: {song1_artist}") # 下载第二首歌 (file_6) song2_name = first_record.get("贰歌名") song2_artist = first_record.get("贰歌手") success2 = False if song2_name and song2_artist: print(f"\n准备下载第二首歌: {song2_name} - {song2_artist},保存为 file_6.mp3") try: success2 = download_song(song2_name, song2_artist, folder_name, "file_6") except Exception as e: print(f"下载失败: {song2_name} - {song2_artist}, 错误: {e}") else: print("第二首歌信息完整,无法下载") print(f"歌名: {song2_name}, 歌手: {song2_artist}") return success1, success2 # 创建全局事件循环 loop = asyncio.new_event_loop() asyncio.set_event_loop(loop) async def convert_text_to_speech(text, output_file): """将文本转换为语音并保存为MP3文件""" if text: try: # 使用全局事件循环 communicate = edge_tts.Communicate(text, voice="zh-CN-YunyangNeural") await communicate.save(output_file) print(f"语音已保存为:{output_file}") return True except Exception as e: print(f"文本转语音失败: {e}") return False else: print(f"文本为空,跳过转换: {output_file}") return False def run_async_task(coro): """在独立线程中运行异步任务""" def run(): loop = asyncio.new_event_loop() asyncio.set_event_loop(loop) try: loop.run_until_complete(coro) finally: loop.close() thread = threading.Thread(target=run) thread.start() thread.join() async def convert_broadcast_texts_async(data, folder_name): """异步转换播报文本为语音文件""" if not data or len(data) == 0: print("没有可转换的播报文本数据") return first_record = data[0] # 确保目标目录存在 if not os.path.exists(folder_name): os.makedirs(folder_name) # 转换第一个播报文本 (file_3) text1 = first_record.get("需更新文案-播报", "") output_file1 = os.path.join(folder_name, "file_3.mp3") print(f"\n准备转换第一个播报文本,保存为 file_3.mp3") print(f"文本内容: {text1[:50]}...") # 只显示前50个字符 success1 = await convert_text_to_speech(text1, output_file1) if success1: print("第一个播报文本转换成功") # 转换第二个播报文本 (file_5) text2 = first_record.get("需更新文案2-播报", "") output_file2 = os.path.join(folder_name, "file_5.mp3") print(f"\n准备转换第二个播报文本,保存为 file_5.mp3") print(f"文本内容: {text2[:50]}...") # 只显示前50个字符 success2 = await convert_text_to_speech(text2, output_file2) if success2: print("第二个播报文本转换成功") return success1, success2 def convert_broadcast_texts(data, folder_name): """转换播报文本为语音文件 - 兼容同步和异步环境""" # 创建新线程运行异步任务,避免事件循环冲突 run_async_task(convert_broadcast_texts_async(data, folder_name)) def create_folder(folder_name): """创建文件夹""" if not os.path.exists(folder_name): logger.info(f"创建文件夹: {folder_name}") os.makedirs(folder_name) def download_file(file_token, save_path, authorization): """下载文件""" url = f"https://open.feishu.cn/open-apis/drive/v1/medias/{file_token}/download" headers = {"Authorization": "Bearer " + authorization} try: logger.info(f"开始下载文件: {file_token}") response = requests.get(url, headers=headers, stream=True, timeout=60) if response.status_code == 200: with open(save_path, 'wb') as f: for chunk in response.iter_content(chunk_size=8192): f.write(chunk) logger.info(f"文件已成功下载到: {save_path}") return True else: logger.error(f"请求失败,状态码: {response.status_code}") logger.error(f"错误信息: {response.text}") except requests.exceptions.Timeout: logger.error(f"下载文件超时: {file_token}") except requests.exceptions.ConnectionError: logger.error(f"连接错误,无法下载文件: {file_token}") except Exception as e: logger.error(f"下载文件发生异常: {str(e)}", exc_info=True) return False def get_audio_duration(file_path): """获取音频时长""" try: if file_path.endswith(".mp3"): audio = MP3(file_path) elif file_path.endswith(".mp4"): audio = MP4(file_path) else: logger.error(f"支持的文件格式: {file_path}") return 0 return audio.info.length except Exception as e: logger.error(f"获取音频时长失败: {e}", exc_info=True) return 0 def kill_previous_players(): """清理之前残留的播放器进程""" system = platform.system() # 获取当前操作系统类型 try: logger.info(f"清理之前残留的播放器进程,操作系统: {system}") if system == "Windows": subprocess.run(['taskkill', '/F', '/IM', 'wmplayer.exe'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) subprocess.run(['taskkill', '/F', '/IM', 'vlc.exe'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) subprocess.run(['taskkill', '/F', '/IM', 'Music.UI.exe'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) elif system == "Darwin": # macOS subprocess.run(['killall', 'afplay'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) subprocess.run(['killall', 'Music'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) elif system == "Linux": subprocess.run(['pkill', 'mpg123'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) subprocess.run(['pkill', 'vlc'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) logger.info("已清理之前残留的播放器进程") except Exception as e: logger.error(f"清理播放器进程时发生错误: {e}", exc_info=True) def play_music_in_folder(folder_path): """播放文件夹中的音频文件,并在播放完成后关闭播放器""" audio_files = [f for f in os.listdir(folder_path) if f.endswith((".mp3", ".mp4"))] processes = [] # 用于存储播放器进程 # 按照文件名排序 audio_files.sort() for file_name in audio_files: full_file_path = os.path.join(folder_path, file_name) try: duration = get_audio_duration(full_file_path) if duration <= 0: logger.error(f"无法获取 {file_name} 的时长,跳过播放") continue logger.info(f"播放 {file_name},预计播放时长:{duration} 秒") if os.name == 'nt': # Windows 系统 process = subprocess.Popen(['start', '', full_file_path], shell=True) elif os.name == 'posix': # MacOS 或 Linux 系统 process = subprocess.Popen(['afplay', full_file_path]) # 对于 MacOS 使用 afplay else: logger.error(f"支持的操作系统类型: {os.name}") continue processes.append(process) # 保存进程对象 time.sleep(duration + 1) # 等待音频播放完成,额外增加1秒缓冲 except Exception as e: logger.error(f"无法播放 {full_file_path}: {e}", exc_info=True) # 关闭所有播放器进程 for process in processes: try: if process.poll() is None: # 检查进程是否仍在运行 process.kill() # 强制终止进程 logger.info("播放器已强制关闭") except Exception as e: logger.error(f"关闭播放器失败: {e}", exc_info=True) def wait_until(target_time_str, target_name="目标时间"): """等待直到指定时间""" # 获取当前时间(上海时区) tz = pytz.timezone('Asia/Shanghai') now = datetime.now(tz) # 正确解析目标时间并设置时区 time_part = datetime.strptime(target_time_str, "%H:%M").time() naive_target_time = datetime.combine(now.date(), time_part) target_time = tz.localize(naive_target_time) # 如果目标时间已经过去,则将其设置为明天的同一时间 if target_time <= now: target_time = tz.localize(naive_target_time + timedelta(days=1)) # 计算需要等待的总秒数 total_wait_seconds = (target_time - now).total_seconds() logger.info(f"当前时间: {now.strftime('%H:%M:%S')}, 等待 {target_name}: {target_time.strftime('%H:%M:%S')}, 预计等待 {total_wait_seconds/60:.1f} 分钟") # 等待策略优化 if total_wait_seconds > 600: # 超过10分钟 # 先等待大部分时间 initial_wait = total_wait_seconds - 300 # 预留5分钟精确检查 logger.info(f"先等待 {initial_wait/60:.1f} 分钟,然后进行精确检查") time.sleep(initial_wait) # 剩余时间每秒检查一次 remaining_wait = total_wait_seconds - initial_wait logger.info(f"剩余 {remaining_wait:.1f} 秒,每秒检查一次") while datetime.now(tz) < target_time: time.sleep(1) else: # 短时间等待,每秒检查一次 while datetime.now(tz) < target_time: current_time = datetime.now(tz) elapsed = (current_time - now).total_seconds() # 每分钟输出一次日志 if int(elapsed) % 60 == 0: logger.info(f"已等待 {elapsed/60:.1f} 分钟,当前时间: {current_time.strftime('%H:%M:%S')}, 目标时间: {target_time.strftime('%H:%M:%S')}") time.sleep(1) logger.info(f"已到达 {target_name}: {target_time_str}") def process_time_segment(segment, download_offset, play_offset, data, authorization, folder_name): """处理一个时间段的下载和播放""" try: logger.info(f"开始处理时间段: {segment}") target_data = next((entry for entry in data if entry["时间段"] == segment), None) if not target_data: logger.error(f"未找到时间段 {segment} 的文件数据!") return current_time = datetime.now(pytz.timezone('Asia/Shanghai')).strftime('%H:%M') segment_start_time = segment.split("-")[0] logger.info(f"当前时间: {current_time}, 时间段开始时间: {segment_start_time}") # 如果当前时间已经超过该时间段的开始时间,则跳过 if current_time > segment_start_time: logger.warning(f"当前时间已超过时间段 {segment} 的开始时间,跳过该时间段的处理") return # 计算下载时间 download_time = (datetime.strptime(segment_start_time, "%H:%M") - timedelta(minutes=download_offset)).strftime("%H:%M") # 等待到达下载时间 wait_until(download_time, "下载时间") # 在下载前3分钟,重新获取新的 token new_authorization = get_auth_token() if not new_authorization: logger.error("获取新的 token 失败,无法继续执行操作。") return logger.info(f"使用新的 token 开始获取 {segment} 的数据") # 使用新的 token 获取飞书数据 new_data = get_broadcast_data_with_token(new_authorization) if not new_data: logger.error("获取新的飞书数据失败!") return # 获取当前时间段的数据 target_data = next((entry for entry in new_data if entry["时间段"] == segment), None) if not target_data: logger.error(f"未找到时间段 {segment} 的文件数据!") return logger.info(f"开始下载 {segment} 的文件") files = [] # 下载固定的file_token文件 file_tokens = [ ("开播音乐file_token", "mp3"), ("开场白-播报file_token", "mp3"), ("结束语-播报file_token", "mp3"), ("结束音乐file_token", "mp3") ] for i, (key, file_format) in enumerate(file_tokens): token = target_data.get(key) if token: # 前两个文件保持原命名,后两个文件命名为file_7和file_8 if i < 2: # 索引0和1(前两个文件) file_number = i + 1 else: # 索引2和3(后两个文件) file_number = i + 5 # 2+5=7, 3+5=8 save_path = os.path.join(folder_name, f"file_{file_number}.{file_format}") if download_file(token, save_path, new_authorization): files.append(save_path) # 尝试下载或生成播报文件 # 需更新文案-播报 (file_3) broadcast_text = target_data.get("需更新文案-播报") broadcast_token = target_data.get("需更新文案-播报file_token") save_path = os.path.join(folder_name, "file_3.mp3") if broadcast_text: # 如果有文本内容,优先使用文本转语音 print(f"使用文本转语音生成 file_3.mp3") try: # 使用改进的异步任务执行方法 run_async_task(convert_text_to_speech(broadcast_text, save_path)) files.append(save_path) except Exception as e: print(f"文本转语音失败: {e}") # 如果文本转语音失败,尝试使用file_token下载 if broadcast_token: print(f"尝试使用 file_token 下载 file_3.mp3") if download_file(broadcast_token, save_path, new_authorization): files.append(save_path) elif broadcast_token: # 如果没有文本内容但有file_token,直接下载 print(f"使用 file_token 下载 file_3.mp3") if download_file(broadcast_token, save_path, new_authorization): files.append(save_path) # 需更新文案2-播报 (file_5) broadcast_text2 = target_data.get("需更新文案2-播报") broadcast_token2 = target_data.get("需更新文案2-播报file_token") save_path2 = os.path.join(folder_name, "file_5.mp3") if broadcast_text2: # 如果有文本内容,优先使用文本转语音 print(f"使用文本转语音生成 file_5.mp3") try: # 使用改进的异步任务执行方法 run_async_task(convert_text_to_speech(broadcast_text2, save_path2)) files.append(save_path2) except Exception as e: print(f"文本转语音失败: {e}") # 如果文本转语音失败,尝试使用file_token下载 if broadcast_token2: print(f"尝试使用 file_token 下载 file_5.mp3") if download_file(broadcast_token2, save_path2, new_authorization): files.append(save_path2) elif broadcast_token2: # 如果没有文本内容但有file_token,直接下载 print(f"使用 file_token 下载 file_5.mp3") if download_file(broadcast_token2, save_path2, new_authorization): files.append(save_path2) # 下载两首歌曲 (调整为file_4和file_6) song1_token = target_data.get("首歌file_token") song2_token = target_data.get("贰首歌file_token") song1_name = target_data.get("歌名") song1_artist = target_data.get("歌手") song2_name = target_data.get("贰歌名") song2_artist = target_data.get("贰歌手") # 尝试下载歌曲 if song1_token: save_path = os.path.join(folder_name, "file_4.mp3") if download_file(song1_token, save_path, new_authorization): files.append(save_path) elif song1_name and song1_artist: # 如果没有file_token但有歌名和歌手,尝试从网络搜索下载 print(f"从网络搜索下载第一首歌: {song1_name} - {song1_artist}") if download_song(song1_name, song1_artist, folder_name, "file_4"): files.append(os.path.join(folder_name, "file_4.mp3")) if song2_token: save_path = os.path.join(folder_name, "file_6.mp3") if download_file(song2_token, save_path, new_authorization): files.append(save_path) elif song2_name and song2_artist: print(f"从网络搜索下载第二首歌: {song2_name} - {song2_artist}") if download_song(song2_name, song2_artist, folder_name, "file_6"): files.append(os.path.join(folder_name, "file_6.mp3")) if not files: logger.error(f"没有成功下载任何文件!") return # 按照数字顺序排序文件 files.sort(key=lambda x: int(re.search(r'file_(\d+)', x).group(1))) logger.info(f"所有文件下载完成,共下载 {len(files)} 个文件") # 计算播放时间 play_time = (datetime.strptime(segment_start_time, "%H:%M") - timedelta(minutes=play_offset)).strftime("%H:%M") # 等待到达播放时间 wait_until(play_time, "播放时间") # 清理之前残留的播放器进程 kill_previous_players() # 播放所有下载的文件 logger.info(f"开始播放 {segment} 的所有文件") play_music_in_folder(folder_name) logger.info(f"时间段 {segment} 处理完成") # 清理下载的文件 logger.info(f"开始清理时间段 {segment} 下载的文件") for file in files: if os.path.exists(file): os.remove(file) logger.info(f"已删除文件: {file}") else: logger.warning(f"文件存在,无法删除: {file}") logger.info(f"时间段 {segment} 文件清理完成") except Exception as e: logger.error(f"处理时间段 {segment} 时发生错误: {e}", exc_info=True) def process_day(data, authorization, process_past_segments=False): """处理一天的所有时间段""" if not data: logger.error("没有可处理的数据") return # 获取所有同的时间段 time_segments = sorted(list({item["时间段"] for item in data if item["时间段"]})) if not time_segments: logger.error("没有找到任何时间段数据") return logger.info(f"找到 {len(time_segments)} 个时间段需要处理") # 处理每个时间段 for segment in time_segments: # 创建时间段文件夹 segment_folder = os.path.join(TARGET_DIR, segment.replace(":", "")) create_folder(segment_folder) # 检查是否需要处理已过去的时间段 current_time = datetime.now(pytz.timezone('Asia/Shanghai')).strftime('%H:%M') segment_start_time = segment.split("-")[0] if not process_past_segments and current_time > segment_start_time: logger.warning(f"时间段 {segment} 已过去,跳过处理") continue # 处理时间段 process_time_segment(segment, download_offset=5, play_offset=0, data=data, authorization=authorization, folder_name=segment_folder) # 清理时间段文件夹(如果为空) if os.path.exists(segment_folder) and not os.listdir(segment_folder): os.rmdir(segment_folder) logger.info(f"已删除空文件夹: {segment_folder}") def main(): """主函数""" logger.info("===== 广播播放系统启动 =====") # 获取认证token authorization = get_auth_token() if not authorization: logger.error("获取认证token失败,程序退出") return # 获取广播数据 logger.info("获取广播数据...") data = get_broadcast_data_with_token(authorization) if not data: logger.error("没有获取到任何广播数据,程序退出") return logger.info(f"成功获取 {len(data)} 条广播数据") # 获取今天的日期 today = datetime.now(pytz.timezone('Asia/Shanghai')).strftime("%Y-%m-%d") # 过滤出今天的数据 today_data = [item for item in data if item["播音日期"] == today] if not today_data: logger.warning(f"今天 ({today}) 没有安排广播,程序退出") return logger.info(f"今天 ({today}) 有 {len(today_data)} 条广播数据") # 处理今天的所有时间段(包括已过去的) process_day(today_data, authorization, process_past_segments=False) logger.info("===== 广播播放系统完成所有任务 =====") if __name__ == "__main__": main()
最新发布
06-25
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值