#!/usr/bin/python
|
# -*- coding: GBK -*-
|
#-------------------------------------------------------------------------------
|
# writer:ifo
|
# @copyright: Copyright (c) 2004, 2010 Xiamen Wangyou Network Technology Co.,Ltd., All rights reserved.
|
#-------------------------------------------------------------------------------
|
#
|
|
|
|
#ÅúÁ¿×¢²áÕʺÅ
|
|
import os
|
import sys
|
import md5
|
import uuid
|
import logging
|
import pymongo
|
import MangoDBCommon
|
import AccountInfo
|
from time import localtime
|
|
import MongoDBConfig
|
import CommFunc
|
import tagDSAccountReader
|
import tagDBPlayerReader
|
import tagRoleItemReader
|
import tagRolePetReader
|
import tagPetSkillReader
|
import tagRoleSkillReader
|
|
now = localtime()
|
file = "%s-%d-%d-%d.log" % ( "MongoGMAcc", now.tm_year, now.tm_mon, now.tm_mday )
|
logging.basicConfig(level=logging.DEBUG,
|
format='%(name)-12s %(asctime)s %(name)-12s %(levelname)-8s %(message)s',
|
datefmt='%Y-%m-%dT%H:%M:%S',
|
filename=file,
|
filemode='a+')
|
|
tagDSAccountReader.ReadConfig("tagDSAccount.txt")
|
tagDSAccountConfig = tagDSAccountReader.GetConfig()
|
|
tagDBPlayerReader.ReadConfig("tagDBPlayer.txt")
|
tagDBPlayerConfig = tagDBPlayerReader.GetConfig()
|
|
tagRoleItemReader.ReadConfig("tagRoleItem.txt")
|
tagRoleItemConfig = tagRoleItemReader.GetConfig()
|
|
tagRolePetReader.ReadConfig("tagRolePet.txt")
|
tagRolePetConfig = tagRolePetReader.GetConfig()
|
|
tagPetSkillReader.ReadConfig("tagPetSkill.txt")
|
tagPetSkillConfig = tagPetSkillReader.GetConfig()
|
|
tagRoleSkillReader.ReadConfig("tagRoleSkill.txt")
|
tagRoleSkillConfig = tagRoleSkillReader.GetConfig()
|
|
class MongoGMAcc:
|
|
def __init__(self):
|
self.connection = 0
|
self.userDb = 0
|
self.regDB = None
|
self.userList = []
|
|
def LoginMongoDB(self,user, pwd, ip = '', port = '', connection = None ):
|
try:
|
if not connection:
|
connection = pymongo.Connection(ip, port)
|
db = connection.admin
|
return db.authenticate(user, pwd)
|
except Exception,e:
|
print 'Login mongodb failed:user = %s,pwd = %s'%(user, pwd)
|
logging.error('Login mongodb failed:user = %s,pwd = %s'%(user, pwd))
|
return False
|
|
def ConnectServer(self):
|
try:
|
self.connection = pymongo.Connection(MongoDBConfig.USER_DB_IP, MongoDBConfig.USER_DB_PORT)
|
except:
|
print 'Connect mongo db server %s:%s failed!'%(MongoDBConfig.USER_DB_IP, MongoDBConfig.USER_DB_PORT)
|
logging.error('Connect mongo db server %s:%s failed!'%(MongoDBConfig.USER_DB_IP, MongoDBConfig.USER_DB_PORT))
|
return
|
|
print 'Connect mongo db server %s:%s ok!'%(MongoDBConfig.USER_DB_IP, MongoDBConfig.USER_DB_PORT)
|
logging.info('Connect mongo db server %s:%s ok!'%(MongoDBConfig.USER_DB_IP, MongoDBConfig.USER_DB_PORT))
|
isTrue,psw = CommFunc.Decrypt(MongoDBConfig.USER_DB_Password)
|
|
if not isTrue:
|
print 'decode psw fail!'
|
logging.error('decode psw fail!')
|
return
|
|
if not self.LoginMongoDB(MongoDBConfig.USER_DB_UserName, psw, connection = self.connection):
|
return
|
|
self.userDb = self.connection[MongoDBConfig.USER_DB_NAME]
|
self.regDB = self.connection[MongoDBConfig.REG_DB_NAME]
|
return True
|
|
def CreateAccountInfo(self,curUser,commentLine):
|
|
rec = AccountInfo.tagDSAccount()
|
|
for i in range(len(commentLine)):
|
print '---',commentLine[i],getattr(curUser,commentLine[i]), '++++'
|
setattr(rec,commentLine[i],getattr(curUser,commentLine[i]))
|
|
accid = getattr(curUser,"ACCID")
|
setattr(rec,"ACCID",accid.strip().lower())
|
|
lastExceptionCnt = MangoDBCommon.getADOExceptionCount()
|
lastSIDErrorCnt = MangoDBCommon.getSIDErrorCnt()
|
if rec.adoLoadC(self.userDb.tagDSAccount):
|
print 'Accid = %s exists!'%accid
|
logging.warning('Accid = %s exists!'%accid)
|
return
|
else:
|
if lastExceptionCnt != MangoDBCommon.getADOExceptionCount():
|
#Êý¾Ý¿â·ÃÎÊÒì³£
|
print 'DB access excption!'
|
logging.error('DB access excption!')
|
return
|
if lastSIDErrorCnt != MangoDBCommon.getSIDErrorCnt():
|
#Óû§È·Êµ²»´æÔÚ
|
print 'accid = %s not found!'%accid
|
logging.error('accid = %s not found!'%accid)
|
return
|
|
|
#ÃÜÂë½øÐÐ2´ÎMD5
|
psw = getattr(curUser,"Psw")
|
psw = psw.strip().lower()
|
md5Obj1 = md5.new()
|
md5Obj1.update(psw)
|
psw = md5Obj1.hexdigest()
|
|
md5Obj2 = md5.new()
|
md5Obj2.update(psw)
|
psw = md5Obj2.hexdigest()
|
|
setattr(rec,"Psw",psw.strip().lower())
|
|
if not rec.adoInsertC(self.userDb.tagDSAccount):
|
print 'Reg accid = %s failed!'%accid
|
logging.warning('Reg accid = %s failed!'%accid)
|
return
|
|
print 'Reg accid = %s ok'%accid
|
logging.info('Reg accid = %s ok'%accid)
|
|
|
def hasPlayerByPlayerName(self, db, playerName):
|
collection = db.tagDBPlayer
|
#ÕýÔò±í´ïʽ£¬ÊµÏÖ´óСд²»Ãô¸Ð²éѯ
|
name = '^%s$'%playerName
|
resultCollection = collection.find({'PlayerName':{'$regex':MangoDBCommon.fix_incomingText(playerName), '$options':'i'}, 'IsDeleted':0})
|
return (resultCollection.count() != 0)
|
|
def CreatePlayer(self,curUser,commentLine):
|
|
createPlayer = AccountInfo.tagDBPlayer()
|
|
createPlayer.IsDeleted = 0
|
createPlayer.AccState = 1
|
createPlayer.GMLevel = 90
|
createPlayer.HPRestoreSetting = 20
|
createPlayer.MPRestoreSetting = 20
|
createPlayer.UseGoldType = 2
|
createPlayer.UseSilverType = 4
|
createPlayer.SID = 2266520
|
|
for i in range(len(commentLine)):
|
setattr(createPlayer,commentLine[i],getattr(curUser,commentLine[i]))
|
|
#ÕʺÅͳһתСд
|
accid = getattr(curUser,"AccID")
|
setattr(createPlayer,"AccID",accid.strip().lower())
|
|
self.DBPlayerCollection = self.userDb.tagDBPlayer
|
|
#¼ì²é¸ÃÕʺÅÊÇ·ñÒÑÓÐÓÐЧ½ÇÉ«
|
if createPlayer.adoLoadCEx(self.userDb.tagDBPlayer, {'AccID':createPlayer.AccID, 'IsDeleted':createPlayer.IsDeleted}):
|
print 'player of accid = %s exists!'%createPlayer.AccID
|
logging.warning('player of accid = %s exists!'%createPlayer.AccID)
|
return
|
|
#¼ì²é½ÇÉ«ÃûÊÇ·ñÖظ´
|
if self.hasPlayerByPlayerName(self.userDb, createPlayer.PlayerName):
|
print 'player of playerName = %s exists!'%createPlayer.PlayerName
|
logging.warning('player of playerName = %s exists!'%createPlayer.PlayerName)
|
return
|
|
#Éú³É×ÔÔö³¤µÄplayerID
|
createPlayer.PlayerID = MangoDBCommon.seq(self.userDb, 'tagDBPlayer', 'PlayerID', MangoDBCommon.PLAYERID_FEED, MangoDBCommon.PLAYERID_STEP)
|
|
#УÑéͨ¹ý£¬ÍùÊý¾Ý¿âÖвåÈë½ÇÉ«
|
if not createPlayer.adoInsertC(self.userDb.tagDBPlayer):
|
print 'insert player failed!accid = %s'%createPlayer.AccID
|
logging.warning('insert player failed!accid = %s'%createPlayer.AccID)
|
return
|
|
#²åÈë½ÇÉ«³É¹¦
|
#¶ÁÈ¡½ÇÉ«ID
|
if not createPlayer.adoLoad(self.userDb.tagDBPlayer):
|
print 'insert player failed!accid = %s'%createPlayer.AccID
|
logging.warning('insert player failed!accid = %s'%createPlayer.AccID)
|
return
|
|
self.userList.append(createPlayer)
|
print 'insert role ok!accid = %s'%createPlayer.AccID
|
logging.info('insert role ok!accid = %s'%createPlayer.AccID)
|
|
|
def MakeRoleItem(self,curUser,commentLine,userItemDataDict):
|
createItem = AccountInfo.tagRoleItem()
|
|
userDataList = userItemDataDict.get((curUser.Sex,curUser.Job))
|
|
if userDataList == None:
|
print 'not this itemData Sex:%d,Job:%d accid = %s'%(curUser.Sex,curUser.Job,curUser.AccID)
|
logging.warning('not this itemData Sex:%d,Job:%d accid = %s'%(curUser.Sex,curUser.Job,curUser.AccID))
|
return
|
|
for userData in userDataList:
|
createItem.ItemGUID = str(uuid.uuid1())
|
createItem.PlayerID = curUser.PlayerID
|
|
for i in range(2,len(commentLine)):
|
setattr(createItem,commentLine[i],getattr(userData,commentLine[i]))
|
|
if not createItem.adoCheckUpdateC(self.userDb.tagRoleItem):
|
print 'insert Item failed!ItemGUID = %s'%createItem.ItemGUID
|
logging.warning('insert Item failed!ItemGUID = %s'%createItem.ItemGUID)
|
continue
|
|
print 'insert Item ok!accid = %s,ItemGUID = %s'%(curUser.AccID,createItem.ItemGUID)
|
logging.info('insert Item ok!accid = %s,ItemGUID = %s'%(curUser.AccID,createItem.ItemGUID))
|
|
|
|
def AddRolePet(self,curUser,commentLine,rolePetDataDict,petSkillCommentLine,petSkillDataDict):
|
createRolePet = AccountInfo.tagRolePet()
|
|
userDataList = rolePetDataDict.get((curUser.Sex,curUser.Job))
|
|
if userDataList == None:
|
print 'not this rolePetData Sex:%d,Job:%d accid = %s'%(curUser.Sex,curUser.Job,curUser.AccID)
|
logging.warning('not this rolePetData Sex:%d,Job:%d accid = %s'%(curUser.Sex,curUser.Job,curUser.AccID))
|
return
|
|
for userData in userDataList:
|
createRolePet.PlayerID = curUser.PlayerID
|
createRolePet.PetID = createRolePet.PlayerID*10
|
|
for i in range(2,len(commentLine)):
|
setattr(createRolePet,commentLine[i],getattr(userData,commentLine[i]))
|
|
|
if not createRolePet.adoCheckUpdate(self.userDb.tagRolePet):
|
print 'insert RolePet failed!PlayerID = %s,Name = %s'%(createRolePet.PlayerID,createRolePet.Name)
|
logging.warning('insert RolePet failed!PlayerID = %s,Name = %s'%(createRolePet.PlayerID,createRolePet.Name))
|
continue
|
|
print 'insert RolePet ok!PlayerID = %s,Name = %s'%(createRolePet.PlayerID,createRolePet.Name)
|
logging.info('insert RolePet ok!PlayerID = %s,Name = %s'%(createRolePet.PlayerID,createRolePet.Name))
|
|
createPetSkill = AccountInfo.tagPetSkill()
|
key = getattr(createRolePet,"NPCID")
|
if petSkillDataDict.has_key(key):
|
curPetSkill = petSkillDataDict[key]
|
for j in range(1,len(petSkillCommentLine)):
|
curPetSkillInfoList = getattr(curPetSkill,petSkillCommentLine[j])
|
for k in range(len(curPetSkillInfoList)):
|
createPetSkill.PetID = createRolePet.PetID
|
createPetSkill.PlayerID = createRolePet.PlayerID
|
setattr(createPetSkill,petSkillCommentLine[j],curPetSkillInfoList[k])
|
|
if not createPetSkill.adoCheckUpdate(self.userDb.tagPetSkill):
|
print 'insert PetSkill failed!PlayerID = %s,Name = %s,SkillID = %s'%(createRolePet.PlayerID,createRolePet.Name,createPetSkill.SkillID)
|
logging.warning('insert PetSkill failed!PlayerID = %s,Name = %s,SkillID = %s'%(createRolePet.PlayerID,createRolePet.Name,createPetSkill.SkillID))
|
continue
|
|
print 'insert PetSkill ok!PlayerID = %s,Name = %s,SkillID = %s'%(createRolePet.PlayerID,createRolePet.Name,createPetSkill.SkillID)
|
logging.info('insert PetSkill ok!PlayerID = %s,Name = %s,SkillID = %s'%(createRolePet.PlayerID,createRolePet.Name,createPetSkill.SkillID))
|
|
def AddRoleSkill(self,user,commentLine,roleSkillDataDict):
|
|
createRoleSkill = AccountInfo.tagRoleSkill()
|
|
key = getattr(user,"AccID")
|
if roleSkillDataDict.has_key(key):
|
curRoleSkill = roleSkillDataDict[key]
|
for j in range(1,len(commentLine)):
|
curRoleSkillInfoList = getattr(curRoleSkill,commentLine[j])
|
for k in range(len(curRoleSkillInfoList)):
|
createRoleSkill.PlayerID = user.PlayerID
|
setattr(createRoleSkill,commentLine[j],curRoleSkillInfoList[k])
|
|
if not createRoleSkill.adoCheckUpdate(self.userDb.tagRoleSkill):
|
print 'insert RoleSkill failed!PlayerID = %s,SkillID = %s'%(createRoleSkill.PlayerID,createRoleSkill.SkillID)
|
logging.warning('insert RoleSkill failed!PlayerID = %s,SkillID = %s'%(createRoleSkill.PlayerID,createRoleSkill.SkillID))
|
continue
|
|
print 'insert RoleSkill ok!PlayerID = %s,SkillID = %s'%(createRoleSkill.PlayerID,createRoleSkill.SkillID)
|
logging.info('insert RoleSkill ok!PlayerID = %s,SkillID = %s'%(createRoleSkill.PlayerID,createRoleSkill.SkillID))
|
|
|
|
if __name__ == '__main__':
|
|
newAccount = MongoGMAcc()
|
|
if not newAccount.ConnectServer():
|
os.system('pause')
|
|
UserCount = tagDSAccountConfig.GetRecordCount()
|
CommentLine = tagDSAccountConfig.GetCommentLine()
|
for i in range(UserCount):
|
curUser = tagDSAccountConfig.GetRecordByIndex(i)
|
newAccount.CreateAccountInfo(curUser,CommentLine)
|
|
UserCount = tagDBPlayerConfig.GetRecordCount()
|
CommentLine = tagDBPlayerConfig.GetCommentLine()
|
for i in range(UserCount):
|
curUser = tagDBPlayerConfig.GetRecordByIndex(i)
|
newAccount.CreatePlayer(curUser,CommentLine)
|
|
userItemDataCount = tagRoleItemConfig.GetRecordCount()
|
userItemDataDict={}
|
for i in range(userItemDataCount):
|
curUserItemIndex = tagRoleItemConfig.GetRecordByIndex(i)
|
userItemDataDict.update({(curUserItemIndex.Sex,curUserItemIndex.Job):[]})
|
|
for i in range(userItemDataCount):
|
curUserItemIndex = tagRoleItemConfig.GetRecordByIndex(i)
|
key = (curUserItemIndex.Sex,curUserItemIndex.Job)
|
if userItemDataDict.has_key(key):
|
userItemDataDict[key].append(curUserItemIndex)
|
|
CommentLine = tagRoleItemConfig.GetCommentLine()
|
for user in newAccount.userList:
|
newAccount.MakeRoleItem(user,CommentLine,userItemDataDict)
|
|
rolePetDataCount = tagRolePetConfig.GetRecordCount()
|
rolePetDataDict={}
|
for i in range(rolePetDataCount):
|
curRolePetIndex = tagRolePetConfig.GetRecordByIndex(i)
|
rolePetDataDict.update({(curRolePetIndex.Sex,curRolePetIndex.Job):[]})
|
|
for i in range(rolePetDataCount):
|
curRolePetIndex = tagRolePetConfig.GetRecordByIndex(i)
|
key = (curRolePetIndex.Sex,curRolePetIndex.Job)
|
if rolePetDataDict.has_key(key):
|
rolePetDataDict[key].append(curRolePetIndex)
|
|
petSkillDataCount = tagPetSkillConfig.GetRecordCount()
|
petSkillDataDict={}
|
for i in range(petSkillDataCount):
|
curPetSkillIndex = tagPetSkillConfig.GetRecordByIndex(i)
|
petSkillDataDict.update({curPetSkillIndex.NPCID:curPetSkillIndex})
|
|
CommentLine = tagRolePetConfig.GetCommentLine()
|
PetSkillCommentLine = tagPetSkillConfig.GetCommentLine()
|
for user in newAccount.userList:
|
newAccount.AddRolePet(user,CommentLine,rolePetDataDict,PetSkillCommentLine,petSkillDataDict)
|
|
roleSkillDataCount = tagRoleSkillConfig.GetRecordCount()
|
roleSkillDataDict={}
|
for i in range(roleSkillDataCount):
|
curRoleSkillIndex = tagRoleSkillConfig.GetRecordByIndex(i)
|
roleSkillDataDict.update({curRoleSkillIndex.AccID:curRoleSkillIndex})
|
|
CommentLine = tagRoleSkillConfig.GetCommentLine()
|
for user in newAccount.userList:
|
newAccount.AddRoleSkill(user,CommentLine,roleSkillDataDict)
|
|
|
