def dialog(self, qtbot, example_file_path) -> DataFrameModelImportDialog:
dfm = DataFrameModelManager()
dialog = DataFrameModelImportDialog(dfm, file_path=example_file_path)
dialog.show()
qtbot.add_widget(dialog)
return dialog
def __init__(self, settings_ini: (str, SettingsINI), parent=None):
"""
:param settings_ini: (str, SettingsINI)
can be a settings_ini file path or configured SettingsINI object.
"""
self.df_manager = DataFrameModelManager()
QtGui.QMainWindow.__init__(self, parent=parent)
self.setupUi(self)
self.icons = Icons()
self.dialog_settings = SettingsDialog(settings=settings_ini)
self.dialog_merge_purge = MergePurgeDialog(self.df_manager)
self.dialog_export = DataFrameModelExportDialog(self.df_manager,
parent=self)
self.dialog_import = DataFrameModelImportDialog(self.df_manager,
parent=self)
self.dialog_new_folder = DirectoryPathCreateDialog(self.treeView,
parent=self)
self.dialog_cloud = None
self.key_delete = QtGui.QShortcut(self)
self.key_enter = QtGui.QShortcut(self)
self.key_zip = QtGui.QShortcut(self)
self.key_rename = QtGui.QShortcut(self)
self.connect_window_title()
self.connect_actions()
self.connect_treeview()
self.connect_icons()
self.connect_settings_dialog()
self.connect_import_dialog()
self.connect_export_dialog()
self.connect_cloud_dialog()
self.current_model = None
def __init__(self,
directory,
settings_ini,
default_dirs=True,
tree_view=None,
main_control=None,
**kwargs):
self.parent = kwargs.get('parent', None)
self.main_control = main_control
if default_dirs is True:
settings_ini.set_safe('GENERAL', 'ROOT_DIRECTORY', directory)
settings_ini.set_safe('GENERAL', 'LOG_DIRECTORY',
os.path.join(directory, 'logs'))
self._directory = directory
self._df_manager = DataFrameModelManager()
self._file_tree_model = FileTreeModel(root_dir=directory,
parent=self.parent)
self._dialog_export_df_model = DataFrameModelExportDialog(
self.df_manager)
self._dialog_merge_purge = MergePurgeDialog(self.df_manager,
parent=self.parent)
self._dialog_add_directory = DirectoryPathCreateDialog(
base_dirname=directory, parent=self.parent)
self._dialog_import_df_model = DataFrameModelImportDialog(
self.df_manager, parent=self.parent)
self._dialog_import_df_model.signalImported.connect(
self._handle_imported_df_model)
self._dialog_settings = ProjectSettingsDialog(settings_ini)
self._tree_view = None
self.tree_view = tree_view
def main_window(self, qtbot, sqlite_db_path):
"""
Creates & returns a DatabasesMainWindow object loaded with one test sqlite connection
named "con1" - any existing tables are dropped.
:param qtbot:
:param sqlite_db_path:
:return:
"""
c = AlchemyConnection(name="con1")
c.configure("sqlite:///" + sqlite_db_path, reset=True)
c.meta.drop_all()
cm = AlchemyConnectionManager()
cm.add_connection(connection=c)
window = DatabasesMainWindow(df_manager=DataFrameModelManager(), connection_manager=cm)
qtbot.addWidget(window)
return window
def configure(self):
"""
called once on __init__
- Sets AlchemyConnectionManager and/or DataFrameModelManager if they
were not set in the __init__(**kwargs)
- Connects default actions
- sets treeView model.
:return: (None)
"""
self.setupUi(self)
if self.con_manager is None:
self.con_manager = AlchemyConnectionManager()
if self.df_manager is None:
self.df_manager = DataFrameModelManager()
self._key_enter.setKey('return')
self._key_ctrl_t.setKey('ctrl+T')
self._key_enter.activated.connect(self.open_query_alchemyview)
self._key_ctrl_t.activated.connect(self.open_table_description_dialog)
self.treeView.setModel(self.con_manager.get_standard_item_model())
self.actionRemove.triggered.connect(self.delete)
self.actionRefreshSchemas.triggered.connect(self.refresh_schemas)
self.actionSaveText.triggered.connect(self.save_last_sql_text)
self.actionSaveTextAs.triggered.connect(self.save_sql_text)
self.actionOpenFile.triggered.connect(self.open_sql_text)
self.actionOpenQueryData.triggered.connect(self.open_query_fileview)
self.actionConnectToDatabase.triggered.connect(self.connect_database)
self.actionDisconnectFromDatabase.triggered.connect(
self.disconnect_database)
self.actionExportFile.triggered.connect(self.export_table)
self.actionImportFile.triggered.connect(self.open_import_dialog)
self.actionAddDatabase.triggered.connect(
self.open_add_connection_dialog)
self.actionExecuteQuery.triggered.connect(self.execute_query)
self.actionExecuteSelectedQuery.triggered.connect(
self.execute_query_selected)
self.treeView.expanded.connect(self.sync_current_database)
self.connect_default_databases()
def manager(self) -> DataFrameModelManager:
return DataFrameModelManager()
def df_manager(self) -> DataFrameModelManager:
if self._df_manager is None:
self._df_manager = DataFrameModelManager()
return self._df_manager