test_rxdp_test.py 678 B

1234567891011121314151617181920212223
  1. import os
  2. from tmp.mysql.sql_util import Database
  3. # 查询所有表
  4. db = Database('mysql.rxdptest.svc.k5.bigtree.zone', 'test_user', 'tmOxmmc+3jznq2cX', 'dip_message')
  5. query = db.query(f"SELECT table_name FROM information_schema.tables WHERE table_schema = DATABASE();")
  6. allTable = set([x[0] for x in query])
  7. print(allTable)
  8. # 应该有的表
  9. sql_file_table = set("message_record message_record_detail".split(" "))
  10. print(sql_file_table)
  11. # 差集
  12. differenceTable = allTable.difference(sql_file_table)
  13. print(differenceTable)
  14. # 生产sql
  15. with open("tmp.sql", "w") as f:
  16. for x in differenceTable:
  17. f.write(f"DROP TABLE IF EXISTS `{x}`;")
  18. f.write(os.linesep)