mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-25 04:32:33 -06:00
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
|
from pandac.PandaModules import *
|
||
|
|
from otp.chat.SequenceList import SequenceList
|
||
|
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
|
||
|
|
class TTSequenceList(SequenceList):
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.notify = DirectNotifyGlobal.directNotify.newCategory('TTSequenceList')
|
||
|
|
sequenceListURL = config.GetString('blacklist-sequence-url', '')
|
||
|
|
if sequenceListURL == '':
|
||
|
|
self.notify.warning('No Sequence BL URL specified! Continuing with local sequence.')
|
||
|
|
SequenceList.__init__(self, self.loadSquencesLocally())
|
||
|
|
else:
|
||
|
|
SequenceList.__init__(self, self.downloadSequences(sequenceListURL))
|
||
|
|
|
||
|
|
def downloadSequences(self, url):
|
||
|
|
fs = Ramfile()
|
||
|
|
http = HTTPClient.getGlobalPtr()
|
||
|
|
self.ch = http.makeChannel(True)
|
||
|
|
self.ch.getHeader(DocumentSpec(url))
|
||
|
|
doc = self.ch.getDocumentSpec()
|
||
|
|
self.ch.getDocument(doc)
|
||
|
|
self.ch.downloadToRam(fs)
|
||
|
|
return fs.getData().split('\r\n')
|
||
|
|
|
||
|
|
def loadSquencesLocally(self):
|
||
|
|
vfs = VirtualFileSystem.getGlobalPtr()
|
||
|
|
filename = Filename('tsequence.dat')
|
||
|
|
searchPath = DSearchPath()
|
||
|
|
searchPath.appendDirectory(Filename('/server'))
|
||
|
|
found = vfs.resolveFilename(filename, searchPath)
|
||
|
|
if not found:
|
||
|
|
self.notify.warning("Couldn't find blacklist sequence data file!")
|
||
|
|
return
|
||
|
|
data = vfs.readFile(filename, True)
|
||
|
|
return data.split('\n')
|