import csv
import sys
import json
import os
import re
import traceback
import requests
import subprocess
from ffmpy import FFprobe
from subprocess import PIPE
from sys import stdout
from termcolor import colored, RESET
from datetime import datetime
dt=datetime.now()
# Channel	Group	Source	Link

SKIP_FFPROBE_MESSAGES = [re.compile(pattern) for pattern in (
	'Last message repeated',
	'mmco: unref short failure',
	'number of reference frames .+ exceeds max',
)]

def check_channel(clist,num):
    # clist 为一行 csv
    uri = clist[3]
    try:
        r = requests.get(clist[3], timeout=2) # 先测能不能正常访问
        if(r.status_code == requests.codes.ok):
            #ffprobe = FFprobe(inputs={uri: '-v warning'})
            #errors = tuple(filter(
			#    lambda line: not (line in ('', RESET) or any(regex.search(line) for regex in SKIP_FFPROBE_MESSAGES)),
			#    ffprobe.run(stderr=PIPE)[1].decode('utf-8').split('\n')
		    #))
            #if errors: # https://github.com/Jamim/iptv-checker/blob/master/iptv-checker.py#L26
            #    print('[{}] {}({}) Error:{}'.format(str(num), clist[0], clist[2], str(errors)))
            #    return False
            #else: # 查视频信息
            ffprobe = FFprobe(inputs={uri: '-v error -show_format -show_streams -print_format json'})
            cdata = json.loads(ffprobe.run(stdout=PIPE,stderr=PIPE)[0].decode('utf-8'))
            flagAudio = 0
            flagVideo = 0
            vwidth = 0
            vheight = 0
            for i in cdata['streams']:
                if i['codec_type'] == 'video':
                    flagVideo = 1
                    if vwidth <= i['coded_width']: # 取最高分辨率
                        vwidth = i['coded_width']
                        vheight = i['coded_height']
                elif i['codec_type'] == 'audio':
                    flagAudio = 1
            if flagAudio == 0:
                print('[{}] {}({}) Error: Video Only!'.format(str(num), clist[0], clist[2]))
                return False
            if flagVideo == 0:
                print('[{}] {}({}) Error: Audio Only!'.format(str(num), clist[0], clist[2]))
                return False
            print('[{}] {}({}) PASS: {}*{}'.format(str(num), clist[0], clist[2], vwidth, vheight))
            return [vwidth,vheight]
        else:
            print('[{}] {}({}) Error:{}'.format(str(num), clist[0], clist[2], str(r.status_code)))
            return False
    except Exception as e:
        #traceback.print_exc()
        print('[{}] {}({}) Error:{}'.format(str(num), clist[0], clist[2], str(e)))
        return False

def print_info():
    print('Time: {}-{}-{} {}:{}'.format(dt.year,dt.month,dt.day,dt.hour,dt.minute))
    #subprocess.run(['ffprobe'])

def main():
    print_info()
    Total = 0
    fulltimes = '-{}{}{}{}{}'.format(dt.year,dt.month,dt.day,dt.hour,dt.minute) # 时间后缀
    # times = fulltimes # 有时间后缀
    times = '' # 无时间后缀
    with open('data.csv') as f:
        f_csv = csv.reader(f)
        headers = next(f_csv)
        num = 1
        with open('data{}.csv'.format(fulltimes), 'a+') as f0: # 写入检测后新data
           # print('Channel,Group,Source,Link', file=f0)
            for row in f_csv:
                ret = check_channel(row,num)
                if(ret): # 通过，写入
                    with open('groups/{}{}.txt'.format(row[1],times), 'a+') as f1:
                        print('{}({}-{}*{}),{}'.format(row[0],row[2],ret[0],ret[1],row[3]), file=f1)
                    with open('groups/{}-simple{}.txt'.format(row[1],times), 'a+') as f1:
                        print('{},{}'.format(row[0],row[3]), file=f1)
                    with open('merged{}.txt'.format(times),'a+') as f1:
                        print('{}({}-{}*{}),{}'.format(row[0],row[2],ret[0],ret[1],row[3]), file=f1)
                    with open('merged-simple{}.txt'.format(times),'a+') as f1:
                        print('{},{}'.format(row[0],row[3]), file=f1)
                    print('{},{},{},{}'.format(row[0],row[1],row[2],row[3]), file=f0)
                    Total = Total + 1
                num = num + 1
    print('Total: {}'.format(Total))
if __name__ == '__main__':
    main()