fenrir/src/fenrir-package/core/punctuationManager.py
2016-09-22 16:53:31 +02:00

58 lines
1.9 KiB
Python

#!/bin/python
# -*- coding: utf-8 -*-
# Fenrir TTY screen reader
# By Chrys, Storm Dragon, and contributers.
from core import debug
class punctuationManager():
def __init__(self):
pass
def initialize(self, environment):
self.env = environment
self.punctuation = {
'currLevel':'1',
'levels':{
'1':',',
'2':'.',
'3':'.,',
},
'punctuationDict':{
'.':'punkt',
',':'komma'
},
'customDict':{
'chrys':'awsome',
'cool':'mr chrys'
}
}
def shutdown(self):
pass
def removeUnused(self, text):
return text.translate(text.maketrans(string.punctuation, ' '*len(string.punctuation)))
def useCustomDict(self, text, customDict):
resultText = str(text)
if customDict:
for key,item in customDict.items():
resultText = resultText.replace(str(key),str(item))
return resultText
def usePunctuationDict(self, text, punctuationDict, punctuation):
resultText = str(text)
if punctuationDict and punctuation != '':
for key,item in punctuationDict.items():
if key in punctuation:
resultText = resultText.replace(str(key),' ' +str(item) +' ')
return resultText
def proceedPunctuation(self, text, ignoreLevel=False):
resultText = self.useCustomDict(text, self.punctuation['customDict'])
currPunctLevel = ''
if not ignoreLevel:
currPunctLevel = self.punctuation['levels'][self.punctuation['currLevel']]
else:
currPunctLevel = string.punctuation
resultText = self.usePunctuationDict(resultText, self.punctuation['punctuationDict'], currPunctLevel)
resultText = self.removeUnused(resultText)
return resultText