From 6c84efa0dab03fab6f3ee8f08b727a8258d592d4 Mon Sep 17 00:00:00 2001
From: xw_y_am <xw_y_am@163.com>
Date: Sat, 12 May 2018 17:51:55 +0800
Subject: [PATCH] use a function instead of tuple to get sites list

---
 sites.py | 23 +++++++++++++++++++----
 ux.py    |  5 +++--
 2 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/sites.py b/sites.py
index 5a29809..36f4587 100644
--- a/sites.py
+++ b/sites.py
@@ -3,7 +3,18 @@ import requests
 from bs4 import BeautifulSoup
 
 
-class Sites:
+def lst():
+    def get():
+        for _, value in globals().items():
+            try:
+                if value.site_enable:
+                    yield value
+            except:
+                pass
+    return tuple(get())
+
+
+class Sites():
     def __init__(self, signal):
         self.send_item = signal
         print(self.name, 'start ...')
@@ -70,6 +81,7 @@ class Sites:
 
 class Bobobt(Sites):
     name = 'Bobobt'
+    site_enable = True
 
     @staticmethod
     def trns_key(key_word):
@@ -104,6 +116,7 @@ class Bobobt(Sites):
 
 class BTcerise(Sites):
     name = 'BTcerise'
+    site_enable = True
 
     @staticmethod
     def trns_key(key_word):
@@ -136,6 +149,7 @@ class BTcerise(Sites):
 
 class Cililianc(Sites):
     name = 'Cililianc'
+    site_enable = False
 
     @staticmethod
     def trns_key(key_word):
@@ -168,6 +182,7 @@ class Cililianc(Sites):
 
 class BTdao(Sites):
     name = 'BTdao'
+    site_enable = True
 
     @staticmethod
     def trns_key(key_word):
@@ -206,6 +221,7 @@ class BTdao(Sites):
 
 class BTrabbit(Sites):
     name = 'BTrabbit'
+    site_enable = True
 
     @staticmethod
     def trns_key(key_word):
@@ -244,6 +260,7 @@ class BTrabbit(Sites):
 
 class BTanw(Sites):
     name = 'BTanw'
+    site_enable = True
 
     @staticmethod
     def trns_key(key_word):
@@ -282,6 +299,7 @@ class BTanw(Sites):
 
 class Ciliba(Sites):
     name = 'Ciliba'
+    site_enable = True
 
     @staticmethod
     def trns_key(key_word):
@@ -316,6 +334,3 @@ class Ciliba(Sites):
     @staticmethod
     def get_link(soup):
         return soup.find('a', 'download').get('href')
-
-
-lst = (Bobobt, BTcerise, BTdao, BTrabbit, BTanw, Ciliba)
diff --git a/ux.py b/ux.py
index 92e41bf..a4b05ae 100644
--- a/ux.py
+++ b/ux.py
@@ -18,7 +18,8 @@ class Ux(QMainWindow, window.Ui_MainWindow):
         self.tb_result.setColumnWidth(0, 70)
         self.tb_result.setColumnWidth(1, 90)
         self.tb_result.setColumnWidth(2, 480)
-        for it in sites.lst:
+        self.site_list = sites.lst()
+        for it in self.site_list:
             self.cb_source.addItem(it.name)
 
     def update_state(self):
@@ -35,7 +36,7 @@ class Ux(QMainWindow, window.Ui_MainWindow):
 
     def on_cb_source_currentIndexChanged(self, index):
         if type(index) == type(1):
-            self.site = sites.lst[index]
+            self.site = self.site_list[index]
             self.stop_search()
 
     def on_btn_search_clicked(self, b=True):