'''
    from   oneclass.onedata import Onedata
    import os,math,sys,time,re,json,datetime,requests,warnings,logging,xlrd,traceback
    import numpy as np
    from   sqlalchemy import create_engine
    import sqlalchemy  #初始化数据库连接，使用pymysql模块
    import pymysql
    import multiprocessing as mp
'''
from   oneclass.onedb   import Onedb
import pandas as pd
import tushare as ts

ts.set_token('8f6e67ec8a48922bbb531b95687625d99ee263940a07561ae95f4648')  # Tushare Pro
pd.set_option('mode.chained_assignment', None)
# warnings.filterwarnings("ignore")

'''
pro = ts.pro_api()
stocklist = pro.query('stock_basic', exchange='', list_status='L')

stocklist = stocklist.append([{'ts_code': "000001.SH", 'name': "上证综指"}], ignore_index=True)                       
stocklist = stocklist.append([{'ts_code': "399006.SZ", 'name': "创业板指"}], ignore_index=True)
stocklist = stocklist.append([{'ts_code': "399001.SZ", 'name': "深圳成指"}], ignore_index=True)

print(stocklist)
'''
sqlstring="select tusharestocklist.name,pe.stockname,tusharestocklist.ts_code,pe.stockcode from tusharestocklist left join pe on tusharestocklist.ts_code=pe.stockcode"
onedb=Onedb()
pestocklist=onedb.execute_sqlnew(sqlstring)
print("pestocklist",pestocklist)
pestocklist.to_csv('pestocklist.csv', index=False)


'''
UPDATE tusharestocklist JOIN pe ON tusharestocklist.ts_code = pe.stockcode SET tusharestocklist.name = pe.stockname
WHERE table1.data IS NULL OR table1.data = '';
'''