Poodletooth-iLand/toontown/launcher/TTSLauncher.py

from panda3d.core import *
from direct.directnotify import DirectNotifyGlobal
import os
import sys
import time

class LogAndOutput:
    def __init__(self, orig, log):
        self.orig = orig
        self.log = log

    def write(self, str):
        self.log.write(str)
        self.log.flush()
        self.orig.write(str)
        self.orig.flush()

    def flush(self):
        self.log.flush()
        self.orig.flush()

class TTSLauncher:
    notify = DirectNotifyGlobal.directNotify.newCategory('TTSLauncher')

    def __init__(self):
        self.http = HTTPClient()

        self.logPrefix = 'stride-'

        ltime = 1 and time.localtime()
        logSuffix = '%02d%02d%02d_%02d%02d%02d' % (ltime[0] - 2000,  ltime[1], ltime[2], ltime[3], ltime[4], ltime[5])

        if not os.path.exists('user/logs/'):
            os.mkdir('user/logs/')
            self.notify.info('Made new directory to save logs.')

        logfile = os.path.join('user/logs', self.logPrefix + logSuffix + '.log')

        log = open(logfile, 'a')
        logOut = LogAndOutput(sys.stdout, log)
        logErr = LogAndOutput(sys.stderr, log)
        sys.stdout = logOut
        sys.stderr = logErr

    def getPlayToken(self):
        return self.getValue('TTS_PLAYCOOKIE')

    def getGameServer(self):
        if config.GetString('game-type', 'developer') != 'remote':
            return self.getValue('TTS_GAMESERVER')
        return self.getServer(self.getValue('TTS_GAMESERVER'))

    def getServer(self, gameserver):
        import DNS
        DNS.DiscoverNameServers()
        srv_req = DNS.Request(qtype = 'srv')
        srv_result = srv_req.req('_toontown._tcp.'+ gameserver)

        for result in srv_result.answers:
            if result['typename'] == 'SRV':
                data = result['data']
                return data[3] +  ":"  +  str(data[2])

    def getValue(self, key, default = None):
        return os.environ.get(key, default)

    def setPandaErrorCode(self):
        pass

    def setDisconnectDetails(self, disconnectCode, disconnectMsg):
        self.disconnectCode = disconnectCode
        self.disconnectMsg = disconnectMsg

    def setDisconnectDetailsNormal(self):
        self.setDisconnectDetails(0, 'normal')