wind__质押股份

# coding=utf-8
from WindPy import *
import pandas as pd
import time
from sqlalchemy import create_engine
import pymysql
import datetime


def pledged_shares():       # 质押股份

    # 本代码已经经过测试 数据收取无误  质押股份是按照周频来收取的
    # 运行本代码endtime 需要设置为周五 因为质押股份为周频数据
    # 数据一般都是周五更新 如果endtime不是周五 最后一天的数据会和上一个周五数据相同

    w.start()
    today_time = time.strftime('%Y-%m-%d', time.localtime(time.time()))

    engine = create_engine(
        'mysql+pymysql://public:123456@192.168.0.22:3307/wind?charset=utf8')
    stock_list = ["000002.SZ"]

    df_stock = pd.read_csv('stock_list.csv')
    list_df = df_stock['list'].values
    stock_list = list_df.tolist()

    start_time = '2010-01-01'
    end_time = '2018-06-20'
    for stock in stock_list:

        datas = w.wsd(stock, "share_pledgeda,share_pledgeda_pct,share_liqa_pledged,share_liqa_pledgedpct,share_restricteda_pledged,share_restricteda_pledgedpct",
                      start_time, end_time, "unit=1;Period=W")

        if datas.Data[0] != ['CWSDService: No data.'] or datas.Data[0] != ['CWSDService: quota exceeded.']:
            today_time_h = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
            df = pd.DataFrame(index=None)

            df['datetime'] = datas.Times

            df['share_pledgeda'] = datas.Data[0]
            df['share_pledgeda_pct'] = datas.Data[1]
            df['share_liqa_pledged'] = datas.Data[2]
            df['share_liqa_pledgedpct'] = datas.Data[3]
            df['share_restricteda_pledged'] = datas.Data[4]
            df['share_restricteda_pledgedpct'] = datas.Data[5]

            df['insertime'] = today_time_h
            df['stock_code'] = stock

            print(df)
            df.to_sql('pledged_shares', engine,
                      if_exists='append', index=None)

            del df
        else:
            print(stock)
            break

    w.stop()


if __name__ == '__main__':
    pledged_shares()