100-24-literal-9.py

__author__ = 'rui'
#coding=utf-8
import urllib
import json
import re
from pylab import *

mpl.rcParams['font.sans-serif'] = ['SimHei']  # 指定默认字体
mpl.rcParams['axes.unicode_minus'] = False  # 解决保存图像是负号'-'显示为方块的问题


def tidy_json(jsonstr):
    jsonres = re.sub(r"([,{]?)([a-z]+):", r"\1'\2':", jsonstr)
    jsonres = jsonres.replace("'", "\"")
    return jsonres


def get_stock_hexun(stock_num):
    url = 'http://bdcjhq.hexun.com/quote?s2=%s' % stock_num
    values = urllib.urlopen(url).read()
    # 使用Pattern匹配文本，获得匹配结果，无法匹配时将返回None
    match = re.search(u'\{".+\}\}', values)
    jsonData = {}
    if match:
        # 使用Match获得分组信息
        data = match.group()
        print(tidy_json(data))
        print(json.loads(tidy_json(data), encoding='GBK', strict=False))
        jsonData = json.loads(tidy_json(data), encoding='GBK', strict=False)["002041.sz"]
    return jsonData


def get_stock_info(stock_num):
    yahooflag = 'l1c1va2xj1b4j4dyekjm3m4rr5p5p6s7n'
    url = 'http://finance.yahoo.com/d/quotes.csv?s=%s&f=%s' % (stock_num, yahooflag)
    values = urllib.urlopen(url).read().strip().split(',')
    print(values)
    data = {'price': values[0], 'change': values[1], 'volume': values[2], 'avg_daily_volume': values[3],
            'stock_exchange': values[4], 'market_cap': values[5], 'book_value': values[6], 'ebitda': values[7],
            'dividend_per_share': values[8], 'dividend_yield': values[9], 'earnings_per_share': values[10],
            '52_week_high': values[11], '52_week_low': values[12], '50day_moving_avg': values[13],
            '200day_moving_avg': values[14], 'price_earnings_ratio': values[15],
            'price_earnings_growth_ratio': values[16], 'price_sales_ratio': values[17], 'price_book_ratio': values[18],
            'short_ratio': values[19], 'name': values[20]}
    return data


def quote_track(stock_num):
    from matplotlib.dates import DateFormatter, WeekdayLocator, DayLocator, MONDAY
    from matplotlib.finance import quotes_historical_yahoo, candlestick
    from matplotlib.font_manager import FontProperties

    font = FontProperties(fname=r"c:\windows\fonts\simsun.ttc", size=18)
    # 定义起始、终止日期和股票代码
    date2 = datetime.date.today()
    date1 = date2 - datetime.timedelta(days=30)
    # 定义日期格式
    mondays = WeekdayLocator(MONDAY)
    alldays = DayLocator()
    weekFormatter = DateFormatter('%b %d')
    # 获取股票数据
    quotes = quotes_historical_yahoo(stock_num, date1, date2)
    if len(quotes) == 0:
        raise SystemExit
    print(quotes)
    # 绘制蜡烛线或美国线
    fig = figure()
    fig.subplots_adjust(bottom=0.2)
    ax = fig.add_subplot(111)
    ax.xaxis.set_major_locator(mondays)
    ax.xaxis.set_minor_locator(alldays)
    ax.xaxis.set_major_formatter(weekFormatter)
    #注释掉下面的其中一行，可以得到蜡烛线或美国线
    candlestick(ax, quotes, width=0.6)
    #plot_day_summary(ax, quotes, ticksize=3)
    ax.xaxis_date()
    ax.autoscale_view()
    setp(gca().get_xticklabels(), rotation=45, horizontalalignment='right')
    stockdata = get_stock_hexun(stock_num)
    title(
        u'%s %s 昨收盘：%s 今开盘：%s 当前价：%s' % (stock_num, stockdata['na'], stockdata['pc'], stockdata['op'], stockdata['la']),
        fontproperties=font)
    show()


if __name__ == "__main__":
    #get_stock_hexun("002041.sz")
    quote_track("002041.sz")
    print(re.sub(r"(,?)([^:{,\"]+?)\s?:", r"\1'\2':", '{"002041.sz":{na:"登海种业",pc:"30.15",op:"30.15",vo:"41364",'))