Python开发游戏开服脚本

#!/usr/bin/env python
#coding:utf-8

from pymongo import MongoClient
import datetime
import random
import shutil
import time
import sys
import re
import os

class readyTostart(object):

    def Countdown(self):
        print('''
        ######################################
        #        拜见包大人一健开服脚本      #
        ######################################
        ''')
        print('                   \033[1;31;40m脚本即将开始执行...\033[0m')
        time.sleep(2)
        print
        for i in range(3,0,-1):
            time.sleep(1)
            print('              脚本还有\033[5;31;40m%s\033[0m秒开始执行,请稍等...' % i)
            print

    def ParametersCheck(self):
        global openNum,openYear,openMonth,openDay
        try:
            openNum = sys.argv[1]
            if not re.match('^[0-9]+$', openNum):
                if not re.match('(^[0-9]+)-([0-9]+$)', openNum):
                    print('Please output in the specified format. Example num1-num2,Does not support decimals')
                    sys.exit(1)
                else:
                    parmet1 = re.match('(^[0-9]+)-([0-9]+$)', openNum)
                    if int(parmet1.group(1)) > int(parmet1.group(2)):
                        print('The first opening number %s cannot be greater than the second opening number %s' % (parmet1.group(1),parmet1.group(2)))
                        sys.exit(1)
            if re.match('^[0]', openNum):
                print('You cannot enter numbers starting with 0')
                sys.exit(1)
        except Exception as e:
            print e
            values = sys.argv[0]
            print('Please enter at least one parameter')
            Exaples = 'Exaples: python {key} 10 | python {key} 1-10'
            print(Exaples.format(key=values))
            sys.exit(1)
        nowDate = datetime.datetime.now() + datetime.timedelta(days=1)
        openYear = nowDate.strftime("%Y")
        try:
            cusDate = sys.argv[2]
        except Exception as e:
            cusDate = nowDate
            openMonth = cusDate.strftime("%m")
            openDay = cusDate.strftime("%d")
            if re.match('0\d', openMonth): openMonth = openMonth[1:]
            if re.match('0\d', openDay): openDay = openDay[1:]
        else:
            if not re.match('^[0-9]+\.[0-9]+',cusDate):
                print 'The second parameter format is => month.day'
                sys.exit(1)
            openMonth = cusDate.split('.')[0]
            openDay = cusDate.split('.')[1]
            if int(openMonth) < 1 or int(openMonth) > 12:
                print 'The month you entered cannot be greater than twelve and less than one'
                sys.exit(1)
            if int(openDay) < 1 or int(openDay) > 31:
                print("You can't enter more than 31 or less than 1")
                sys.exit(1)
        openMonth = '=' + openMonth
        openDay = '=' + openDay

