107 lines
4.1 KiB
Python
107 lines
4.1 KiB
Python
from PIL import Image
|
||
import numpy as np
|
||
import cv2
|
||
import base64
|
||
import io,os
|
||
import requests
|
||
import time,json
|
||
import string,random
|
||
import glob
|
||
##for CeKanYuan
|
||
def get_offlineUrls(taskUrl,offlineFile,jsonfile='SendLog/platformQueryOfftask.json'):
|
||
with open(offlineFile,'r') as fp:
|
||
lines=fp.readlines()
|
||
doneCodes=[line.strip().split(' ')[2] for line in lines]
|
||
try:
|
||
res = requests.get(taskUrl,timeout=10).json()
|
||
offlines=res['data'] ##offlines[0]['code'],offlines[0]['videoUrl']
|
||
with open(jsonfile,'w') as fp:
|
||
json.dump(res,fp, ensure_ascii=False)
|
||
except Exception as ee:
|
||
timestr=time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
||
print('###line43 %s read taskUrl:%s error:%s '%(timestr,taskUrl,ee))
|
||
offlines=[]
|
||
outOfflines=[]
|
||
for off in offlines:
|
||
off['port']=1935;
|
||
off.update({'name':'off-' +off.pop("code")})
|
||
if off['name'] in doneCodes:
|
||
continue
|
||
off.update({'url': off.pop("videoUrl")})
|
||
outOfflines.append(off)
|
||
#off['url']=off['videoUrl']
|
||
|
||
return outOfflines
|
||
def platurlToJsonfile(taskUrl,jsonfile='SendLog/platformQuery.json'):
|
||
try:
|
||
res = requests.get(taskUrl,timeout=10).json()
|
||
offlines=res['data'] ##offlines[0]['code'],offlines[0]['videoUrl']
|
||
with open(jsonfile,'w') as fp:
|
||
json.dump(res,fp, ensure_ascii=False)
|
||
except Exception as ee:
|
||
timestr=time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
||
print('###line43 %s read taskUrl:%s error:%s '%(timestr,taskUrl,ee))
|
||
|
||
def get_websource_fromTxt(txtfile):
|
||
with open(txtfile,'r') as fp:
|
||
lines = fp.readlines()
|
||
sources=[ ]
|
||
for line in lines:
|
||
sous={}
|
||
try:
|
||
sps = line.strip().split(' ')
|
||
sous['url']=sps[0];sous['port']=sps[1]
|
||
|
||
#webs.append(sps[0])
|
||
if 'rtmp' in sps[0]:
|
||
name = sps[0].split('/')[4]
|
||
|
||
else:
|
||
name = sps[0][-3:]
|
||
sous['name']='live-'+name.replace('_','')
|
||
sous['port']=sps[1]
|
||
sources.append(sous)
|
||
except Exception as ee:
|
||
|
||
print('####format error : %s, line:%s , in file:%s#####'%(ee,line,txtfile))
|
||
assert len(sources)>0
|
||
return sources
|
||
|
||
def update_websource_offAndLive(platform_query_url,sourceFile,offlineFile,jsonfile='SendLog/platformQuery.json'):
|
||
|
||
#platform_query_url='http://47.96.182.154:9051/api/suanfa/getPlatformInfo'
|
||
txtSource=get_websource_fromTxt(sourceFile)
|
||
try:
|
||
res = requests.get(platform_query_url,timeout=10).json()
|
||
questionUrl = res['data']['questionUrl'] ###直播流时，问题图片的推送地址
|
||
offlineUrl = res['data']['offlineUrl'] ###http离线视频时，问题图片的推送地址
|
||
taskUrl = res['data']['taskUrl'] ###http离线视频时，离线视频存放的地址
|
||
with open(jsonfile,'w') as fp:
|
||
json.dump(res,fp, ensure_ascii=False)
|
||
except Exception as ee:
|
||
timestr=time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
||
print('######line83: %s: file:geturlPlatform: error %s ,url:%s #####'%(timestr,ee,platform_query_url))
|
||
taskUrl='http://121.40.249.52:9050/api/analyse/getAiInspectionList'
|
||
if taskUrl:
|
||
offlines=get_offlineUrls(taskUrl,offlineFile)
|
||
txtSource.extend(offlines)
|
||
#[{'url': 'rtmp://demoplay.yunhengzhizao.cn/live/THSA_HD5M', 'port': '1935', 'name': 'live-THSAHD5M'}]
|
||
outlist=[]
|
||
for sourss in txtSource :
|
||
source_url = sourss['url']
|
||
vid = cv2.VideoCapture(source_url)
|
||
if vid.isOpened():
|
||
outlist.append( sourss )
|
||
return outlist
|
||
#print('##line65:',txtSource)
|
||
|
||
|
||
|
||
if __name__=='__main__':
|
||
platform_query_url='http://47.96.182.154:9051/api/suanfa/getPlatformInfo'
|
||
sourceFile='../config/source.txt'
|
||
offlineFile='../mintors/offlines/doneCodes.txt'
|
||
jsonfile='../SendLog/platformQuery.json'
|
||
txtSource=update_websource_offAndLive(platform_query_url,sourceFile,offlineFile,jsonfile=jsonfile)
|
||
print(txtSource)
|