-
-
Notifications
You must be signed in to change notification settings - Fork 75
/
play.py
153 lines (143 loc) · 12.6 KB
/
play.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
#coding:utf-8
import os
import sys
import time
import random
from mutagen.mp3 import MP3
import var_set
import shutil
import _thread
import ass_maker
path = var_set.path #引入设置路径
rtmp = var_set.rtmp #引入设置的rtmp网址
live_code = var_set.live_code #引入设置rtmp参数
deviceType = var_set.deviceType
def get_v():
if deviceType == "pi":
return "h264_omx"
elif deviceType == "vps":
return "libx264"
#格式化时间，暂时没啥用，以后估计也没啥用
def convert_time(n):
s = n%60
m = int(n/60)
return '00:'+"%02d"%m+':'+"%02d"%s
#移动放完的视频到缓存文件夹
def remove_v(filename):
try:
shutil.move(path+'/downloads/'+filename,path+'/default_mp3/')
except Exception as e:
print(e)
try:
os.remove(path+'/downloads/'+filename.replace(".flv",'')+'ok.ass')
os.remove(path+'/downloads/'+filename.replace(".flv",'')+'ok.info')
except Exception as e:
print(e)
print('delete error')
while True:
try:
if (time.localtime()[3] >= 22 or time.localtime()[3] <= 5) and var_set.play_videos_when_night:
print('night is comming~') #晚上到咯~
night_files = os.listdir(path+'/night') #获取所有缓存文件
night_files.sort() #排序文件
night_ran = random.randint(0,len(night_files)-1) #随机抽一个文件
if(night_files[night_ran].find('.flv') != -1): #如果为flv视频
#直接暴力推流
print('ffmpeg -threads 0 -re -i "'+path+"/night/"+night_files[night_ran]+'" -vcodec copy -acodec copy -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -i "'+path+"/night/"+night_files[night_ran]+'" -vcodec copy -acodec copy -f flv "'+rtmp+live_code+'"')
if(night_files[night_ran].find('.mp3') != -1): #如果为mp3
pic_files = os.listdir(path+'/default_pic') #获取准备的图片文件夹中的所有图片
pic_files.sort() #排序数组
pic_ran = random.randint(0,len(pic_files)-1) #随机选一张图片
audio = MP3(path+'/night/'+night_files[night_ran]) #获取mp3文件信息
seconds=audio.info.length #获取时长
print('mp3 long:'+convert_time(seconds))
if not os.path.isfile(path+'/night/'+night_files[night_ran]+'.ass'):
ass_maker.make_ass('../night/'+night_files[night_ran].replace('.mp3',''),'当前是晚间专属时间哦~时间范围：晚上22点-凌晨5点\\N大家晚安哦~做个好梦~\\N当前文件名：'+night_files[night_ran],path)
print('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/night/'+night_files[night_ran]+'" -vf ass="'+path+'/night/'+night_files[night_ran]+'.ass" -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/night/'+night_files[night_ran]+'" -vf ass="'+path+'/night/'+night_files[night_ran].replace('.mp3','')+'.ass" -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
continue
files = os.listdir(path+'/downloads') #获取文件夹下全部文件
files.sort() #排序文件，按文件名（点播时间）排序
count=0 #总共匹配到的点播文件统计
for f in files:
if((f.find('.mp3') != -1) and (f.find('.download') == -1)): #如果是mp3文件
print(path+'/downloads/'+f)
seconds = 600
bitrate = 0
try:
audio = MP3(path+'/downloads/'+f) #获取mp3文件信息
seconds=audio.info.length #获取时长
bitrate=audio.info.bitrate #获取码率
except Exception as e:
print(e)
bitrate = 99999999999
print('mp3 long:'+convert_time(seconds))
if((seconds > 600) | (bitrate > 400000)): #大于十分钟就不播放/码率限制400k以下
print('too long/too big,delete')
else:
pic_files = os.listdir(path+'/default_pic') #获取准备的图片文件夹中的所有图片
pic_files.sort() #排序数组
pic_ran = random.randint(0,len(pic_files)-1) #随机选一张图片
#推流
#如果存在封面
if os.path.isfile(path+'/downloads/'+f.replace(".mp3",'')+'.jpg'):
print('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/downloads/'+f.replace(".mp3",'')+'.jpg'+'" -filter_complex "[0:v][1:v]overlay=30:390[cover];[cover]ass='+path+"/downloads/"+f.replace(".mp3",'')+'.ass'+'[result]" -i "'+path+'/downloads/'+f+'" -map "[result]" -map 2,0 -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v h264_omx -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/downloads/'+f.replace(".mp3",'')+'.jpg'+'" -filter_complex "[0:v][1:v]overlay=30:390[cover];[cover]ass='+path+"/downloads/"+f.replace(".mp3",'')+'.ass'+'[result]" -i "'+path+'/downloads/'+f+'" -map "[result]" -map 2,0 -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v h264_omx -f flv "'+rtmp+live_code+'"')
else:#如果不存在封面
print('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/downloads/'+f+'" -vf ass="'+path+"/downloads/"+f.replace(".mp3",'')+'.ass'+'" -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/downloads/'+f+'" -vf ass="'+path+"/downloads/"+f.replace(".mp3",'')+'.ass'+'" -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
try: #放完后删除mp3文件、删除字幕、删除点播信息、封面图片
shutil.move(path+'/downloads/'+f,path+'/default_mp3/')
shutil.move(path+'/downloads/'+f.replace(".mp3",'')+'.ass',path+'/default_mp3/')
if os.path.isfile(path+'/downloads/'+f.replace(".mp3",'')+'.jpg'):
shutil.move(path+'/downloads/'+f.replace(".mp3",'')+'.jpg',path+'/default_mp3/')
except Exception as e:
print(e)
try:
os.remove(path+'/downloads/'+f.replace(".mp3",'')+'.info')
os.remove(path+'/downloads/'+f)
os.remove(path+'/downloads/'+f.replace(".mp3",'')+'.ass')
os.remove(path+'/downloads/'+f.replace(".mp3",'')+'.jpg')
except:
print('delete error')
count+=1 #点播统计加一
break
if((f.find('ok.flv') != -1) and (f.find('.download') == -1) and (f.find('rendering') == -1)): #如果是有ok标记的flv文件
print('flv:'+f)
#直接推流
print('ffmpeg -threads 0 -re -i "'+path+"/downloads/"+f+'" -vcodec copy -acodec copy -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -i "'+path+"/downloads/"+f+'" -vcodec copy -acodec copy -f flv "'+rtmp+live_code+'"')
os.rename(path+'/downloads/'+f,path+'/downloads/'+f.replace("ok","")) #修改文件名，以免下次循环再次匹配
_thread.start_new_thread(remove_v, (f.replace("ok",""),)) #异步搬走文件，以免推流卡顿
count+=1 #点播统计加一
break
if(count == 0): #点播统计为0，说明点播的都放完了
print('no media')
mp3_files = os.listdir(path+'/default_mp3') #获取所有缓存文件
mp3_files.sort() #排序文件
mp3_ran = random.randint(0,len(mp3_files)-1) #随机抽一个文件
if(mp3_files[mp3_ran].find('.mp3') != -1): #如果是mp3文件
pic_files = os.listdir(path+'/default_pic') #获取准备的图片文件夹中的所有图片
pic_files.sort() #排序数组
pic_ran = random.randint(0,len(pic_files)-1) #随机选一张图片
audio = MP3(path+'/default_mp3/'+mp3_files[mp3_ran]) #获取mp3文件信息
seconds=audio.info.length #获取时长
print('mp3 long:'+convert_time(seconds))
#推流
if(os.path.isfile(path+'/default_mp3/'+mp3_files[mp3_ran].replace(".mp3",'')+'.ass')):
if os.path.isfile(path+"/default_mp3/"+mp3_files[mp3_ran].replace(".mp3",'')+'.jpg'):
print('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+"/default_mp3/"+mp3_files[mp3_ran].replace(".mp3",'')+'.jpg'+'" -filter_complex "[0:v][1:v]overlay=30:390[cover];[cover]ass='+path+"/default_mp3/"+mp3_files[mp3_ran].replace(".mp3",'')+'.ass'+'[result]" -i "'+path+'/default_mp3/'+mp3_files[mp3_ran]+'" -map "[result]" -map 2,0 -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+"/default_mp3/"+mp3_files[mp3_ran].replace(".mp3",'')+'.jpg'+'" -filter_complex "[0:v][1:v]overlay=30:390[cover];[cover]ass='+path+"/default_mp3/"+mp3_files[mp3_ran].replace(".mp3",'')+'.ass'+'[result]" -i "'+path+'/default_mp3/'+mp3_files[mp3_ran]+'" -map "[result]" -map 2,0 -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
else:
print('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/default_mp3/'+mp3_files[mp3_ran]+'" -vf ass="'+path+"/default_mp3/"+mp3_files[mp3_ran].replace(".mp3",'')+'.ass'+'" -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/default_mp3/'+mp3_files[mp3_ran]+'" -vf ass="'+path+"/default_mp3/"+mp3_files[mp3_ran].replace(".mp3",'')+'.ass'+'" -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
else:
print('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/default_mp3/'+mp3_files[mp3_ran]+'" -vf ass="'+path+'/default.ass" -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -loop 1 -r 2 -t '+str(int(seconds))+' -f image2 -i "'+path+'/default_pic/'+pic_files[pic_ran]+'" -i "'+path+'/default_mp3/'+mp3_files[mp3_ran]+'" -vf ass="'+path+'/default.ass" -pix_fmt yuv420p -preset ultrafast -maxrate '+var_set.maxbitrate+'k -acodec copy -c:v '+get_v()+' -f flv "'+rtmp+live_code+'"')
if(mp3_files[mp3_ran].find('.flv') != -1): #如果为flv视频
#直接推流
print('ffmpeg -threads 0 -re -i "'+path+"/default_mp3/"+mp3_files[mp3_ran]+'" -vcodec copy -acodec copy -f flv "'+rtmp+live_code+'"')
os.system('ffmpeg -threads 0 -re -i "'+path+"/default_mp3/"+mp3_files[mp3_ran]+'" -vcodec copy -acodec copy -f flv "'+rtmp+live_code+'"')
except Exception as e:
print(e)