class deployService(object):
    def __init__(self):
        self.toolsdirctory = '/data/tools/new_server/'
        self.servicedir = '/data/bz-tw-and/s'
        self.codedir = '/data/bz-tw-and'
        self.logdir = '/data/logs/s'
        self.userInputSpecified = []
        self.userInputall = []
        self.alreadyExist = []

    def startDeploy(self):
        global inputNum,newService,scriptList
        if not os.path.exists(self.codedir):
            print('%s not exists,script exit' % self.codedir)
            sys.exit(0)
        if not os.path.exists(self.toolsdirctory + 'Config.lua') or not os.path.exists(self.toolsdirctory + 'bin'):
            print('(%s and %s)file or dirctory not exists,script exit' % (self.toolsdirctory + 'Config.lua', self.toolsdirctory + 'bin'))
            sys.exit(1)
        inputNum = re.match('(\d+)-(\d+)', openNum)
        inputOnenum = re.match('(^[1-9][0-9]*$)', openNum)
        if inputOnenum:
            num = 1
            for i in os.listdir(self.codedir):
                nowService = re.match('(s)(\d+)',i)
                if nowService:
                    num += 1
                    nowNum = i[1:]
                    if openNum == nowNum:
                        print('       %s Service already existsService already exists' % openNum)
                        sys.exit()
            newNum = str(num) + '-' + openNum
            inputNum = re.match('(\d+)-(\d+)', newNum)
        if  inputNum:
            oneNum = int(inputNum.group(1))
            towNum = int(inputNum.group(2))
            for fuNum in range(1,towNum+1):
                self.userInputall.append(fuNum)
            for fuNum in range(oneNum,towNum+1):
               self.userInputSpecified.append(fuNum)
            for i in os.listdir(self.codedir):
               nowService = re.match('(s)(\d+)',i)
               if nowService:
                   nowNum = i[1:]
                   if int(nowNum) < towNum:
                       self.alreadyExist.append(int(nowNum))
            differenceSet=(set(self.userInputall)^set(self.alreadyExist))
            intersection=(set(self.userInputSpecified)&set(self.alreadyExist))
            newService=list(differenceSet)
            oldService=list(intersection)
            oldService=list(intersection)
            if newService != []:
                for i in newService:
                    gameService = self.servicedir + str(i)
                    os.mkdir(gameService)
                    shutil.copytree(self.toolsdirctory + 'bin', gameService + '/bin')
                    shutil.copytree(self.servicedir + str(1) + '/scripts', self.servicedir + str(i) + '/scripts')
                    shutil.copyfile(self.toolsdirctory + 'Config.lua',gameService +'/Config.lua')
                    if not os.path.exists(self.logdir +str(i)):
                        os.mkdir(self.logdir +str(i))
                    os.symlink(self.logdir+str(i),self.servicedir + str(i) + '/bin/logs')
                    if not os.path.exists(self.toolsdirctory + '/bin/bz-tw-and-s_'):
                        print('%s not exists,Please Check' % (self.toolsdirctory + '/bin/bz-tw-and-s_'))
                    try:
                        shutil.move(self.servicedir + str(i) + '/bin/bz-tw-and-s_',self.servicedir + str(i) + '/bin/bz-tw-and-s' + str(i))
                    except Exception as e:
                        print e
                        print('%s服修改bz-tw-and-s_命名失败' % i)
                    scriptList = ['hotup.sh','hotup_client.sh','start.sh','stop.sh']
                    for scriptName in scriptList:
                        if not os.path.exists(self.servicedir+str(i) + '/bin/' + scriptName):
                            print('%s not exists,Please Check' % scriptName)
                    hotupLine = ''
                    hotupPort = 10000 + int(i)
                    for scriptName in ['hotup.sh','hotup_client.sh']:
                        with open(self.servicedir + str(i) + '/bin/' + scriptName, 'r')as f:
                            for line in f.readlines():
                                if line.find('10000'):
                                    line = re.sub('10000', str(hotupPort), line)
                                    hotupLine += line
                                else:
                                    hotupLine += line
                        with open(self.servicedir + str(i) + '/bin/' + scriptName, 'w')as f:
                            f.write(hotupLine)
                            hotupLine = ''
                    start_stop_Line = ''
                    for scriptName in ['start.sh','stop.sh']:
                        with open(self.servicedir + str(i) + '/bin/' + scriptName, 'r')as f:
                            for line in f.readlines():
                                if line.find('1'):
                                    line = re.sub('1', str(i), line)
                                    start_stop_Line += line
                                else:
                                    start_stop_Line += line
                        with open(self.servicedir + str(i) + '/bin/' + scriptName, 'w')as f:
                            f.write(start_stop_Line)
                            start_stop_Line = ''
            if oldService != []:
                for i in oldService:print('                 already kai fu %s' % i)
        else:
            print('Exaples: python %s 10 | python %s 1-10' % (sys.argv[0],sys.argv[0]))

