def get_sent_context()

in blink/candidate_retrieval/utils.py [0:0]

• 33 lines of code
• 9 McCabe index (conditional complexity)

def get_sent_context(mention, key, solr_escaped=True):
    if not solr_escaped:
        mention_data_key = "sent_context_orig"
    else:
        mention_data_key = "sent_context"

    if key.endswith("next"):
        if key.endswith("prev_next"):
            res = "{} {} {}".format(
                ""
                if mention[mention_data_key][0] is None
                else mention[mention_data_key][0],
                mention[mention_data_key][1],
                ""
                if mention[mention_data_key][2] is None
                else mention[mention_data_key][2],
            )
        else:
            res = "{} {}".format(
                mention[mention_data_key][1],
                ""
                if mention[mention_data_key][2] is None
                else mention[mention_data_key][2],
            )
    elif key.endswith("prev"):
        res = "{} {}".format(
            ""
            if mention[mention_data_key][0] is None
            else mention[mention_data_key][0],
            mention[mention_data_key][1],
        )
    else:
        res = mention[mention_data_key][1]

    return res.strip()