加入延迟获取

This commit is contained in:
EvilCult 2019-05-23 17:32:06 +08:00
parent 4abc8e0f0d
commit 823a9e94e5
3 changed files with 18 additions and 9 deletions

2
db.py
View File

@ -37,7 +37,7 @@ class DataBase :
def create (self) : def create (self) :
if self.connStat == False : return False if self.connStat == False : return False
sql = 'create table ' + self.table + ' (id integer PRIMARY KEY autoincrement, title text, quality text, url text, enable integer, online integer, update text)' sql = 'create table ' + self.table + ' (id integer PRIMARY KEY autoincrement, title text, quality text, url text, enable integer, online integer, delay integer, update text)'
self.cur.execute(sql) self.cur.execute(sql)
def query (self, sql) : def query (self, sql) :

23
main.py
View File

@ -2,6 +2,7 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import tools import tools
import time
import re import re
class Iptv : class Iptv :
@ -27,12 +28,17 @@ class Iptv :
for item in sourceList : for item in sourceList :
playable = self.chkPlayable(item[1]) netstat = self.chkPlayable(item[1])
if playable == True : if netstat > 0 :
info = self.fmtTitle(item[0]) info = self.fmtTitle(item[0])
print('title: ' + str(info['id']) + ' ' + str(info['title']))
print('url: ' + str(item[1])) data = {
'title': str(info['id']) + str(info['title']),
'url': str(item[1]),
'delay': netstat
}
print(data)
else : else :
pass # MAYBE later :P pass # MAYBE later :P
else : else :
@ -40,14 +46,17 @@ class Iptv :
def chkPlayable (self, url) : def chkPlayable (self, url) :
try: try:
startTime = int(round(time.time() * 1000))
res = self.T.getPage(url) res = self.T.getPage(url)
if res['code'] == 200 : if res['code'] == 200 :
return True endTime = int(round(time.time() * 1000))
useTime = endTime - startTime
return int(useTime)
else: else:
return False return 0
except: except:
return False return 0
def baseFilter (self) : def baseFilter (self) :

View File

@ -12,7 +12,7 @@ import gzip
import random import random
import socket import socket
socket.setdefaulttimeout(5.0) socket.setdefaulttimeout(10.0)
class Tools : class Tools :