1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
| """ Created on Wed Nov 25 09:02:55 2020
@author: lixin """
import mysql.connector as mysql import pandas as pd db = mysql.connect( host = "127.0.0.1", user = "root", passwd = "121457", database = "tpc2" )
data = pd.read_sql("SELECT TABLE_NAME,TABLE_COMMENT FROM information_schema.TABLES",db)
dict_comments_tab = pd.read_excel("comments/user_tab_comments.xlsx")
def prc(row, df): dff = df[(df['TABLE_NAME'] == row.upper())] res = [i for i in dff['COMMENTS']] return res[0] if len(res) > 0 else None
data['注释'] = data.apply(lambda x: prc(x['TABLE_NAME'],dictjj),axis = 1) data = data[pd.notnull(data['注释'])] data2 = data[data['TABLE_COMMENT'] == ''] data2.to_csv('tab_commentsII.csv',index=0)
cursor = db.cursor() res = [] for index, row in data2.iterrows(): val = row['注释'] table = row['TABLE_NAME'] sql = ''' alter table {} comment '{}';'''.format(table,val) res.append(sql)
file_name = 'write_tab_comment.txt' with open(file_name,'w') as file_obj: for i in res: file_obj.write(i+'\n')
|