hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

import_time_zone_info.py adapt to python3

Browse Source
This commit is contained in:
sdc
2024-07-10 10:15:06 +00:00
committed by ob-robot
parent 790b27297f
commit 85a4d29528
1 changed files with 17 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
tools/import_time_zone_info.py
View File

@ -4,8 +4,8 @@ import re
import os import os
import getopt import getopt
import mysql.connector import mysql.connector
import argparse
from mysql.connector import errorcode from mysql.connector import errorcode
import argparse
class TimeZoneInfoImporter: class TimeZoneInfoImporter:
def get_args(self): def get_args(self):
@ -66,22 +66,22 @@ class TimeZoneInfoImporter:
def connect_server(self): def connect_server(self):
self.conn = mysql.connector.connect(user='root', password=self.pwd, host=self.host, port=self.port, database='mysql') self.conn = mysql.connector.connect(user='root', password=self.pwd, host=self.host, port=self.port, database='mysql')
self.cur = self.conn.cursor(buffered=True) self.cur = self.conn.cursor(buffered=True)
print "INFO : sucess to connect server {0}:{1}".format(self.host, self.port) print ("INFO : sucess to connect server {0}:{1}".format(self.host, self.port))
try: try:
sql = "select value from oceanbase.__all_sys_parameter where name = 'enable_upgrade_mode';" sql = "select value from oceanbase.__all_sys_parameter where name = 'enable_upgrade_mode';"
self.cur.execute(sql) self.cur.execute(sql)
print "INFO : execute sql -- {0}".format(sql) print ("INFO : execute sql -- {0}".format(sql))
result = self.cur.fetchall() result = self.cur.fetchall()
if 1 == len(result) and 1 == result[0][0]: if 1 == len(result) and 1 == result[0][0]:
self.upgrade_mode = True self.upgrade_mode = True
else: else:
self.upgrade_mode = False self.upgrade_mode = False
sql = "select tenant_id from oceanbase.__all_tenant where tenant_name = '{0}';".format(str(self.tenant)) sql = "select tenant_id from oceanbase.__all_tenant where tenant_name = '{0}';".format(str(self.tenant))
print "INFO : execute sql -- {0}".format(sql) print ("INFO : execute sql -- {0}".format(sql))
self.cur.execute(sql) self.cur.execute(sql)
result = self.cur.fetchall() result = self.cur.fetchall()
if 1 == len(result): if 1 == len(result):
print "tenant_id = {0}".format(str(result[0][0])) print ("tenant_id = {0}".format(str(result[0][0])))
self.tenant_id = result[0][0] self.tenant_id = result[0][0]
else: else:
self.tenant_id = 0 self.tenant_id = 0
@ -93,18 +93,17 @@ class TimeZoneInfoImporter:
except mysql.connector.Error as err: except mysql.connector.Error as err:
print("ERROR : " + sql) print("ERROR : " + sql)
print(err) print(err)
raise raise
def execute_sql(self): def execute_sql(self):
try: try:
for sql in self.sql_list: for sql in self.sql_list:
self.cur.execute(sql); self.cur.execute(sql);
print "INFO : execute sql -- {0}".format(sql) print ("INFO : execute sql -- {0}".format(sql))
except mysql.connector.Error as err: except mysql.connector.Error as err:
print("ERROR : " + sql) print("ERROR : " + sql)
print(err) print(err)
print("ERROR : fail to import time zone info") print("ERROR : fail to import time zone info")
raise raise
else: else:
print("INFO : success to import time zone info") print("INFO : success to import time zone info")
@ -112,7 +111,7 @@ class TimeZoneInfoImporter:
self.cur.execute("select count(*) from {0}".format(table_name)) self.cur.execute("select count(*) from {0}".format(table_name))
result = self.cur.fetchone() result = self.cur.fetchone()
self.result_count[idx] = result[0] self.result_count[idx] = result[0]
print "INFO : {0} record count -- {1}, expect count -- {2}".format(table_name, result[0], self.expect_count[idx]) print ("INFO : {0} record count -- {1}, expect count -- {2}".format(table_name, result[0], self.expect_count[idx]))
def check_result(self): def check_result(self):
self.result_count = [0, 0, 0, 0] self.result_count = [0, 0, 0, 0]
@ -127,12 +126,12 @@ class TimeZoneInfoImporter:
try: try:
for sql in self.tz_version_sql_list: for sql in self.tz_version_sql_list:
self.cur.execute(sql) self.cur.execute(sql)
print "INFO : execute sql -- {0}".format(sql) print ("INFO : execute sql -- {0}".format(sql))
except mysql.connector.Error as err: except mysql.connector.Error as err:
print("ERROR : " + sql) print("ERROR : " + sql)
print(err) print(err)
print("ERROR : fail to insert time zone version") print("ERROR : fail to insert time zone version")
raise raise
else: else:
print("INFO : success to insert time zone version") print("INFO : success to insert time zone version")
@ -140,13 +139,13 @@ def main():
tz_info_importer = TimeZoneInfoImporter() tz_info_importer = TimeZoneInfoImporter()
tz_info_importer.get_args() tz_info_importer.get_args()
try: try:
tz_info_importer.connect_server() tz_info_importer.connect_server()
if False == tz_info_importer.upgrade_mode: if False == tz_info_importer.upgrade_mode:
tz_info_importer.generate_sql() tz_info_importer.generate_sql()
tz_info_importer.execute_sql() tz_info_importer.execute_sql()
tz_info_importer.check_result() tz_info_importer.check_result()
except: except:
print("except error in main") print("except error in main")
if __name__ == "__main__": if __name__ == "__main__":
main() main()