[python]抓取股票交易记录

#!/usr/bin/python#DownByDate.py sh600115 2014-12-29 2015-3-15#DownByDate.py stock_num start_date end_date#http://stock.gtimg.cn/data/index.php?appn=detail&action=download&c=sh600115&d=20141229#sh600115_2014-12-29.txtimport sysimport urllibimport datetimedef download_date(src_url,dest_file):download=urllib.FancyURLopener();download_page=download.open(src_url);savefile=file(dest_file,'wb+');while True:arr = download_page.read();if len(arr)==0:break;savefile.write(arr);savefile.flush();savefile.close();returnstock_code=sys.argv[1]str_0='''http://stock.gtimg.cn/data/index.php?appn=detail&action=download&c='''str_0=str_0 + stock_code + '&d='date_start=datetime.datetime.strptime(sys.argv[2],'%Y-%m-%d')if len(sys.argv)>3:date_end=datetime.datetime.strptime(sys.argv[3],'%Y-%m-%d')else:date_end=date_start+datetime.timedelta(days=1)while date_start<date_end:str_date=date_start.strftime('%Y%02m%02d')str_url=str_0+str_datestr_file=stock_code + '_' + date_start.strftime('%Y-%02m-%02d') + '.txt'download_date(str_url,str_file)print str_filedate_start=date_start+datetime.timedelta(days=1)