Merge pull request egrcc#16 from LuoZijun/patch-4

egrcc · egrcc · commit 33773a010445 · 2015-09-11T22:52:41.000+08:00
修正重新登录问题
diff --git a/README.rst b/README.rst
@@ -1,6 +1,14 @@
 zhihu-python：获取知乎信息
 ===============================
 
+:Author: `egrcc <https://github.com/egrcc>`_ ( `微博 <http://weibo.com/u/2948739432>`_ | `电邮 <zhaolujun1994@gmail.com>`_ )
+:Committer: `Eureka22 <https://github.com/Eureka22>`_ ,  `lufo816 <https://github.com/lufo816>`_ , `LuoZijun <https://github.com/LuoZijun>`_
+:Update: 09/09 2015
+
+
+.. contents::
+
+
 介绍
 ----
 
@@ -40,42 +48,74 @@ zhihu-python 采用 python2.7 编写，用来方便地获取知乎上各种内
         
 导出的markdown，txt文件示例请见该项目的markdown，text文件夹。当然，想要知道某大V关注了那些人，提了什么问题也不在话下，详情请见：快速开始。        
 
-依赖
------
 
-- 使用 `Beautiful Soup 4 <http://www.crummy.com/software/BeautifulSoup/>`_ 解析 html 文档
-- 使用 `requests <https://github.com/kennethreitz/requests>`_ 处理 http 请求
-- 使用 `html2text <https://github.com/aaronsw/html2text>`_ 进行格式转换
 
-没有的话可以使用 pip 安装：
+快速开始
+---------
 
-.. code-block:: bash
+**准备**
 
-    $ pip install requests
-    $ pip install beautifulsoup4
-    $ pip install html2text
 
-或者使用
+**Tips** :
 
-.. code-block:: bash
+1.  确保你的系统里面已经安装了 `Python2.7 <https://www.python.org/>`_ ，不同作业系统如何安装不再赘述。
+2.  检查你系统中 `python` 和 `pip` 的版本, 如果不属于 `python2.7` , 请在执行代码范例时，自行将 `python` 和 `pip` 分别替换成 `python2.7` 和 `pip2` 。
+3.  确保你的系统中安装了 `git` 程序 以及 `python-pip` 。
 
-    $ pip install -r requirements.txt
 
-**注意** ：如果同时安装了 python3 和 python2 ， pip 命令可能默认安装的是 python3 版本的包，故需将上述命令中的 pip 换成pip2 （比如在我的Ubuntu上）。
+**克隆本项目**
 
-快速开始
----------
 
-zhihu-python 主要文件为 zhihu.py ，配置文件为 config.ini , 将这两个文件下载到你的工作目录，并修改
-config.ini 文件中的 email 为你的知乎账户邮箱，修改 password 为你的知乎账户密码（用作模拟登录）。
+.. code:: bash
+
+  git clone git@github.com:egrcc/zhihu-python.git
+  cd zhihu-python
+
+
+**解决依赖**
+
+* `Beautiful Soup 4 <http://www.crummy.com/software/BeautifulSoup/>`_
+* `requests <https://github.com/kennethreitz/requests>`_
+* `html2text <https://github.com/aaronsw/html2text>`_
+* `termcolor <https://pypi.python.org/pypi/termcolor>`_
+
+
+.. code:: bash
+
+  sudo pip install -r requirements.txt
+
+
+Or
+
+.. code:: bash
+
+  sudo pip2 install -r requirements.txt
+
+
+
+**登录知乎**
+
+
+登录 `知乎` 生成身份信息, 保存在当前目录的 `cookies` 文件中。
+
+.. code:: bash
+  
+  python auth.py
+
+
+**执行测试**
+
+
+.. code:: bash
+
+  python test.py
+
+不出意外，一切应该完美运行 :))
 
-(2015.2.1更新)由 `@Eureka22 <https://github.com/Eureka22>`_ 添加了cookies支持，若由于验证码原因出现登录失败，请查看浏览器的cookies，填写 config.ini 文件中的 cookies 项后重试（当然，你的cookies可能与  cogfig.ini 文件中的某些项不同，比如没有“c_c”，可能会多出“_ga”，把你的补充上去即可）。如能正常登录，cookies 项可以不填写。
 
-**注意** ：一定记得修改config.ini文件，否则无法正常使用。
 
- 
 Question：获取问题信息
-~~~~~~~~~~~~~~~~~~~~~~~~
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 Question 代表一个问题，处理知乎问题相关操作。创建一个 Question 对象需传入该问题的 url ，如：
 
@@ -142,7 +182,7 @@ Question 代表一个问题，处理知乎问题相关操作。创建一个 Ques
 
    
 Answer：获取答案信息
-~~~~~~~~~~~~~~~~~~~~~
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 Answer 代表了一个答案，处理知乎答案相关操作。创建一个 Answer 对象需传入该答案的 url ，如：
 
@@ -193,7 +233,7 @@ Answer 代表了一个答案，处理知乎答案相关操作。创建一个 Ans
 
 
 User：获取用户信息
-~~~~~~~~~~~~~~~~~~~~~~~
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 User 代表一个用户，处理用户相关操作。创建一个 User 对象需传入该用户的 url ，如：
 
@@ -268,7 +308,7 @@ User 代表一个用户，处理用户相关操作。创建一个 User 对象需
 
 
 Collection：获取收藏夹信息
-~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 Collection 代表一个收藏夹，处理收藏夹相关操作。创建一个 Collection 对象需传入该收藏夹的 url ，如：
 
@@ -312,7 +352,7 @@ Collection 代表一个收藏夹，处理收藏夹相关操作。创建一个 Co
     
 
 综合实例
-~~~~~~~~~~~~~~~
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
 将 Question ， Answer ， User ， Collection 结合起来使用。实例如下：
 
@@ -625,6 +665,6 @@ zhihu.Collection ---- 知乎收藏夹操作类
 联系我
 ----------
 
-- 微博：`@egrcc <http://weibo.com/u/2948739432>`_
-- github：`@egrcc <https://github.com/egrcc>`_
+- 微博：http://weibo.com/u/2948739432
+- github：https://github.com/egrcc
 - email：zhaolujun1994@gmail.com
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
