加入第三个视频源

EvilCult · web-flow · commit 67e7c48977be · 2019-05-29T18:16:52.000+08:00
diff --git a/main.py b/main.py
@@ -7,6 +7,7 @@
 import re
 from plugins import base
 from plugins import lista
+from plugins import listb
 from plugins import dotpy
 
 class Iptv (object):
@@ -26,6 +27,11 @@ def run(self) :
         for item in urlList :
             self.addData(item)
 
+        listB = listb.Source()
+        urlList = listB.getSource()
+        for item in urlList :
+            self.addData(item)
+
         Dotpy = dotpy.Source()
         urlList = Dotpy.getSource()
         for item in urlList :
@@ -48,7 +54,7 @@ def addData (self, data) :
     def outPut (self) :
         sql = """SELECT * FROM
             (SELECT * FROM %s WHERE online = 1 ORDER BY delay DESC) AS delay
-            GROUP BY delay.title
+            GROUP BY LOWER(delay.title)
             HAVING delay.title != '' and delay.title != 'CCTV-' AND delay.delay < 500
             ORDER BY level ASC, length(title) ASC, title ASC
             """ % (self.DB.table)
@@ -64,14 +70,17 @@ def outPut (self) :
                     className = '地方频道'
                 elif item[4] == 3 :
                     className = '地方频道'
+                elif item[4] == 7 :
+                    className = '广播频道'
                 else :
                     className = '其他频道'
 
                 f.write("#EXTINF:-1, group-title=\"%s\", %s\n" % (className, item[1]))
                 f.write("%s\n" % (item[3]))
 
-obj = Iptv()
-obj.run()
+if __name__ == '__main__':
+    obj = Iptv()
+    obj.run()
 
 
 
diff --git a/plugins/listb.py b/plugins/listb.py
@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import tools
+import time
+import re
+
+class Source (object) :
+
+    def __init__ (self):
+        self.T = tools.Tools()
+        self.now = int(time.time() * 1000)
+
+    def getSource (self) :
+        urlList = []
+
+        url = 'https://www.sheng521.top/zyfx/zbyfx'
+        req = [
+            'user-agent: Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/73.0.3683.86 Mobile Safari/537.36',
+        ]
+        res = self.T.getPage(url, req)
+
+        if res['code'] == 200 :
+            pattern = re.compile(r"<h2 class=\"entry-title\">.*?href=\"(.*?)\".*?<\/h2>", re.I|re.S)
+            postList = pattern.findall(res['body'])
+
+            for post in postList :
+                url = post
+                req = [
+                    'user-agent: Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/73.0.3683.86 Mobile Safari/537.36',
+                ]
+                res = self.T.getPage(url, req)
+
+                if res['code'] == 200 :
+                    pattern = re.compile(r"<blockquote(.*?)</blockquote>", re.I|re.S)
+                    tmp = pattern.findall(res['body'])
+
+                    pattern = re.compile(r"span .*?>(.*?),(.*?)<\/span>", re.I|re.S)
+
+                    sourceList = pattern.findall(tmp[0])
+
+                    i = 1
+                    total = len(sourceList)
+                    for item in sourceList :
+                        print('Checking[ %s / %s ]: %s' % (i, total,  str(item[0])))
+                        i = i + 1
+                        netstat = self.T.chkPlayable(item[1])
+
+                        if netstat > 0 :
+                            info = self.T.fmtTitle(item[0])
+
+                            data = {
+                                'title'  : str(info['id']) if info['id'] != '' else str(info['title']),
+                                'url'    : str(item[1]),
+                                'quality': str(info['quality']),
+                                'delay'  : netstat,
+                                'level'  : str(info['level']),
+                                'online' : 1,
+                                'udTime' : self.now,
+                            }
+                            urlList.append(data)
+                        else :
+                            pass # MAYBE later :P
+                else :
+                    pass # MAYBE later :P
+
+        return urlList
diff --git a/tools.py b/tools.py
@@ -119,6 +119,9 @@ def fmtTitle (self, string) :
             if '+' in result['title'] :
                 result['id'] = result['id'] + str('+')
 
+        pattern = re.compile(r"\[\d+\*\d+\]", re.I)
+        result['title'] = re.sub(pattern, "", result['title'])
+
         Area = area.Area()
         result['level'] = Area.classify(str(result['id']) + str(result['title']))
 
diff --git a/tv.m3u8 b/tv.m3u8
