__author__ = 'felix'
# 原作者為 felix

import requests
# requests 用於爬取新浪股票 API
import time
import sys
# sys 用於在直譯器中互動
import threading
# threading 用於多執行緒處理    

from Queue import Queue
# Queue 是有關佇列的庫
from optparse import OptionParser
# OptionParser 用於在命令列中新增選項

class Worker(threading.Thread):
# 建立類 Work，多執行緒獲取

    def __init__(self, work_queue, result_queue):
    # 寫入必須繫結的強制屬性 self， work_queue， result_queue
    # __init__ 方法的第一個引數一定是 self，用於表示建立的例項本身
    # 其中，例項是根據類來建立的
    # 在 __init__ 方法內部，可將各種屬性繫結到 self，因為 self 指向建立的例項本身

        threading.Thread.__init__(self)
        self.work_queue = work_queue
        self.result_queue = result_queue
        self.start()

    def run(self):
    # 增加一個新方法 run

        while True:
            func, arg, code_index = self.work_queue.get()
            # 獲取 func, arg, code_index
            res = func(arg, code_index)
            self.result_queue.put(res)
            if self.result_queue.full():
                res = sorted([self.result_queue.get() for i in range(self.result_queue.qsize())], key=lambda s: s[0], reverse=True)
                # sorted() 是用於排序的方法，返回副本，原始輸入不變
                # Queue.get() 用於獲取佇列
                # Queue.qsize() 返回佇列大小         
                # key=lambda s: s[0]：關鍵詞為 lambda s: s[0]
                # lambda s: s[0]：匿名函式，返回第一個元素
                # reverse=True：降序排列      
                res.insert(0, ('0', u'名稱     股價'))
                # list.insert() 用於將指定物件插入列表的指定位置        
                print '***** start *****'
                for obj in res:
                    print obj[1]
                print '***** end *****\n'
            self.work_queue.task_done()
            # 在完成一項工作後，Queue.task_done() 會向任務已經完成的佇列傳送一個訊號       


class Stock(object):
# 股票實時價格獲取

    def __init__(self, code, thread_num):
        self.code = code
        self.work_queue = Queue()
        self.threads = []
        self.__init_thread_poll(thread_num)

    def __init_thread_poll(self, thread_num):
        self.params = self.code.split(',')
        # parmas 會向函式傳入一個字典
        self.params.extend(['s_sh000001', 's_sz399001'])  
        # 預設獲取滬指、深指
        # extend()： 擴充套件，與 append() 的區別為，extend() 加入的元素是分別單個加入的
        self.result_queue = Queue(maxsize=len(self.params[::-1]))
        for i in range(thread_num):
            self.threads.append(Worker(self.work_queue, self.result_queue))

    def __add_work(self, stock_code, code_index):
        self.work_queue.put((self.value_get, stock_code, code_index))
        # self.value_get 涉及到下面的裝飾器        

    def del_params(self):
        for obj in self.params:
            self.__add_work(obj, self.params.index(obj))

    def wait_all_complete(self):
        for thread in self.threads:
            if thread.isAlive():
            # 判斷執行緒是否是啟用的
            # 從呼叫 start() 方法啟動執行緒，到 run() 方法執行完畢或者遇到未處理異常而中斷，這段時間內執行緒是啟用的           
                thread.join()
                # join() 的作用是阻塞程序直到執行緒執行完畢
                # 依次檢查執行緒池中的執行緒是否接觸，沒有結束就阻塞執行緒直到執行緒結束
                # 如果結束則跳轉執行下一個執行緒的 join() 函式                 

    @classmethod
    # 裝飾器，返回函式的高階函式    
    def value_get(cls, code, code_index):
        slice_num, value_num = 21, 3
        name, now = u'——無——', u'  ——無——'
        if code in ['s_sh000001', 's_sz399001']:
            slice_num = 23
            value_num = 1
        r = requests.get("http://hq.sinajs.cn/list=%s" % (code,))
        # 爬取新浪股票 API                
        res = r.text.split(',')
        if len(res) > 1:
            name, now = r.text.split(',')[0][slice_num:], r.text.split(',')[value_num]
        return code_index, name + ' ' + now


if __name__ == '__main__':
# 該指令碼用於直接執行，而不能被 import
    parser = OptionParser(description="Query the stock's value.", usage="%prog [-c] [-s] [-t]", version="%prog 1.0")
    # 生成命令列說明
    # %prog 將會以當前程式名的字串來代替    
    parser.add_option('-c', '--stock-code', dest='codes',
                      help="the stock's code that you want to query.")
    # 使用 add_option() 來定義命令列引數，即加入選項
    # dest 是儲存的變數                     
    parser.add_option('-s', '--sleep-time', dest='sleep_time', default=6, type="int",
                      help='How long does it take to check one more time.')
    parser.add_option('-t', '--thread-num', dest='thread_num', default=3, type='int',
                      help="thread num.")
    options, args = parser.parse_args(args=sys.argv[1:])
    # 設定好命令列後，用 parse_args() 來解析命令列
    assert options.codes, "Please enter the stock code!"  
    # 是否輸入股票程式碼
    if filter(lambda s: s[:-6] not in ('sh', 'sz', 's_sh', 's_sz'), options.codes.split(',')):  
    # 股票程式碼輸入是否正確
        raise ValueError

    stock = Stock(options.codes, options.thread_num)

    while True:
        stock.del_params()
        time.sleep(options.sleep_time)
        # sleep() 用於使程式休眠