#!/usr/bin/python
# -*- coding: GBK -*-
#-------------------------------------------------------------------------------
#
#-------------------------------------------------------------------------------
#

#-------------------------------------------------------------------------------
#¼ì²é²¢¸üÐÂË÷Òý
#-------------------------------------------------------------------------------

from CollectionDefine import *
from Common import(mylog,)
DBConfig = __import__('Config.DBConfig')
import os
from DBCommon import (GlobalFunctions, error)
import pymongo

def ReadIndexInfo(indexInfoFile):
    indexInfoDict = {}
    if not os.path.isfile(indexInfoFile):
        mylog.info('%s not found!'%indexInfoFile)
        return indexInfoDict
    mylog.info('Reading index file %s...'%indexInfoFile)
    try:
        fileObj = open(indexInfoFile)
        indexInfo = fileObj.read()
    except IOError, e:
#        print "open file %s failed!%s"%(indexInfoFile, e)
        msg = error.formatMsg('fatal', error.ERROR_NO_164, 'Open %s exception %s'%(indexInfoFile, e))
        mylog.fatal(msg)
        return indexInfoDict
    finally:
        fileObj.close()
    mylog.info('Parsing index file %s...'%indexInfoFile)
    try:
        indexInfoDict = eval(indexInfo)
    except:
#        print 'file %s content eval error!'%indexInfoFile
        msg = error.formatMsg('fatal', error.ERROR_NO_165, 'file %s content eval error %s'%(indexInfoFile, e))
        mylog.fatal(msg)
        return indexInfoDict

    mylog.info('Read index file %s ok!'%indexInfoFile)
    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]
#    print 'checking index info on %s...'%colName
    mylog.info('checking index info on %s...'%colName)
    '''¼ì²é±íÖÐÊÇ·ñÓжàÓàµÄË÷Òý'''
    indexDict = col.index_information()
    for k, v in indexDict.items():
        if k == '_id_':
            continue
        if k not in expectIndexInfo:
#            print 'unexpect index %s:%s found!'%(k,v)
            msg = error.formatMsg('fatal', error.ERROR_NO_166, 'unexpect index %s.%s found!'%(colName, k))
            mylog.fatal(msg)
            return False
        if not CompareIndex(v, expectIndexInfo[k]):
#            print '%s unmatched index found!'%colName
            msg = error.formatMsg('fatal', error.ERROR_NO_167, 'unmatched index %s.%s found! index in db:\r\n%s\r\n expect index:\r\n%s\r\n'%(colName, k, indexDict, expectIndexInfo))
            mylog.fatal(msg)
            return False
#    print 'check indx info on %s ok!'%colName
    mylog.info('check indx info on %s ok!'%colName)
    return True

def AddIndexes(db, colName, expectIndexes):
    '''Ìí¼ÓË÷Òý'''
    col = db[colName]
#    print 'adding index on %s'%colName
    mylog.info('adding index on %s'%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:
#                print 'create index %s failed!'%(k)
                msg = error.formatMsg('fatal', error.ERROR_NO_168, 'create index %s.%s failed!%s'%(colName, k, e))
                mylog.fatal(msg)
                return False
#            print 'create index %s ok!'%k
            mylog.info('create index %s ok!'%k)
#    print 'add index on %s ok!'%colName
    mylog.info('add index on %s ok!'%colName)
    return True     

def CheckAndUpdateIndexOnDb(db, indexInfoFile):
    '''¶ÔÓÚË÷ÒýÎļþÖеÄÿ¸ö±í£¬Êý¾Ý¿âÖиñíÓеÄË÷ÒýÐÅÏ¢£¬Ë÷ÒýÎļþÖбØÐëÓв¢ÇÒÆ¥ÅäuniqueºÍ(key, direction)list'''
    indexInfoDict = ReadIndexInfo(indexInfoFile)
    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

def CheckAndUpdateIndex():
    indexPath = os.path.join(GlobalFunctions.getAppPath(), 'index')
    
    usrDBCon = pymongo.Connection(DBConfig.USER_DB_IP, DBConfig.USER_DB_PORT, auto_start_request=False)
    if not GlobalFunctions.LoginMongoDB(DBConfig.userdb_user, DBConfig.userdb_pwd, connection = usrDBCon):
        msg = error.formatMsg('DeployError', error.ERROR_NO_13,
                              "LoginMongoDB failed:ip = %s, port = %s, user = %s,pwd = %s"%(DBConfig.USER_DB_IP,
                                                                                            DBConfig.USER_DB_PORT,
                                                                                            DBConfig.userdb_user,
                                                                                            DBConfig.userdb_pwd))
        mylog.DeployError(msg, True)
        return
    
    userDB = usrDBCon[DBConfig.USER_DB_NAME]
    userDBIndexFileName = os.path.join(indexPath, 'GameUser.txt')
    mylog.info('CheckAndUpdateIndexOnDb(%s, %s)...'%(DBConfig.USER_DB_NAME, userDBIndexFileName))
    if not CheckAndUpdateIndexOnDb(userDB, userDBIndexFileName):
        mylog.info('CheckAndUpdateIndexOnDb(%s, %s) failed!'%(DBConfig.USER_DB_NAME, userDBIndexFileName))
        return False
    mylog.info('CheckAndUpdateIndexOnDb(%s, %s) ok!'%(DBConfig.USER_DB_NAME, userDBIndexFileName))
    
    
    return True

def test():
    import pymongo
    
    con = pymongo.Connection()
    db = con.admin
    if not db.authenticate('sa', 'sa'):
        print 'auth failed!'
        os.system('pause')
        return
    
#    dbName = 'test'
#    con.drop_database(dbName)
#    db = con[dbName]
    
    if not CheckAndUpdateIndex():
        print '#info CheckAndUpdateIndex failed!'
    else:
        print '#info CheckAndUpdateIndex ok'
    os.system('pause')

if __name__ == "__main__":
    test()
    os.system('pause')