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)