mirror of
https://github.com/midoks/mdserver-web.git
synced 2026-10-18 23:45:52 +08:00
162 lines
4.2 KiB
Python
Executable File
162 lines
4.2 KiB
Python
Executable File
#!/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")
|
||
|
||
|
||
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()
|
||
|
||
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 ffmpeg(self, file=''):
|
||
#vfile = '/Users/midoks/Desktop/www/btplayer/public/video/test.mp4'
|
||
md5file = self.md5(file)
|
||
m3u8_dir = FILE_TO + '/m3u8/' + md5file
|
||
os.system('mkdir -p ' + m3u8_dir)
|
||
m3u8_file = m3u8_dir + '/' + md5file + '.m3u8'
|
||
|
||
tofile = FILE_TO + '/m3u8/' + md5file + '/%03d.ts'
|
||
cmd = 'ffmpeg -i ' + file + \
|
||
' -c copy -map 0 -f segment -segment_list ' + \
|
||
m3u8_file + ' -segment_time 5 ' + tofile
|
||
print cmd
|
||
self.execShell(cmd)
|
||
|
||
def video_do(self, dir):
|
||
return ''
|
||
|
||
def checkTask(self):
|
||
while True:
|
||
torrents = self.qb.torrents()
|
||
for torrent in torrents:
|
||
print torrent
|
||
print time.time(), "no task!"
|
||
time.sleep(10)
|
||
|
||
def completed(self):
|
||
while True:
|
||
torrents = self.qb.torrents(filter='completed')
|
||
# print torrents
|
||
for torrent in torrents:
|
||
print torrent
|
||
self.ffmpeg(
|
||
'/Users/midoks/Desktop/www/btplayer/public/video/test.mp4')
|
||
print time.time(), "no task!"
|
||
time.sleep(60)
|
||
|
||
|
||
def test():
|
||
while True:
|
||
print time.time(), "no download task!",
|
||
time.sleep(1)
|
||
test()
|
||
|
||
if __name__ == "__main__":
|
||
|
||
dl = downloadBT()
|
||
|
||
import threading
|
||
# t = threading.Thread(target=dl.checkTask)
|
||
# t.start()
|
||
|
||
completed = threading.Thread(target=dl.completed)
|
||
completed.start()
|
||
|
||
# test()
|