修正auth.py 中的会重新登录问题

Luo · Luo · commit b2964d30bafb · 2015-09-09T23:29:41.000+08:00
diff --git a/auth.py b/auth.py
@@ -7,12 +7,7 @@
 
 # requirements
 import requests, termcolor
-# try:
-#     from bs4 import BeautifulSoup
-# except:
-#     import BeautifulSoup
-# Darwin platform
-# BeautifulSoup = BeautifulSoup.BeautifulSoup
+
 
 requests = requests.Session()
 requests.cookies = cookielib.LWPCookieJar('cookies')
@@ -80,9 +75,12 @@ def download_captcha():
     """
         System platform: https://docs.python.org/2/library/platform.html
     """
+    Logging.info(u"正在调用外部程序渲染验证码 ... ")
     if platform.system() == "Linux":
-        os.system("see %s &" % image_name )
+        Logging.info(u"Command: xdg-open %s &" % image_name )
+        os.system("xdg-open %s &" % image_name )
     elif platform.system() == "Darwin":
+        Logging.info(u"Command: open %s &" % image_name )
         os.system("open %s &" % image_name )
     elif platform.system() == "SunOS":
         os.system("open %s &" % image_name )
@@ -180,7 +178,7 @@ def read_account_from_config_file(config_file="config.ini"):
     if os.path.exists(config_file) and os.path.isfile(config_file):
         Logging.info(u"正在加载配置文件 ...")
         cf.read(config_file)
-        #cookies = cf._sections['cookies']
+
         email = cf.get("info", "email")
         password = cf.get("info", "password")
         if email == "" or password == "":
@@ -197,6 +195,7 @@ def read_account_from_config_file(config_file="config.ini"):
 def login(account=None, password=None):
     if islogin() == True:
         Logging.success(u"你已经登录过咯")
+        return True
 
     if account == None:
         (account, password) = read_account_from_config_file()
@@ -228,4 +227,5 @@ def login(account=None, password=None):
         return True
 
 if __name__ == "__main__":
-    login()
+    # login(account="xxxx@email.com", password="xxxxx")
+    login()
diff --git a/zhihu.py b/zhihu.py
@@ -59,39 +59,38 @@
     from bs4 import BeautifulSoup
 except:
     import BeautifulSoup
-# Darwin platform
-#BeautifulSoup = BeautifulSoup.BeautifulSoup
 
-reload(sys)
-sys.setdefaultencoding('utf8')
+# module
+from auth import islogin
+from auth import Logging
+
+
+"""
+    Note:
+        1. 身份验证由 `auth.py` 完成。
+        2. 身份信息保存在当前目录的 `cookies` 文件中。
+        3. `requests` 对象可以直接使用，身份信息已经自动加载。
 
+    By Luozijun (https://github.com/LuoZijun), 09/09 2015
 
+"""
 requests = requests.Session()
 requests.cookies = cookielib.LWPCookieJar('cookies')
 try:
     requests.cookies.load(ignore_discard=True)
 except:
-    print u"ERROR: 你还没有登录知乎哦 ..."
-    print u"INFO: 执行 `python auth.py` 即可以完成登录。"
+    Logging.error(u"你还没有登录知乎哦 ...")
+    Logging.info(u"执行 `python auth.py` 即可以完成登录。")
     raise Exception("无权限(403)")
 
-"""
-    Note:
-        全局变量 `session` 以及 `cookie` 还有 `create_session` 函数 可以逐步在代码移除，当前用作过渡。
-        
-        User/Question ... 等 抓取类 可以直接 使用 `requests.get/requests.post`
-        来抓取知乎链接。身份cookies已经自动设定。
-
-        身份Cookies文件保存于 `cookies`。
-"""
-session = requests
-
 
-def create_session():
-    print "NOTE: 该函数已经不再需要。"
-    return True
+if islogin() != True:
+    Logging.error(u"你的身份信息已经失效，请重新生成身份信息( `python auth.py` )。")
+    raise Exception("无权限(403)")
 
 
+reload(sys)
+sys.setdefaultencoding('utf8')
 
 class Question:
     url = None
