#!/usr/bin/env python # encoding: utf-8 """ 下载检测 """ import hashlib import os import time import datetime import traceback import sys import json import socket import threading from hashlib import sha1 from random import randint from struct import unpack from socket import inet_ntoa from threading import Timer, Thread from time import sleep reload(sys) sys.setdefaultencoding("utf-8") sys.path.append('/usr/local/lib/python2.7/site-packages') # import pygeoip import MySQLdb as mdb from configparser import ConfigParser cp = ConfigParser() cp.read("../qb.conf") section_db = cp.sections()[0] DB_HOST = cp.get(section_db, "DB_HOST") DB_USER = cp.get(section_db, "DB_USER") DB_PORT = cp.getint(section_db, "DB_PORT") DB_PASS = cp.get(section_db, "DB_PASS") DB_NAME = cp.get(section_db, "DB_NAME") section_qb = cp.sections()[1] QB_HOST = cp.get(section_qb, "QB_HOST") QB_PORT = cp.get(section_qb, "QB_PORT") QB_USER = cp.get(section_qb, "QB_USER") QB_PWD = cp.get(section_qb, "QB_PWD") section_file = cp.sections()[2] FILE_TO = cp.get(section_file, "FILE_TO") FILE_TRANSFER_TO = cp.get(section_file, "FILE_TRANSFER_TO") FILE_OWN = cp.get(section_file, "FILE_OWN") FILE_GROUP = cp.get(section_file, "FILE_GROUP") section_task = cp.sections()[3] TASK_RATE = cp.getint(section_task, "TASK_RATE") TASK_COMPLETED_RATE = cp.getint(section_task, "TASK_COMPLETED_RATE") def formatTime(): return time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) class downloadBT(Thread): def __init__(self): Thread.__init__(self) self.setDaemon(True) self.dbconn = mdb.connect( DB_HOST, DB_USER, DB_PASS, DB_NAME, port=DB_PORT, charset='utf8') self.dbconn.autocommit(False) self.dbcurr = self.dbconn.cursor() self.dbcurr.execute('SET NAMES utf8') self.qb = self.qb() self.has_suffix = ['.mp4'] def query(self, sql): self.dbcurr.execute(sql) result = self.dbcurr.fetchall() data = map(list, result) return data def qb(self): from qbittorrent import Client url = 'http://' + QB_HOST + ':' + QB_PORT + '/' qb = Client(url) qb.login(QB_USER, QB_PWD) return qb def execShell(self, cmdstring, cwd=None, timeout=None, shell=True): import subprocess if shell: cmdstring_list = cmdstring else: cmdstring_list = shlex.split(cmdstring) if timeout: end_time = datetime.datetime.now() + datetime.timedelta(seconds=timeout) sub = subprocess.Popen(cmdstring_list, cwd=cwd, stdin=subprocess.PIPE, shell=shell, bufsize=4096, stdout=subprocess.PIPE, stderr=subprocess.PIPE) while sub.poll() is None: time.sleep(0.1) if timeout: if end_time <= datetime.datetime.now(): raise Exception("Timeout：%s" % cmdstring) return sub.communicate() def md5(self, str): # 生成MD5 try: m = hashlib.md5() m.update(str) return m.hexdigest() except: return False def get_transfer_ts_file(self, to): return FILE_TRANSFER_TO + '/' + to + '.ts' def get_transfer_mp4_file(self, to): return FILE_TRANSFER_TO + '/' + to + '.mp4' def get_transfer_m3u5_dir(self, dirname): return FILE_TO + '/m3u8/' + dirname def fg_transfer_mp4_cmd(self, sfile, dfile): cmd = 'ffmpeg -y -i ' + sfile + ' -c:v libx264 -strict -2 ' + dfile return cmd def fg_transfer_ts_cmd(self, file, to_file): cmd = 'ffmpeg -y -i ' + file + \ ' -vcodec copy -acodec copy -vbsf h264_mp4toannexb ' + to_file return cmd def fg_m3u8_cmd(self, ts_file, m3u8_file, to_file): cmd = 'ffmpeg -y -i ' + ts_file + ' -c copy -map 0 -f segment -segment_list ' + \ m3u8_file + ' -segment_time 3 ' + to_file return cmd def ffmpeg(self, file=''): md5file = self.md5(file)[0:6] m3u8_dir = self.get_transfer_m3u5_dir(md5file) self.execShell('mkdir -p ' + m3u8_dir) mp4file = self.get_transfer_mp4_file(md5file) if not os.path.exists(mp4file): cmd_mp4 = self.fg_transfer_mp4_cmd(file, mp4file) print 'cmd_mp4:', cmd_mp4 data_mp4 = self.execShell(cmd_mp4) print 'mp4:', data_mp4[1] print formatTime(), 'mp4 exists:', mp4file tsfile = self.get_transfer_ts_file(md5file) cmd_ts = self.fg_transfer_ts_cmd(mp4file, tsfile) print 'cmd_ts:', cmd_ts data_ts = self.execShell(cmd_ts) print 'ts:', data_ts[1] m3u8_file = m3u8_dir + '/' + md5file + '.m3u8' tofile = m3u8_dir + '/%03d.ts' cmd_m3u8 = self.fg_m3u8_cmd(tsfile, m3u8_file, tofile) print 'cmd_m3u8:', cmd_m3u8 data_m3u8 = self.execShell(cmd_m3u8) print 'm3u8:', data_m3u8[1] self.execShell('chown -R ' + FILE_OWN + ':' + FILE_GROUP + ' ' + m3u8_dir) def file_arr(self, path, filters=['.DS_Store']): file_list = [] flist = os.listdir(path) # print flist for i in range(len(flist)): file_path = os.path.join(path, flist[i]) if flist[i] in filters: continue if os.path.isdir(file_path): tmp = self.file_arr(file_path, filters) file_list.extend(tmp) else: file_list.append(file_path) return file_list def find_dir_video(self, path): flist = self.file_arr(path) video = [] for i in range(len(flist)): t = os.path.splitext(flist[i]) if t[1] in self.has_suffix: video.append(flist[i]) return video def video_do(self, path): if os.path.isfile(path): t = os.path.splitext(path) if t[1] in self.has_suffix: self.ffmpeg(path) else: vlist = self.find_dir_video(path) for i in range(len(vlist)): self.ffmpeg(vlist[i]) return '' def checkTask(self): while True: torrents = self.qb.torrents(filter='downloading') tlen = len(torrents) if tlen > 0: print "downloading torrents count:", tlen for torrent in torrents: print torrent['name'], ' task downloading!' else: print formatTime(), "no downloading task!" time.sleep(TASK_RATE) def completed(self): while True: torrents = self.qb.torrents(filter='completed') tlen = len(torrents) print "completed torrents count:", tlen if tlen > 0: for torrent in torrents: path = torrent['save_path'] + torrent['name'] try: self.video_do(path) except Exception as e: print formatTime(), str(e) print formatTime(), "done task!" else: print formatTime(), "no completed task!" time.sleep(TASK_COMPLETED_RATE) def test(): while True: print formatTime(), "no download task!", time.sleep(1) test() if __name__ == "__main__": dl = downloadBT() import threading task = threading.Thread(target=dl.checkTask) task.start() completed = threading.Thread(target=dl.completed) completed.start() # test()