def indexed_field(self, signature):
return utils.to_indexable(signature, self.caseless) in self.field_index
def record_table(self, table_name):
table_name = to_indexable(table_name)
self.tables.append(table_name)
def __init__(self, node_type, name, n_name=None, caseless=True):
self.node_type = node_type
self.name = name
self.normalized_name = n_name if n_name else name
self.indexable_name = utils.to_indexable(name, caseless)
self.lexical_features = None
def indexed_table(self, signature):
return utils.to_indexable(signature, self.caseless) in self.table_index
def get_schema_pos(self, signature):
if signature == '*':
return 0
signature = utils.to_indexable(signature, self.caseless)
# print(signature, self.bert_feature_idx.get(signature, None))
return self.bert_feature_idx.get(signature, None)
def field_in_table(self, f_name, t_name):
indexable_signature = '{}.{}'.format(
utils.to_indexable(t_name, self.caseless),
utils.to_indexable(f_name, self.caseless))
return indexable_signature in self.field_index
def is_field_name(self, name):
return utils.to_indexable(name, self.caseless) in self.field_names
def is_table_name(self, name):
return utils.to_indexable(name, self.caseless) in self.table_names
def get_field_id(self, signature):
signature = utils.to_indexable(signature, self.caseless)
return self.field_index.get(signature, None)
