MyStem Simple Wrapper Functions

mystem.py

import subprocess as sp
import json

def get_lemma(x):
    lemma = x['analysis'][0].get('lex') if x['analysis'] else x.get('text')
    return lemma

def mylem(x):
    p = sp.Popen(['mystem', '-l', '--format', 'json'], stdin=sp.PIPE, stdout=sp.PIPE, stderr=sp.STDOUT)
    stdout = p.communicate(input=bytes(x, encoding='utf-8'))
    stdout = json.loads(stdout[0].decode('utf-8'))
    return list(filter(None, map(get_lemma, stdout)))