def transform(self, raw_documents):
documents = []
for raw_doc in raw_documents:
everything = []
allWords, required_values, target, conditions, tables, question_type, proper_nouns, component_name = nlp_nlidb.nlp_nlidb(
raw_doc
)
# everything.extend(allWords)
# everything.extend(required_values)
# everything.extend(target)
everything.extend(conditions)
# everything.extend(tables)
# everything.extend(question_type)
# everything.extend(question)
everything.extend(allWords)
# everything.extend(proper_nouns)
if len(everything) == 0:
documents.append(raw_doc)
else:
documents.append(" ".join(everything))
return documents
def to_sql(self, nl_query):
if FuzzyAdventure.easter_egg(nl_query):
return [42]
sql, category = FuzzyAdventure.tc.template(nl_query)
keywords = []
allWords, required_values, target, conditions, tables, question_type, proper_nouns, component_name = nlp_nlidb.nlp_nlidb(nl_query)
keywords.extend(allWords)
keywords.extend(conditions)
keywords.extend(component_name)
keywords.extend(map(lambda x: x.upper(), proper_nouns))
answer = term_selector.TermSelector.fill_in_the_blanks(sql, keywords)
return answer
