class ConnecDialog(ParentDialog):
def __init__(self, dialog, layer, feature):
''' Constructor class '''
super(ConnecDialog, self).__init__(dialog, layer, feature)
self.init_config_form()
def init_config_form(self):
''' Custom form initial configuration '''
# Define local variables
context_name = "ws_connec"
table_element = "v_ui_element_x_connec"
table_document = "v_ui_doc_x_connec"
table_hydrometer = "v_rtc_hydrometer"
table_hydrometer_epanet = "v_edit_rtc_hydro_data_x_connec"
# Define class variables
self.field_id = "connec_id"
self.id = utils_giswater.getWidgetText(self.field_id, False)
self.filter = self.field_id+" = '"+str(self.id)+"'"
self.connec_type = utils_giswater.getWidgetText("cat_connectype_id", False)
self.connecat_id = utils_giswater.getWidgetText("connecat_id", False)
# Get widget controls
self.tab_analysis = self.dialog.findChild(QTabWidget, "tab_analysis")
self.tab_event = self.dialog.findChild(QTabWidget, "tab_event")
self.tab_event_2 = self.dialog.findChild(QTabWidget, "tab_event_2")
self.tab_main = self.dialog.findChild(QTabWidget, "tab_main")
self.tbl_info = self.dialog.findChild(QTableView, "tbl_info")
self.tbl_document = self.dialog.findChild(QTableView, "tbl_connec")
self.tbl_dae = self.dialog.findChild(QTableView, "tbl_dae")
self.tbl_dae_2 = self.dialog.findChild(QTableView, "tbl_dae_2")
# Manage tab visibility
self.set_tabs_visibility()
# Manage i18n
self.translate_form(context_name)
# Define and execute query to populate combo 'cat_connectype_id_dummy'
self.fill_connec_type_id()
# Load data from related tables
self.load_data()
# Set layer in editing mode
self.layer.startEditing()
# Fill the info table
self.fill_table(self.tbl_info, self.schema_name+"."+table_element, self.filter)
# Configuration of info table
self.set_configuration(self.tbl_info, table_element)
# Fill the tab Document
self.fill_tbl_document(self.tbl_document, self.schema_name+"."+table_document, self.filter)
# Configuration of table Document
self.set_configuration(self.tbl_document, table_document)
# Fill tab Hydrometer | feature
self.fill_tbl_hydrometer(self.tbl_dae, self.schema_name+"."+table_hydrometer, self.filter)
# Configuration of table Hydrometer | feature
self.set_configuration(self.tbl_dae, table_hydrometer)
# Fill tab Hydrometer | epanet
self.fill_tbl_hydrometer_epanet(self.tbl_dae_2, self.schema_name+"."+table_hydrometer_epanet, self.filter)
# Configuration of table Hydrometer | epanet
self.set_configuration(self.tbl_dae_2, table_hydrometer_epanet)
# Set signals
self.dialog.findChild(QPushButton, "delete_row_info").clicked.connect(partial(self.delete_records, self.tbl_info, table_element))
self.dialog.findChild(QPushButton, "delete_row_doc").clicked.connect(partial(self.delete_records, self.tbl_document, table_document))
self.dialog.findChild(QPushButton, "btn_delete_hydrometer").clicked.connect(partial(self.delete_records_dae, self.tbl_dae, table_hydrometer))
self.dialog.findChild(QPushButton, "btn_add_hydrometer").clicked.connect(self.insert_records)
def set_tabs_visibility(self):
''' Hide some tabs '''
# Remove tabs: Event, Log
self.tab_main.removeTab(4)
self.tab_main.removeTab(3)
def fill_connec_type_id(self):
''' Define and execute query to populate combo 'cat_connectype_id_dummy' '''
sql = "SELECT connec_type.id"
sql+= " FROM "+self.schema_name+".connec_type INNER JOIN "+self.schema_name+".cat_connec ON connec_type.id = cat_connec.type"
sql+= " GROUP BY connec_type.id ORDER BY connec_type.id"
rows = self.dao.get_rows(sql)
utils_giswater.fillComboBox("cat_connectype_id_dummy", rows, False)
utils_giswater.setWidgetText("cat_connectype_id_dummy", self.connec_type)
def change_connec_type_id(self, index):
''' Define and execute query to populate combo 'cat_nodetype_id' '''
connec_type_id = utils_giswater.getWidgetText("cat_connectype_id_dummy", False)
if connec_type_id:
utils_giswater.setWidgetText("cat_connectype_id", connec_type_id)
sql = "SELECT id FROM "+self.schema_name+".cat_connec"
sql+= " WHERE type = '"+connec_type_id+"' ORDER BY id"
rows = self.dao.get_rows(sql)
utils_giswater.fillComboBox("connecat_id_dummy", rows, False)
if index == -1:
utils_giswater.setWidgetText("connecat_id_dummy", self.connecat_id)
self.change_connec_cat()
def change_connec_cat(self):
''' Just select item to 'real' combo 'connecat_id' (that is hidden) '''
connecat_id_dummy = utils_giswater.getWidgetText("connecat_id_dummy")
utils_giswater.setWidgetText("connecat_id", connecat_id_dummy)
def set_filter_tbl_hydrometer(self):
''' Get values selected by the user and sets a new filter for its table model '''
# Get selected dates
date_from = self.date_dae_from.date().toString('yyyyMMdd')
date_to = self.date_dae_to.date().toString('yyyyMMdd')
if (date_from > date_to):
message = "Selected date interval is not valid"
self.controller.show_warning(message, context_name='ui_message' )
return
# Set filter
expr = self.field_id+" = '"+self.id+"'"
expr+= " AND date >= '"+date_from+"' AND date <= '"+date_to+"'"
# Refresh model with selected filter
self.tbl_dae.model().setFilter(expr)
self.tbl_dae.model().select()
def fill_tbl_hydrometer(self, widget, table_name, filter_):
# Get widgets
self.date_dae_to = self.dialog.findChild(QDateEdit, "date_dae_to")
self.date_dae_from = self.dialog.findChild(QDateEdit, "date_dae_from")
# Set signals
self.date_dae_to.dateChanged.connect(self.set_filter_tbl_hydrometer)
self.date_dae_from.dateChanged.connect(self.set_filter_tbl_hydrometer)
# Fill feature tab of connec
# Filter and fill table related with connec
self.set_model_to_table(widget, table_name, filter_)
def fill_tbl_hydrometer_epanet(self, widget, table_name, filter_):
# Fill EPANET tab of hydrometer
# Filter and fill table related with connec_id
self.set_model_to_table(widget, table_name, filter_)
def insert_records (self):
# Create the dialog and signals
self.dlg_sum = Add_sum()
utils_giswater.setDialog(self.dlg_sum)
# Set signals
self.dlg_sum.findChild(QPushButton, "btn_accept").clicked.connect(self.btn_accept_dae)
self.dlg_sum.findChild(QPushButton, "btn_close").clicked.connect(self.btn_close_dae)
# Open the dialog
self.dlg_sum.exec_()
def btn_accept_dae(self):
# Get widget text - hydtometer_id
widget_hydro = self.dlg_sum.findChild(QLineEdit, "hydrometer_id_new")
self.hydro_id = widget_hydro.text()
# get connec_id
widget_connec = self.dialog.findChild(QLineEdit, "connec_id")
self.connec_id = widget_connec.text()
# Insert hydrometer_id in v_rtc_hydrometer
sql = "INSERT INTO "+self.schema_name+".rtc_hydrometer (hydrometer_id) "
sql+= " VALUES ('"+self.hydro_id+"')"
self.dao.execute_sql(sql)
# insert hydtometer_id and connec_id in rtc_hydrometer_x_connec
sql = "INSERT INTO "+self.schema_name+".rtc_hydrometer_x_connec (hydrometer_id, connec_id) "
sql+= " VALUES ('"+self.hydro_id+"','"+self.connec_id+"')"
self.dao.execute_sql(sql)
# Refresh table in Qtableview
# Fill tab Hydrometer
table_hydrometer = "v_rtc_hydrometer"
self.fill_tbl_hydrometer(self.tbl_dae, self.schema_name+"."+table_hydrometer, self.filter)
self.dlg_sum.close()
def btn_close_dae(self):
''' Close form without saving '''
self.dlg_sum.close()
def delete_records_dae(self, widget, table_name): #@UnusedVariable
''' Delete selected elements of the table '''
# Get selected rows
selected_list = widget.selectionModel().selectedRows()
if len(selected_list) == 0:
message = "Any record selected"
self.controller.show_warning(message, context_name='ui_message' )
return
inf_text = ""
list_id = ""
for i in range(0, len(selected_list)):
row = selected_list[i].row()
id_ = widget.model().record(row).value("hydrometer_id")
inf_text+= str(id_)+", "
list_id = list_id+"'"+str(id_)+"', "
inf_text = inf_text[:-2]
list_id = list_id[:-2]
answer = self.controller.ask_question("Are you sure you want to delete these records?", "Delete records", inf_text)
if answer:
sql= "DELETE FROM "+self.schema_name+".rtc_hydrometer_x_connec WHERE hydrometer_id ='"+id_+"'"
self.dao.execute_sql(sql)
widget.model().select()
sql= "DELETE FROM "+self.schema_name+".v_rtc_hydrometer WHERE hydrometer_id ='"+id_+"'"
self.dao.execute_sql(sql)
widget.model().select()
# Refresh table in Qtableview
# Fill tab Hydrometer
table_hydrometer = "v_rtc_hydrometer"
self.fill_tbl_hydrometer(self.tbl_dae, self.schema_name+"."+table_hydrometer, self.filter)
def update_sum(self):
''' Update contents of the selected widget '''
# Submit all changes made to the table
# View will automatically control with fields can be updated
status = self.tbl_dae_2.model().submitAll()
if status:
self.tbl_dae_2.model().database().commit()
self.tbl_dae_2.model().select()
else:
self.tbl_dae_2.model().database().rollback()
error = self.tbl_dae_2.model().lastError()
print str(error.text())
self.save()