class MongoDabase(object):
    def __init__(self,serverIp='192.168.6.184',databasePort=27017,user='root',password='123456',adminDatabase='admin'):
        self.client = MongoClient(serverIp,databasePort)
        self.db = self.client[adminDatabase]
        self.db.authenticate(user,password)

    def createPassword(self):
        password = ''
        for i in range(8):
            current = random.randrange(0,8)
            if current > i:
                temp = random.randint(0,9)
            elif current < i:
                temp = chr(random.randint(65,90))
            else:
                temp = chr(random.randint(97,122))
            password += str(temp)
        return password

    def createUser(self):
        global passwordList
        passwordList = {}
        deploy = deployService()
        configfileLine = ''
        passwordPath = '/data/script/'
        if not os.path.exists(passwordPath):
            os.mkdir(passwordPath)
        print('                    \033[1;31;40m部署情况如下\033[0m')
        print
        if newService != []:
            for i in newService:
                clientPort = 20000 + int(i)
                httpPort = 10000 + int(i)
                dbName = 's' + str(i)
                SvrId = str(i)
                gameServiceName = 'bz-tw-and-s' + str(i)
                password = self.createPassword()
                passwordList.update({i:password})
                dbUser = self.client[gameServiceName]
                try:
                    dbUser.command("createUser", "bz", pwd=password, roles=["readWrite"])
                except Exception as e:
                    print('''         %s

                请测试账户密码验证是否正确
                    '''% e)
                old = ['20003','10003','s3','zxjRYZGI','sid','2018','=5','=19']
                new = [clientPort,httpPort,dbName,password,SvrId,openYear,openMonth,openDay]
                for old,new  in zip(old,new):
                    with open(deploy.servicedir + str(i) + '/Config.lua','r')as f:
                        for line in f.readlines():
                             if line.find(str(old)):
                                 line = re.sub(str(old),str(new),line)
                                 configfileLine += line
                             else:
                                 configfileLine += line
                    with open(deploy.servicedir + str(i) + '/Config.lua', 'w')as f:
                        f.write(configfileLine)
                        configfileLine = ''
                with open(passwordPath+'mongo_pass.txt','a+')as f:
                    mongoDbname = 'bz-tw-and-' + str(i)
                    f.write(mongoDbname+'='+password+'\n')
                os.remove(deploy.servicedir + str(i)+'/scripts/Config.lua')
                shutil.copyfile(deploy.servicedir + str(i) + '/Config.lua',deploy.servicedir + str(i)+'/scripts/Config.lua')
                print('                 \033[31m%s\033[0m 服已经部署完成...' % i)
                print
        print('             \033[1;31;40m数据库密码文件存放位置如下\033[0m')
        print
        print('            /data/script/mongo_pass.txt')

