From 0cb7ea4410be9c64de6226238838c79662c5ebe3 Mon Sep 17 00:00:00 2001
From: chrys87 <chrys87@web.de>
Date: Thu, 22 Sep 2016 16:06:59 +0200
Subject: [PATCH] Update punctuationManager.py

---
 src/fenrir-package/core/punctuationManager.py | 41 +++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/src/fenrir-package/core/punctuationManager.py b/src/fenrir-package/core/punctuationManager.py
index bbfe65a2..2bd78be8 100644
--- a/src/fenrir-package/core/punctuationManager.py
+++ b/src/fenrir-package/core/punctuationManager.py
@@ -11,5 +11,46 @@ class punctuationManager():
         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):
+        resultText = text.translate(text.maketrans(string.punctuation, ' '*len(string.punctuation)))   
+        return resultText
+    def useCustomDict(self, text, customDict):
+        resultText = text
+        for key,item in customDict.items():
+            resultText = resultText.replace(str(key),str(item))
+        return resultText
+    def usePunctuationDict(self, text, punctuationDict, punctuation):
+        resultText = str(text)
+        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