DANIEL: Whitelist tool shit

This commit is contained in:
John 2015-06-08 19:54:21 +03:00 committed by Loudrob
parent 10a451c883
commit 2fb28f6f09

View file

@ -1,8 +1,7 @@
import os import os
os.chdir('../../../') os.chdir('../../../')
from toontown.chat import WhiteListData from otp.chat import WhiteListData
def acceptWord(): def acceptWord():
word = raw_input('> ').rstrip().lower() word = raw_input('> ').rstrip().lower()
@ -18,6 +17,19 @@ def acceptWord():
else: else:
LOCAL_LIST.remove(word) LOCAL_LIST.remove(word)
print 'Removed "%s" from the whitelist.' % word print 'Removed "%s" from the whitelist.' % word
elif word.startswith('m '):
merge = word.replace('m ', '')
if os.path.isfile(merge):
print 'Opening %s...' % merge
with open(merge) as file:
for line in file.readlines():
line = line.replace('\r', '').replace('\n', '').lower()
print 'Adding %s...' % line
LOCAL_LIST.append(line)
else:
print 'No file named %s!' % merge
elif word in LOCAL_LIST: elif word in LOCAL_LIST:
print 'The word "%s" is already whitelisted.' % word print 'The word "%s" is already whitelisted.' % word
else: else:
@ -26,14 +38,15 @@ def acceptWord():
acceptWord() acceptWord()
def saveChanges(): def saveChanges():
global LOCAL_LIST
print 'Saving the whitelist...' print 'Saving the whitelist...'
with open('toontown/chat/WhiteListData.py', 'w') as f: with open('otp/chat/WhiteListData.py', 'w') as f:
f.write('WHITELIST = [\n') f.write('WHITELIST = [\n')
LOCAL_LIST.sort() LOCAL_LIST.sort()
LOCAL_LIST = list(set(LOCAL_LIST)) # Remove duplicates
addedWords = [] addedWords = []
for word in LOCAL_LIST: for word in LOCAL_LIST:
@ -57,6 +70,7 @@ LOCAL_LIST = WhiteListData.WHITELIST
print 'Welcome to the Toontown Stride Whitelist Tool!' print 'Welcome to the Toontown Stride Whitelist Tool!'
print 'Type any word you want to add to the whitelist.' print 'Type any word you want to add to the whitelist.'
print 'If you wish to remove a word, type "r <word>".' print 'If you wish to remove a word, type "r <word>".'
print 'If you wish to merge a file, type "m <file>".'
print 'When you are done and want to save your changes, type "exit()".' print 'When you are done and want to save your changes, type "exit()".'