Как правильно открывать и читать файлы? - Python - Ответ 5347208

#! /usr/bin/python3
 
# gameservers.py
#       -action     [install|reinstall|start|stop|restart|delete|updatepassword|sysload]
#       -sid
#       -game
#       -ip
#       -port
#       -slots
#       -password
 
# Ports:
# Port #1 = port
# Port #2 = port + 1
# Port #3 = port + 1000
 
import sys
import os
import re
import json
import crypt
import hashlib
import datetime
import argparse
 
# Parse arguments
argParser = argparse.ArgumentParser(description='LitePanel server-side.')
argParser.add_argument("-action", type=str,
                    choices=['install', 'reinstall', 'start', 'stop', 'restart', 'delete', 'updatepassword', 'sysload'],
                    help="Action", required=True)
argParser.add_argument("-sid", type=int,
                    help="Server ID", required=True)
argParser.add_argument("-game", type=str,
                    choices=[os.path.splitext(os.path.basename(f))[0] for f in os.listdir("/home/cp/gameservers/configs/")],
                    help="Game code", required=True)
argParser.add_argument("-ip", type=str,
                    help="IP", required=True)
argParser.add_argument("-port", type=int,
                    help="Port", required=True)
argParser.add_argument("-slots", type=int,
                    help="Slots ammount", required=True)
argParser.add_argument("-password", type=str,
                    help="Password", required=True)
args = argParser.parse_args()
 
action      =   str(args.action)
serverid    =   int(args.sid)
game        =   str(args.game)
ip          =   str(args.ip)
port        =   int(args.port)
slots       =   int(args.slots)
password    =   str(args.password)
 
salt = 'tlas'
username = 'gs' + str(serverid)
 
gameConfig = None
 
def md5file(filePath):
    f = open(filePath, 'rb')
    md5 = hashlib.md5()
    while True:
        data = f.read(8192)
        if not data:
            break
        md5.update(data)
    f.close()
    return md5.hexdigest()
 
def loadGameConfig():
    global gameConfig
    f = open('/home/cp/gameservers/configs/' + game + '.cfg', 'r')
    data = f.read()
    gameConfig = json.loads(data)
    f.close()
 
def serverStatus():
    p = os.popen('su -lc "screen -ls | grep -c gameserver" ' + username)
    count = int(p.readline())
    p.close()
    if count > 0:
        return True
    else:
        return False
 
def serverCheckFiles():
    for file in gameConfig['Files']:
        if not os.path.isfile('/home/' + username + file['File']):
            if file['Required'] == 1:
                return False
            else:
                continue
        
        fileHash = md5file('/home/' + username + file['File'])
        if not fileHash in file['Hashes']:
            return False
    return True
 
def serverConfigure():
    configs = gameConfig['Configs']
    for config in configs:
        # Check config file
        if not os.path.isfile('/home/' + username + config['File']):
            if config['Required'] == 1:
                return False
            else:
                continue
        
        # Read config
        f = open('/home/' + username + config['File'], 'r', encoding='utf-8')
        data = f.read()
        f.close()
        
        # Append exec configs
        if(config['ExecPattern']):
            execPattern = config['ExecPattern'].replace("<value>", "([^\s]*)");
            execConfigs = re.findall(execPattern, data)
            for execConfig in execConfigs:
                configPath = os.path.dirname(config['File']) + '/' + execConfig
                configs.append({
                    "File": configPath,
                    "Required": 0,
                    "ExecPattern": config['ExecPattern'],
                    "Values": [dict(i, Required = 0) for i in config['Values']]
                })
        
        # Check configs values
        for value in config['Values']:
            pattern = value['Pattern'].replace('<value>', '(.*)')
            
            # IP value
            if value['Value'] == '__ip__':
                replace = value['Pattern'].replace('<value>', ip)
            # Port value
            elif value['Value'] == '__port__':
                replace = value['Pattern'].replace('<value>', str(port))
            # Port 2 value
            elif value['Value'] == '__port2__':
                replace = value['Pattern'].replace('<value>', str(port + 1))
            # Port 3 value
            elif value['Value'] == '__port3__':
                replace = value['Pattern'].replace('<value>', str(port + 1000))
            # Slots value
            elif value['Value'] == '__slots__':
                replace = value['Pattern'].replace('<value>', str(slots))
            # Other values
            else:
                replace = value['Pattern'].replace('<value>', value['Value'])
            
            data = re.sub(pattern, replace, data)
            
            # Required, but not found
            if value['Required'] == 1 and not re.search(pattern, data):
                return False
            # Not required, but found
            elif value['Required'] == -1 and re.search(pattern, data):
                return False
        
        # Rewrite config
        f = open('/home/' + username + config['File'], 'w', encoding='utf-8')
        f.write(data)
        f.close()
    return True
 
