comment_converter.py

import re
from collections import defaultdict

regex_n_people = re.compile("[0-9]인")


def word_converter(words, nick_to_human):
    USAGE = ("점심", "저녁", "식사")
    result = {"usage": None, "kor_names": ""} #, "cell_counter": defaultdict(int)}
    try: 
        for word in words:
            if (not word or word == " ") or (regex_n_people.match(word)):
                continue

            if any(usage in word for usage in USAGE) and result["usage"] is None:
                result["usage"] = word
                continue

            if word not in nick_to_human:
                return {"usage": None, "kor_names": None} #, "cell_counter": None}

            if nick_to_human[word]:
                if result["kor_names"]:
                    result["kor_names"] += ","
                result["kor_names"] += nick_to_human[word].kor_name
                # result["cell_counter"][nick_to_human[word].cell_name] += 1
    except:
        return {"usage": None, "kor_names": ""}
    return result


def comment_converter(comment, nick_to_human):
    words = (
        comment.replace("/", " ")
        .replace(",", " ")
        .replace("(", " ")
        .replace(")", " ")
        .replace("\n", " ")
        .split(" ")
    )
    return word_converter(words, nick_to_human)