storm/fenrir
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update punctuationManager.py

Browse Source
This commit is contained in:
chrys87 2016-09-22 16:50:26 +02:00 committed by GitHub
parent 1f263b2cf0
commit 217ec6a2c0
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/fenrir-package/core/punctuationManager.py
View File

@ -30,15 +30,16 @@ class punctuationManager():
def shutdown(self): def shutdown(self):
pass pass
def removeUnused(self, text): def removeUnused(self, text):
resultText = text.translate(text.maketrans(string.punctuation, ' '*len(string.punctuation))) return text.translate(text.maketrans(string.punctuation, ' '*len(string.punctuation)))
return resultText
def useCustomDict(self, text, customDict): def useCustomDict(self, text, customDict):
resultText = text resultText = text
if customDict:
for key,item in customDict.items(): for key,item in customDict.items():
resultText = resultText.replace(str(key),str(item)) resultText = resultText.replace(str(key),str(item))
return resultText return resultText
def usePunctuationDict(self, text, punctuationDict, punctuation): def usePunctuationDict(self, text, punctuationDict, punctuation):
resultText = str(text) resultText = str(text)
if punctuationDict and punctuation != '':
for key,item in punctuationDict.items(): for key,item in punctuationDict.items():
if key in punctuation: if key in punctuation:
resultText = resultText.replace(str(key),' ' +str(item) +' ') resultText = resultText.replace(str(key),' ' +str(item) +' ')