用Python获取沪深两市上市公司股票信息，提取创近10天股价新高的、停牌的、复牌不超过一天或者新发行的股票，并存入mysql数据库

#该脚本可以提取沪深两市上市公司股票信息，并按以下信息分类：（1）当天股价创近10个交易日新高的股票；（2）停牌的股票；（3）复牌不超过一个交易日或者新发行的股票
#将分类后的股票及其信息（股价新高、当前状态等）存入mysql数据库
from time import *
import pandas as pd
import tushare as ts
from datetime import date
import datetime
import MySQLdb
import sys
reload(sys)
sys.setdefaultencoding('utf8')        #将编码方式默认改为utf-8
#-------------------------------定义函数：计算当天往前推Deltadays个交易日所对应的日期--------------------------
def com_StartDate(Deltadays):
    i=0
    da=date.today()
    while i!=Deltadays:
        da=da-datetime.timedelta(days=1)
        if da.isoweekday()==6 or da.isoweekday()==7:   #判断当前日期是否为周末（此处未考虑法定节假日，后续改进）
            pass
        else:
            i+=1
    return da.strftime("%Y-%m-%d")   #返回值数据类型为string

#--------------定义函数：判断某支股票处于什么状态，并返回股票代码、名字和股价（对于创新高的股票）等信息--------
def stock_info(stockID,startdate,index):
    today=date.today()
    df=ts.get_hist_data(stockID,start=startdate,end=today.strftime("%Y-%m-%d")) #获取股票信息
    if type(df)!=type(None) and len(df.index)>1:                       #抓取时间周期内，有数据天数低于2天的，算作新股
        df=df.reindex(index=index,method='ffill',fill_value=int(-1))
        period_high=df['high'].max()
        today_high=df.ix[today.strftime("%Y-%m-%d")]['high']
        if today_high>=period_high and today_high>0 :
            stock_status=int(0)         #非新股，未停牌，股价创新高
            return stock_status,df
        elif today_high==int(-1):
            stock_status=int(1)        #非新股，但停牌
            return stock_status,df
        else:
            stock_status=int(2)       #非新股，未停牌，股价未创新高
            return stock_status,df
    else:
        stock_status=int(3)           #新股或复牌不超过1天的股票
        return stock_status,df

#-----------------------------------------------获取沪深两市所有 上市公司基本信息---------------------------
all_stocks_info=ts.get_stock_basics()
Deltadays=10
startdate=com_StartDate(Deltadays)
newRecStocksID=[]                    #股票价格创新高的股票代码列表
newRecStocksPrice=[]                 #创新高的股票价格列表
newRecStocksName=[]                  #股票价格创新高的股票名称列表
suspendedStocksID=[]                #停牌股票代码列表
suspendedStocksName=[]              #停牌股票名称列表
newORresumeStocksID=[]                      #新发行或者当天复牌的股票代码列表
newORresumeStocksName=[]                    #新发行或者当天复牌的股票名称列表
today=date.today()
index=ts.get_hist_data('002625',start=startdate,end=today.strftime("%Y-%m-%d"),retry_count=10,pause=1).index
for i in all_stocks_info.index:
    stock_status,stockinfo=stock_info(str(i), startdate,index)
    if  stock_status==int(0):                                #当日股价创Deltadays天内新高的股票
        newRecStocksID.append(str(i).zfill(6))
        newRecStocksName.append(all_stocks_info.ix[str(i),'name'])
        newRecStocksPrice.append(stockinfo.ix[today.strftime("%Y-%m-%d"),'high'])
    elif stock_status==int(1):                              #已停牌的股票
        suspendedStocksID.append(str(i).zfill(6))
        suspendedStocksName.append(all_stocks_info.ix[str(i), 'name'])
    elif stock_status==int(2):                             #未停牌，且股价未创Deltadays天内新高的股票
        pass
    else:                                                  #该股票为新发行或者复牌不超过一天的股票
        newORresumeStocksID.append(str(i).zfill(6))
        newORresumeStocksName.append(all_stocks_info.ix[str(i), 'name'])

#-------------------------------------------数据写入DataFrame----------------------------------------------
newRecStocks=pd.DataFrame({'stockID':newRecStocksID,'stockname':newRecStocksName,'high_price':newRecStocksPrice,'status':['NewRecord']*len(newRecStocksID)})
suspendedStocks=pd.DataFrame({'stockID':suspendedStocksID,'stockname':suspendedStocksName,'status':['Suspended']*len(suspendedStocksID)})
newORresumeStocks=pd.DataFrame({'stockID':newORresumeStocksID,'stockname':newORresumeStocksName,'status':['New or Resume']*len(newORresumeStocksID)})

#-----------------------------------将数据存入MySQL数据库，数据库名称：stocks------------------------------
db=MySQLdb.connect('localhost','root','900417','stocks',charset='utf8')
pd.io.sql.to_sql(frame=newRecStocks,con=db,name=today.strftime("%Y%m%d")+'newRecordStocks',flavor='mysql',schema=None,if_exists='replace',index=True,index_label=None,chunksize=1000,dtype=None)
pd.io.sql.to_sql(frame=suspendedStocks,con=db,name=today.strftime("%Y%m%d")+'suspendedStocks',flavor='mysql',schema=None,if_exists='replace',index=True,index_label=None,chunksize=1000,dtype=None)
pd.io.sql.to_sql(frame=newORresumeStocks,con=db,name=today.strftime("%Y%m%d")+'newORresumeStocks',flavor='mysql',schema=None,if_exists='replace',index=True,index_label=None,chunksize=1000,dtype=None)
db.commit()
db.close()