class configCheck(object):
    def luaConfigcheck(self):
        deploy = deployService()
        Month = openMonth[1:]
        Day = openDay[1:]
        data = ['(GAME_IO_LISTEN_PORT = )(\d+)',
                '(GAME_HTTP_LISTEN_PORT=)(\d+)',
                '(DBNAME="bz-tw-and-)(.*)(")',
                '(DBPWD=")(.*)(")',
                '(SvrId = )(.*)',
                '(activityStartDate={year=)(\d{4})(,month=)(\d+)(,day=)(\d+)(.*)']
        if newService != []:
            for i in newService:
                print ('''
       ############################################
       #             %s服配置文件检查             #
       ############################################
                            ''' % i)
                with open(deploy.servicedir + str(i) + '/scripts/Config.lua', 'r')as f:
                    for line in f.readlines():
                        for d in data:
                            if re.match(d, line):
                                m = re.match(d, line)
                                try:
                                    if int(m.group(2)) == 10000 + int(i) or int(m.group(2)) == 20000 + int(i) or int(m.group(2)) == int(i):
                                        if int(m.group(2)) == int(i):
                                            result = '                  [\033[32mID正确\033[0m]'
                                        else:
                                            result = '[\033[32m端口正确\033[0m]'
                                    else:
                                        result = ' [\033[5;31m配置错误\033[0m]'
                                        if int(m.group(2)) == int(openYear) and int(m.group(4)) == int(Month) and int(m.group(6)) == int(Day):
                                            yresult = '[\033[32m年月日正确\033[0m]'
                                        else:
                                            yresult = ' [\033[5;31m年或月或日错误\033[0m]'
                                except Exception as e:
                                    if str(m.group(2)) == str('s'+str(i)) or str(m.group(2)) == str(passwordList[i]):
                                        if str(m.group(2)) == str(passwordList[i]):
                                            result = '      [\033[32m密码正确\033[0m]'
                                        else:
                                            result = ' [\033[32m名字正确\033[0m]'
                                    else:
                                        result = '      [\033[5;31m配置错误\033[0m]'
                                try:
                                    time.sleep(0.5)
                                    print('         %s\033[31m%s\033[0m%s\033[31m%s\033[0m%s\033[31m%s\033[0m%s%s' % (m.group(1), m.group(2), m.group(3), m.group(4), m.group(5), m.group(6),m.group(7), yresult))
                                except Exception as e:
                                    try:
                                        time.sleep(0.5)
                                        print('         %s\033[31m%s\033[0m%s        %s' % (m.group(1), m.group(2), m.group(3), result))
                                    except Exception as e:
                                        time.sleep(0.5)
                                        print('         %s\033[31m%s\033[0m   %s' % (m.group(1), m.group(2), result))

    def scriptCheck(self):
        deploy = deployService()
        hotup = '(.*:)(\d{5})(/.*)'
        startstop = '(.*)(s)(\d+)(")'
        if newService != []:
            for i in newService:
                print ('''
        ############### %s服脚本检查 ###############
                             ''' % i)
                for script_name in scriptList:
                    with open( deploy.servicedir + str(i) + '/bin/'+ script_name, 'r')as f:
                        for line in f.readlines():
                            if re.match(hotup,line):
                                m = re.match(hotup,line)
                                if int(m.group(2)) == (10000 + int(i)):
                                    time.sleep(0.5)
                                    result='端口正确'
                                    print('         %s\033[31m%s\033[0m%s [\033[32m%s\033[0m]' % (m.group(1),m.group(2),m.group(3),result))
                                else:
                                    time.sleep(0.5)
                                    result = '端口错误'
                                    print('          %s\033[31m%s\033[0m%s [\033[5;31m%s\033[0m]' % (m.group(1), m.group(2), m.group(3), result))
                            if re.match(startstop,line):
                                m = re.match(startstop, line)
                                if int(m.group(3)) == (int(i)):
                                    time.sleep(0.5)
                                    result = '开关脚本服数正确'
                                    print('         %s%s\033[31m%s\033[0m%s [\033[32m%s\033[0m]' % (m.group(1), m.group(2), m.group(3), m.group(4), result))
                                else:
                                    time.sleep(0.5)
                                    result = '开关脚本服数正确'
                                    print('         %s\033[31m%s\033[0m%s [\033[5;31m%s\033[0m]' % (m.group(1), m.group(2), m.group(3), m.group(4), result))
        for i in newService:
            for file_name in os.listdir(deploy.servicedir + str(i) + '/bin/'):
                if re.match('(.*s)(\d+)',file_name):
                    m = re.match('(.*s)(\d+)',file_name)
                    if m.group(2) == str(i):
                        result = '服数正确'
                        time.sleep(1)
                        print
                        print("        ########## %s服二进制文件命名检查 ##########" % i)
                        print("         %s\033[31m%s\033[0m [\033[32m%s\033[0m]" % (m.group(1),m.group(2),result))
                    else:
                        time.sleep(1)
                        result = '服数错误'
                        print("         %s\033[31m%s\033[0m [\033[5;31m%s\033[0m]" % (m.group(1), m.group(2), result))

        for i in newService:
            print ('''
        ############### %s软链接人工检查 ###############''' % i)
            if os.path.exists(deploy.servicedir + str(i) + '/bin/logs') and os.path.exists(deploy.logdir+str(i)):
                print('         \033[32m软链接已存在\033[0m')
                os.system('ls -l %s'% (deploy.servicedir + str(i) + '/bin/logs'))
            else:
                print('         \033[31m软链接目录不存在，请检查\033[0m')

if __name__ == '__main__':
    start = readyTostart()
    start.ParametersCheck()
    start.Countdown()
    deploy = deployService()
    deploy.startDeploy()
    db = MongoDabase()
    db.createUser()
    check = configCheck()
    check.luaConfigcheck()
    check.scriptCheck()