Upload New File

stbi/Lib/site-packages/Sastrawi/Stemmer/Context/Visitor/RemoveDerivationalSuffix.py

+import re
+from Sastrawi.Stemmer.Context.Removal import Removal
+
+class RemoveDerivationalSuffix(object):
+    """Remove Derivational Suffix.
+    Asian J. (2007) "Effective Techniques for Indonesian Text Retrieval". page 61
+
+    @link http://researchbank.rmit.edu.au/eserv/rmit:6312/Asian.pdf
+    """
+
+    def visit(self, context):
+        result = self.remove(context.current_word)
+        if result != context.current_word:
+            removedPart = re.sub(result, '', context.current_word, 1)
+
+            removal = Removal(self, context.current_word, result, removedPart, 'DS')
+
+            context.add_removal(removal)
+            context.current_word = result
+
+    def remove(self, word):
+        """Remove derivational suffix
+        Original rule : i|kan|an
+        Added the adopted foreign suffix rule : is|isme|isasi
+        """
+        return re.sub(r'(is|isme|isasi|i|kan|an)$', '', word, 1)
+
+