mirror of
https://github.com/midoks/mdserver-web.git
synced 2026-10-18 23:45:52 +08:00
127 lines
3.1 KiB
Python
Executable File
127 lines
3.1 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 ffmpeg(self, file=''):
|
|
vfile = '/Users/midoks/Desktop/www/btplayer/public/video/test.mp4'
|
|
m3u8_dir = FILE_TO + '/m3u8'
|
|
m3u8_file = m3u8_dir + '/test.m3u8'
|
|
os.system('mkdir -p ' + m3u8_dir)
|
|
tofile = FILE_TO + '/m3u8/' + '%03d.ts'
|
|
cmd = 'ffmpeg -i ' + vfile + \
|
|
' -c copy -map 0 -f segment -segment_list ' + \
|
|
m3u8_file + ' -segment_time 5 ' + tofile
|
|
print cmd
|
|
os.system(cmd)
|
|
|
|
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()
|
|
print time.time(), "no task!"
|
|
time.sleep(10)
|
|
|
|
|
|
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()
|