def __init__(self, config, treeview, parent=None):
from stdm.data import (
foreign_key_parent_tables,
numeric_varchar_columns,
spatial_tables
)
super(EntityNodeFormatter, self).__init__(config, treeview, parent)
self._str_ref = "social_tenure_relationship"
self._str_title = QApplication.translate("STRFormatterBase",
"Social Tenure Relationship")
self._str_model = DeclareMapping.instance().tableMapping(self._str_ref)
'''
Set STR display mapping due to a bug in the 'displayMapping'
function
'''
self._str_model_disp_mapping = {}
if not self._str_model is None:
self._str_model_disp_mapping = self._str_model.displayMapping()
self._str_num_char_cols = numeric_varchar_columns(self._str_ref)
self._fk_references = foreign_key_parent_tables(self._str_ref)
self._current_data_source_fk_ref = self._current_data_source_foreign_key_reference()
self._numeric_char_cols = numeric_varchar_columns(config.data_source_name)
self._spatial_data_sources = spatial_tables()
def db_tables(self):
"""
Returns both textual and spatial table names.
"""
tables = pg_tables(exclude_lookups=False)
tables.extend(spatial_tables(exclude_views=True))
return tables
def loadTables(self,type):
#Load textual or spatial tables
self.lstDestTables.clear()
if type=="textual":
tables = pg_tables(excludeLookups=False)
elif type=="spatial":
tables = spatial_tables(excludeViews=True)
for t in tables:
tabItem = QListWidgetItem(t,self.lstDestTables)
tabItem.setCheckState(Qt.Unchecked)
tabItem.setIcon(QIcon(":/plugins/stdm/images/icons/table.png"))
self.lstDestTables.addItem(tabItem)
