Python gets current data based on stock code-Python Tutorial-php.cn

Python gets current data based on stock code

大家讲道理

Release： 2016-11-09 11:16:53

Original

1759 people have browsed it

If you open a stock website through a browser during working hours, you may be afraid of being seen by others. It doesn't matter. Just try executing the code on the command line to see the data.
Enter sh to view the Shanghai Composite Index
Enter sz to view the Shenzhen Index
Enter cyb to view the GEM Index
Other stock codes can be customized and added to the dictionary

Python version 2.7 .3

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import urllib2
import re
import datetime
 
def getStockInfo(url):
    """根据url获取信息"""
    stockList = []
    request = urllib2.Request(url)
    response = urllib2.urlopen(request)
 
    stockStr = response.read()
    stockList = stockStr.split(&#39;,&#39;)
    return stockList
 
def printStock(List):
    """打印相关信息"""
    print &#39;***********price*****************&#39; + List[1]
    print &#39;***********float_price***********&#39; + List[2]
    print &#39;***********float_perct***********&#39; + List[3] + &#39;%&#39;
    print &#39;***********succ_unit*************&#39; + List[4]+&#39; shou&#39;
    print &#39;***********succ_price************&#39; + List[5]
 
def getUrlByCode(code):
    """根据代码获取详细的url"""
    url = &#39;&#39;
    stockCode = &#39;&#39;
    if code == &#39;sh&#39;:
        url = &#39;http://hq.sinajs.cn/list=s_sh000001&#39;
    elif code == &#39;sz&#39;:
        url = &#39;http://hq.sinajs.cn/list=s_sz399001&#39;
    elif code == &#39;cyb&#39;:
        url = &#39;http://hq.sinajs.cn/list=s_sz399006&#39;
    else:
        pattern = re.compile(r&#39;^60*&#39;)
        match = pattern.match(code)
        if match:
            stockCode = &#39;sh&#39;+ code
        else:
            stockCode = &#39;sz&#39; + code
        url = &#39;http://hq.sinajs.cn/list=s_&#39;+stockCode
 
    return url
 
 
#输入stock代码输出对应的价格信息
#code = raw_input(&#39;code: &#39;)
codeDict = {
    &#39;sh&#39;     : &#39;shang hai zq&#39;,
    &#39;sz&#39;     : &#39;shen zheng zq&#39;,
    &#39;cyb&#39;    : &#39;chang ye ban&#39;,
    &#39;601788&#39; : &#39;guang da zheng quan&#39;,
    &#39;000651&#39; : &#39;ge li dian qi&#39;,
}
 
#http://hq.sinajs.cn/list=s_sh000001 (上海大盘查询)
#http://hq.sinajs.cn/list=s_sz399001 (深圳大盘查询)
 
count = 0;
while (count<=100):#循环100次后再退出
    # 循环字典
    for key in codeDict:
        print key + &#39;--&#39;+codeDict[key]
 
    code = raw_input(&#39;please select a code: &#39;)
    now_time = datetime.datetime.now()
 
    #打印该code的信息
    url = getUrlByCode(code)
    stockInfo = getStockInfo(url)
    #print stockInfo
    printStock(stockInfo)
 
    end_time = datetime.datetime.now()
    costTime =  (end_time - now_time).seconds
    print &#39;总共花费时间&#39;+str(costTime)+&#39;秒&#39;
    count +=1

Copy after login