def serverInstall():
    os.system('useradd -m -g gameservers -p ' + crypt.crypt(password, salt) + ' ' + username)
    for archive in gameConfig['Archives']:
        os.system('tar -xf /home/cp/gameservers/files/' + archive + '.tar -C /home/' + username + '/')
    os.system('chown ' + username + ' -Rf /home/' + username)
    os.system('chmod 700 /home/' + username)
    return True
 
def serverReinstall():
    os.system('rm -Rf /home/' + username + '/*')
    for archive in gameConfig['Archives']:
        os.system('tar -xf /home/cp/gameservers/files/' + archive + '.tar -C /home/' + username + '/')
    os.system('chown ' + username + ' -Rf /home/' + username)
    return True
 
def serverStart():
    execCmd = gameConfig['ExecCmd']
    execCmd = execCmd.replace('@ip@', ip)
    execCmd = execCmd.replace('@port@', str(port))
    execCmd = execCmd.replace('@port2@', str(port+1))
    execCmd = execCmd.replace('@port3@', str(port+1000))
    execCmd = execCmd.replace('@slots@', str(slots))
    os.system('su -lc "screen -AmdS gameserver ' + execCmd + '" ' + username)
    return True
 
def serverStop():
    os.system('su -lc "screen -r gameserver -X quit " ' + username)
    return True
 
def serverUpdatePassword():
    os.system('usermod -p ' + crypt.crypt(password, salt) + ' ' + username)
    return True
 
def serverDelete():
    os.system('userdel -rf ' + username)
    return True
 
def serverSysLoad():
    cpu = 0.0
    ram = 0.0
    
    p = os.popen("ps u -U gs" + str(serverid) + " | awk '{print $3\"\t\"$4}'")
    line = p.readline()
    
    while line:
        result = re.match("([0-9]+\.[0-9]+)\t([0-9]+\.[0-9]+)", line)
        if(result):
            cpu += float(result.group(1))
            ram += float(result.group(2))
        line = p.readline()
    
    p.close()
    print('[[' + str(cpu) + '::' + str(ram) + ']]')
 
def returnResult(status, description = ''):
    date = datetime.datetime.today()
    filename = date.strftime('%d-%m-%y') + '.xls'
    if os.path.isfile('/home/cp/logs/' + filename):
        # Open file
        f = open('/home/cp/logs/' + filename, 'a')
    else:
        # Create file
        f = open('/home/cp/logs/' + filename, 'w')
        f.write('TIME,SERVERID,STATUS,DESCRIPTION\n')
    
    f.write(date.strftime('%H:%M:%S') + ',' + str(serverid) + ',' + status + ',' + description + '\n')
    f.close()
    
    # Print result and exit
    print('[[' + status + '::' + description + ']]')
    exit(0)
 
loadGameConfig()
 
if action == 'install':
    if not serverInstall():
        returnResult('ERROR', 'InstallError')
    elif not serverConfigure():
        returnResult('ERROR', 'ConfigError')
    else:
        returnResult('OK')
 
if action == 'reinstall':
    if serverStatus():
        if not serverStop():
            returnResult('ERROR', 'StopError')
    
    if not serverReinstall():
        returnResult('ERROR', 'ReinstallError')
    elif not serverConfigure():
        returnResult('ERROR', 'ConfigError')
    else:
        returnResult('OK')
 
if action == 'start':
    if serverStatus():
        if not serverStop():
            returnResult('ERROR', 'StopError')
    
    if not serverCheckFiles():
        returnResult('ERROR', 'FilesError')
    elif not serverConfigure():
        returnResult('ERROR', 'ConfigError')
    elif not serverStart():
        returnResult('ERROR', 'StartError')
    else:
        returnResult('OK')
 
if action == 'stop':
    if not serverStop():
        returnResult('ERROR', 'StopError')
    else:
        returnResult('OK')
 
if action == 'restart':
    if not serverStop():
        returnResult('ERROR', 'StopError')
    elif not serverCheckFiles():
        returnResult('ERROR', 'FilesError')
    elif not serverConfigure():
        returnResult('ERROR', 'ConfigError')
    elif not serverStart():
        returnResult('ERROR', 'StartError')
    else:
        returnResult('OK')
 
if action == 'delete':
    if serverStatus():
        serverStop()
    
    if not serverDelete():
        returnResult('ERROR', 'DeleteError')
    else:
        returnResult('OK')
 
if action == 'updatepassword':
    if serverStatus():
        if not serverStop():
            returnResult('ERROR', 'StopError')
    
    if not serverUpdatePassword():
        returnResult('ERROR', 'UpdatePasswordError')
    else:
        returnResult('OK')
 
 
if action == 'sysload':
    serverSysLoad()