#!/usr/bin/python3 """ 批量删除table,只保留最近几天的table """ import pymysql import re def conn_(host='',usr='',passwd='',db='',port=3306,): conn = pymysql.connect(host, usr, passwd, db, port,charset='utf8') return conn def del_table(conn_,table_pre='',table_suff='%Y%m%d',keep_count=3): date_form = None if table_suff == "%Y%m%d": date_form = "_(\d{4}\d{1,2}\d{1,2})$" date_len = 8 elif table_suff == "%Y-%m-%d": date_form = "_(\d{4}-\d{1,2}-\d{1,2})$" date_len = 10 elif table_suff == "%Y%m": date_form = "_(\d{4}\d{1,2})$" date_len = 6 elif table_suff == "%Y-%m": date_form = "_(\d{4}-\d{1,2})$" date_len = 7 else: raise Exception("暂时不支持其他类型的时间后缀") curs = conn_.cursor() curs.execute('SHOW TABLES') data = curs.fetchall() table_ = r'%s'%table_pre+date_form list_table = [] i = 0 for table in data: mt = re.search(table_, table[0]) if mt: if len(mt.groups()[0]) == date_len: list_table.append((table[0], mt.groups()[0])) i += 1 sorted(list_table, key=lambda date: date[1]) #按照表结构后缀时间升序排序 for j in range(i-keep_count): sql = 'DROP TABLE if exists %s'%list_table[j][0] curs.execute(sql) curs.close() conn_.close() if __name__ == '__main__': table_pre = "tree_product" table_suff = "%Y%m%d" # table_suff = "%Y-%m-%d" # table_suff = "%Y%m" # table_suff = "%Y-%m" conn=conn_('10.0.0.11','root','sctele@root','sxf',port=3306) del_table(conn,table_pre=table_pre,table_suff=table_suff,keep_count=1)
批量删除table,只保留最近几天的table
最新推荐文章于 2022-08-17 18:57:12 发布