from tkinter import *

import urllib.request

import re,os

import threading

from tkinter import filedialog

win=Tk()

win.title('海啸投资-股票数据下载')

win.geometry('200x120+400+200')

win.attributes("-toolwindow", 1)

win.resizable(width=False,height=False)

var = StringVar()

var.set('数据量较大，耐心等待！')

Label(win,textvariable=var).grid(row=1,column=0,padx=30,pady=10)

stock_CodeUrl = 'http://quote.eastmoney.com/stocklist.html'

#os.mkdir('stock_data')

#获取股票代码列表

def urlTolist(url):

    allCodeList = []

    html = urllib.request.urlopen(url).read()

    html = html.decode('gbk')

    s = r'<li><a target="_blank" href="http://quote.eastmoney.com/\S\S(.*?).html">'

    pat = re.compile(s)

    code = pat.findall(html)

    for item in code:

        if item[0]=='' or item[0]=='' or item[0]=='':

            allCodeList.append(item)

    return allCodeList

allCodelist = urlTolist(stock_CodeUrl)

def allDown():

    fileDir = filedialog.askdirectory()

    c = len(allCodelist)

    for code in allCodelist:

        stemp = '正在获取'+code+'股票数据...'

        var.set(stemp)

        #print('正在获取%s股票数据...'%code)

        if code[0]=='':

            url = 'http://quotes.money.163.com/service/chddata.html?code=0'+code+\

            '&end=20161231&fields=TCLOSE;HIGH;LOW;TOPEN;LCLOSE;CHG;PCHG;TURNOVER;VOTURNOVER;VATURNOVER;TCAP;MCAP'

        else:

            url = 'http://quotes.money.163.com/service/chddata.html?code=1'+code+\

            '&end=20161231&fields=TCLOSE;HIGH;LOW;TOPEN;LCLOSE;CHG;PCHG;TURNOVER;VOTURNOVER;VATURNOVER;TCAP;MCAP'

        urllib.request.urlretrieve(url,fileDir+'\\'+code+'.csv')

        c-=1

        if c==0:

            var.set('所有股票下载完毕！')

def creatDownP():

    p = threading.Thread(target = allDown)

    p.setDaemon(True)#守护进程

    p.start()

Button(win,text='所有股票下载',width=20,command=creatDownP).grid(row=0,column=0,padx=30,pady=10)

mainloop()