#!/usr/bin/python
|
# -*- coding: GBK -*-
|
#
|
##@package E:/GameSVN/TeamTool/PythonScribe/PyModule/Coupon/CouponDB.py
|
# @todo:
|
#
|
# @author: Alee
|
# @date 2017-6-2 ÏÂÎç05:21:48
|
# @version 1.0
|
#
|
# @note:
|
#
|
#---------------------------------------------------------------------
|
#-------------------------------------------------------------------------------
|
#¼ì²é²¢¸üÐÂË÷Òý
|
#-------------------------------------------------------------------------------
|
from lib import ReadConfig
|
from lib import DBController
|
from lib import mylog
|
import os
|
|
g_dbController = None
|
|
CouponWXColName = "tagCouponWXSC" # ΢ÐÅÉ̳Ç
|
|
CouponBatchColName = "tagCouponBatch"
|
CouponCodeColName = "tagCouponCode"
|
ModuleName = "CouponDB"
|
|
Def_ConfigPath = os.getcwd() + "\\..\\Coupon\\config.ini"
|
|
|
def InitDB():
|
global g_dbController
|
|
if not g_dbController:
|
IP = ReadConfig.ReadConfig(Def_ConfigPath).GetValue("MongoDB", "IP")
|
Port = ReadConfig.ReadConfig(Def_ConfigPath).GetInt("MongoDB", "Port")
|
User = ReadConfig.ReadConfig(Def_ConfigPath).GetValue("MongoDB", "User")
|
Pwd = ReadConfig.ReadConfig(Def_ConfigPath).GetValue("MongoDB", "Pwd")
|
DBName = ReadConfig.ReadConfig(Def_ConfigPath).GetValue("MongoDB", "DBName")
|
g_dbController = DBController.DBController(IP, Port, DBName, User, Pwd, None)
|
|
if g_dbController.connected:
|
#Ö»ÔÚÁ´½ÓDB¼ì²éÒ»´Î
|
result = CheckAndUpdateIndexOnDb(g_dbController.db)
|
if not result:
|
mylog.debug('Ë÷ÒýÓÐÎÊÌâ')
|
return False
|
mylog.debug('»ñȡеÄÊý¾Ý¿âÁ´½Ó')
|
|
if not g_dbController.connected:
|
mylog.debug('ÎÞ·¨Á´½ÓÊý¾Ý¿â')
|
return False
|
|
return True
|
|
def GetDBEventCon():
|
global g_dbController
|
if not InitDB():
|
return None
|
return g_dbController
|
|
|
|
|
|
def ReadIndexInfo():
|
indexInfoDict = {
|
|
CouponBatchColName:
|
{
|
'CouponBatch_1':
|
{
|
'unique':True,
|
'key':[('couponid', 1)] #Ë÷ÒýµÄ×ÖµäºÍµÝÔö(1)»òµÝ¼õ(-1)
|
}
|
},
|
|
}
|
return indexInfoDict
|
|
def CompareIndex(curIndexDict, expectIndexDict):
|
#±È½Ïunique
|
curUnique = curIndexDict.get('unique', None)
|
expectUnique = expectIndexDict.get('unique', None)
|
if curUnique is None:
|
if expectUnique is None:
|
pass
|
elif expectUnique:
|
# print '#debug 1'
|
return False
|
else:
|
pass
|
else:
|
if expectUnique is None:
|
if (not curUnique):
|
return True
|
# print '#debug 2'
|
return False
|
if curUnique and (not expectUnique):
|
# print '#debug 3'
|
return False
|
if (not curUnique) and expectUnique:
|
# print '#debug 4'
|
return False
|
#±È½Ïkeylist
|
curKeyList = curIndexDict.get('key', None)
|
expectKeyList = expectIndexDict.get('key', None)
|
if curKeyList is None:
|
if expectKeyList is None:
|
pass
|
else:
|
# print '#debug 5'
|
return False
|
else:
|
if expectKeyList is None:
|
# print '#debug 6'
|
return False
|
else:
|
for pair in curKeyList:
|
if pair not in expectKeyList:
|
# print '#debug 7'
|
return False
|
for pair in expectKeyList:
|
if pair not in curKeyList:
|
# print '#debug 8'
|
return False
|
return True
|
|
def CheckIndexes(db, colName, expectIndexInfo):
|
col = db[colName]
|
|
'''¼ì²é±íÖÐÊÇ·ñÓжàÓàµÄË÷Òý'''
|
indexDict = col.index_information()
|
for k, v in indexDict.items():
|
if k == '_id_':
|
continue
|
if k not in expectIndexInfo:
|
return False
|
if not CompareIndex(v, expectIndexInfo[k]):
|
return False
|
return True
|
|
def AddIndexes(db, colName, expectIndexes):
|
'''Ìí¼ÓË÷Òý'''
|
col = db[colName]
|
'''¸ø±íÌí¼ÓË÷Òý'''
|
curIndexesDict = col.index_information()
|
for k, v in expectIndexes.items():
|
if k not in curIndexesDict:
|
expectIndexInfo = expectIndexes[k]
|
isUnique = expectIndexInfo.get('unique', False)
|
try:
|
col.create_index(expectIndexInfo['key'], name = k, unique = isUnique)
|
except Exception, e:
|
return False
|
return True
|
|
def CheckAndUpdateIndexOnDb(db):
|
'''¶ÔÓÚË÷ÒýÎļþÖеÄÿ¸ö±í£¬Êý¾Ý¿âÖиñíÓеÄË÷ÒýÐÅÏ¢£¬Ë÷ÒýÎļþÖбØÐëÓв¢ÇÒÆ¥ÅäuniqueºÍ(key, direction)list'''
|
indexInfoDict = ReadIndexInfo()
|
for k, v in indexInfoDict.items():
|
colName = k
|
colIndexDict = v
|
if not CheckIndexes(db, colName, colIndexDict):
|
return False
|
if not AddIndexes(db, colName, colIndexDict):
|
return False
|
return True
|
|
