tools/scripts/build_wid/build_wid.py
2018-12-04 15:40:49 +08:00

118 lines
2.6 KiB
Python

import pymysql
import hashlib
print('runing........')
host_name = "127.0.0.1"
user_name = "root"
password = "keji178"
database_name= "accountdb" + "1"
table_name = "accounts"
database_config = {
'host_name' : host_name,
'user_name' : user_name,
'password' : password,
'database_name' : database_name
}
select_config = [
'country',
'province',
'city',
'sex',
'nickname',
'idx',
'accountid'
]
def getDatabaseCursor(config):
#print('getDatabaseCursor\n')
db = pymysql.connect(config['host_name'],config['user_name'],config['password'],config['database_name'])
return db.cursor()
def getSelectInfo(min_idx,max_idx,s_config):
#print('getSelectInfo\n')
try:
cursor = getDatabaseCursor(s_config)
select_sql = 'select country, province, city, sex, nickname , idx, accountid \
from accounts where idx > %s and idx <%s and accountid is not null' % (min_idx,max_idx)
cursor.execute(select_sql)
ret = cursor.fetchall()
except:
print('sql error!')
return
return ret
def translateToMd5(translate_info):
#print('translateTomd5\n')
if translate_info:
m = hashlib.md5()
b = translate_info.encode(encoding = 'utf-8')
m.update(b)
return m.hexdigest()
return
def saveTofile(save_info,save_name):
#print('saveTofile\n')
if save_info and save_name :
with open(save_name,'a') as f:
f.write("%s\n"%(save_info))
return
def buildWid():
select_min_idx = 10000
select_max_idx = select_min_idx + 100
finsh_flag = 1;
select_count = 0;
save_file_name = 'updata_wid.txt'
while select_count < 100 and finsh_flag:
min_idx = select_min_idx + select_count * 100
max_idx = select_max_idx + select_count * 100
select_info = getSelectInfo(min_idx,max_idx,database_config)
if select_info:
for player_info in select_info:
country = player_info[0]
province = player_info[1]
city = player_info[2]
sex = player_info[3]
nickname = player_info[4]
idx = player_info[5]
accountid = player_info[6]
new_str = '%s%s%s%s%s'%(country, province, city, sex, nickname)
md5_str = translateToMd5(new_str)
update_sql = "UPDATE %s SET wid = '%s' where accountid = '%s';"%(table_name,md5_str,accountid)
saveTofile(update_sql,save_file_name)
select_count = select_count + 1
else:
finsh_flag = 0
print('buidWid END!')
return
buildWid()