Pymorphy2 lemmatizer class.

lemmatized.py

import pymorphy2


class Lemmatizer:
    """
    Pymorphy2 lemmatizer class.
    """
    
    def __init__(self):
        """
        Init lemmatizer.
        """
        self.morph = pymorphy2.MorphAnalyzer()
        
    def __call__(self, word: str) -> str:
        """
        Lemmatize word.
        """
        lemma = self.morph.parse(word)[0].normal_form
        return lemma


class LemmatizerCached:
    """
    Pymorphy2 lemmatizer class with caching.
    """
    
    def __init__(self):
        """
        Init lemmatizer.
        """
        self.cache = {}
        self.morph = pymorphy2.MorphAnalyzer()
        
    def __call__(self, word: str) -> str:
        """
        Lemmatize word.
        """
        if word in self.cache:
            lemma = self.cache[word]
        else:
            lemma = self.morph.parse(word)[0].normal_form
            self.cache[word] = lemma
        return lemma
    
    
lemmatizer = Lemmatizer()
lemmatizer_cached = LemmatizerCached()