def test_dataframe_multiindex(): """Test to validate proper creation and handling of a multiindex.""" arrays = [numpy.array(['bar', 'bar', 'baz', 'baz', 'foo', 'foo', 'qux', 'qux']), numpy.array(['one', 'two', 'one', 'two', 'one', 'two', 'one', 'two'])] tuples = list(zip(*arrays)) index = MultiIndex.from_tuples(tuples, names=['first', 'second']) df = DataFrame(numpy.random.randn(6, 6), index=index[:6], columns=index[:6]) editor = DataFrameEditor(None) editor.setup_and_check(df) header = editor.table_header.model() assert header.headerData(0, Qt.Horizontal, Qt.DisplayRole) == 0 assert data_header(header, 0, 0) == 'bar' assert data_header(header, 1, 0) == 'one' assert data_header(header, 0, 1) == 'bar' assert data_header(header, 1, 1) == 'two' assert data_header(header, 0, 2) == 'baz' assert data_header(header, 1, 2) == 'one' assert data_header(header, 0, 3) == 'baz' assert data_header(header, 1, 3) == 'two' assert data_header(header, 0, 4) == 'foo' assert data_header(header, 1, 4) == 'one' assert data_header(header, 0, 5) == 'foo' assert data_header(header, 1, 5) == 'two'
def test_header_bom(): """Test for BOM data in the headers.""" df = read_csv(os.path.join(FILES_PATH, 'issue_2514.csv')) editor = DataFrameEditor(None) editor.setup_and_check(df) header = editor.table_header.model() assert header.headerData(0, Qt.Horizontal, Qt.DisplayRole) == "Date (MMM-YY)"
def test_change_format_emits_signal(qtbot, monkeypatch): mockQInputDialog = Mock() mockQInputDialog.getText = lambda parent, title, label, mode, text: ('%10.3e', True) monkeypatch.setattr('spyder.plugins.variableexplorer.widgets.dataframeeditor.QInputDialog', mockQInputDialog) df = DataFrame([[0]]) editor = DataFrameEditor(None) editor.setup_and_check(df) with qtbot.waitSignal(editor.sig_option_changed) as blocker: editor.change_format() assert blocker.args == ['dataframe_format', '%10.3e']
def test_sort_dataframe_with_category_dtypes(qtbot): # cf. issue 5361 df = DataFrame({'A': [1, 2, 3, 4], 'B': ['a', 'b', 'c', 'd']}) df = df.astype(dtype={'B': 'category'}) df_cols = df.dtypes editor = DataFrameEditor(None) editor.setup_and_check(df_cols) dfm = editor.dataModel QTimer.singleShot(1000, lambda: close_message_box(qtbot)) editor.dataModel.sort(0) assert data(dfm, 0, 0) == 'int64' assert data(dfm, 1, 0) == 'category'
def test_change_format_with_format_not_starting_with_percent(qtbot, monkeypatch): mockQInputDialog = Mock() mockQInputDialog.getText = lambda parent, title, label, mode, text: ('xxx%f', True) monkeypatch.setattr('spyder.plugins.variableexplorer.widgets.dataframeeditor' '.QInputDialog', mockQInputDialog) monkeypatch.setattr('spyder.plugins.variableexplorer.widgets.dataframeeditor' '.QMessageBox.critical', Mock()) df = DataFrame([[0]]) editor = DataFrameEditor(None) editor.setup_and_check(df) with qtbot.assertNotEmitted(editor.sig_option_changed): editor.change_format()
def test_dataframe_simpleindex(qtbot): """Test to validate proper creation and handling of a simpleindex.""" df = DataFrame(numpy.random.randn(6, 6)) editor = DataFrameEditor(None) editor.setup_and_check(df) header = editor.table_header.model() assert header.headerData(0, Qt.Horizontal, Qt.DisplayRole) == "0" assert header.headerData(1, Qt.Horizontal, Qt.DisplayRole) == "1" assert header.headerData(5, Qt.Horizontal, Qt.DisplayRole) == "5"
def test_sort_dataframe_with_duplicate_column(qtbot): df = DataFrame({'A': [1, 3, 2], 'B': [4, 6, 5]}) df = concat((df, df.A), axis=1) editor = DataFrameEditor(None) editor.setup_and_check(df) dfm = editor.dataModel QTimer.singleShot(1000, lambda: close_message_box(qtbot)) editor.dataModel.sort(0) assert [data(dfm, row, 0) for row in range(len(df))] == ['1', '3', '2'] assert [data(dfm, row, 1) for row in range(len(df))] == ['4', '6', '5'] editor.dataModel.sort(1) assert [data(dfm, row, 0) for row in range(len(df))] == ['1', '2', '3'] assert [data(dfm, row, 1) for row in range(len(df))] == ['4', '5', '6']
def test_dataframe_simpleindex_custom_columns(): """Test to validate proper creation and handling of custom simpleindex.""" df = DataFrame(numpy.random.randn(10, 5), columns=['a', 'b', 'c', 'd', 'e']) editor = DataFrameEditor(None) editor.setup_and_check(df) header = editor.table_header.model() assert header.headerData(0, Qt.Horizontal, Qt.DisplayRole) == "a" assert header.headerData(1, Qt.Horizontal, Qt.DisplayRole) == "b" assert header.headerData(4, Qt.Horizontal, Qt.DisplayRole) == "e"
def test_non_ascii_index(): """ Test that there are no errors when displaying a dataframe with a non-ascii index and header. """ df = read_csv(os.path.join(FILES_PATH, 'issue_5833.csv'), index_col=0) editor = DataFrameEditor(None) editor.setup_and_check(df) index = editor.table_index.model() header = editor.table_header.model() dfm = editor.model() assert header.headerData(0, Qt.Horizontal, Qt.DisplayRole) == "кодирование" assert data_index(index, 0, 0) == 'пример' assert data(dfm, 0, 0) == 'файла'
def test_header_encoding(): """Test for header encoding handling.""" df = read_csv(os.path.join(FILES_PATH, 'issue_3896.csv')) editor = DataFrameEditor(None) editor.setup_and_check(df) header = editor.table_header.model() assert header.headerData(0, Qt.Horizontal, Qt.DisplayRole) == "Unnamed: 0" assert "Unieke_Idcode" in header.headerData(1, Qt.Horizontal, Qt.DisplayRole) assert header.headerData(2, Qt.Horizontal, Qt.DisplayRole) == "a" assert header.headerData(3, Qt.Horizontal, Qt.DisplayRole) == "b" assert header.headerData(4, Qt.Horizontal, Qt.DisplayRole) == "c" assert header.headerData(5, Qt.Horizontal, Qt.DisplayRole) == "d"
def test_no_convert_strings_to_unicode(): """ Test that we don't apply any conversion to strings in headers, indexes or data. """ df = read_csv(os.path.join(FILES_PATH, 'issue_5833.csv'), index_col=0, encoding='koi8_r') editor = DataFrameEditor(None) editor.setup_and_check(df) index = editor.table_index.model() header = editor.table_header.model() dfm = editor.model() assert header.headerData(0, Qt.Horizontal, Qt.DisplayRole) != u"кодирование" assert data_index(index, 0, 0) != u'пример' assert data(dfm, 0, 0) != u'файла'
def test_dataframeeditor_edit_overflow(qtbot, monkeypatch): """ Test that entry of an overflowing integer is caught and handled properly. Integration regression test for issue #6114 . """ MockQMessageBox = Mock() attr_to_patch = ('spyder.plugins.variableexplorer.widgets' + '.dataframeeditor.QMessageBox') monkeypatch.setattr(attr_to_patch, MockQMessageBox) # Numpy doesn't raise the OverflowError for ints smaller than 64 bits if not os.name == 'nt': int32_bit_exponent = 66 else: int32_bit_exponent = 34 test_parameters = [(1, numpy.int32, int32_bit_exponent), (2, numpy.int64, 66)] expected_df = DataFrame([5, 6, 7, 3, 4]) for idx, int_type, bit_exponet in test_parameters: test_df = DataFrame(numpy.arange(0, 5), dtype=int_type) dialog = DataFrameEditor() assert dialog.setup_and_check(test_df, 'Test Dataframe') dialog.show() qtbot.waitForWindowShown(dialog) view = dialog.dataTable qtbot.keyClick(view, Qt.Key_Right) qtbot.keyClicks(view, '5') qtbot.keyClick(view, Qt.Key_Down) qtbot.keyClick(view, Qt.Key_Space) qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace) qtbot.keyClicks(view.focusWidget(), str(int(2 ** bit_exponet))) qtbot.keyClick(view.focusWidget(), Qt.Key_Down) MockQMessageBox.critical.assert_called_with(ANY, "Error", ANY) assert MockQMessageBox.critical.call_count == idx qtbot.keyClicks(view, '7') qtbot.keyClick(view, Qt.Key_Up) qtbot.keyClicks(view, '6') qtbot.keyClick(view, Qt.Key_Down) qtbot.wait(200) dialog.accept() qtbot.wait(500) try: assert numpy.sum(expected_df[0].values == dialog.get_value().values) == len(expected_df) except AttributeError: assert numpy.sum( expected_df[0].as_matrix() == dialog.get_value().as_matrix()) == len(expected_df)
def test_dataframeeditor_with_various_indexes(): for rng_name, rng in generate_pandas_indexes().items(): editor = DataFrameEditor(None) editor.setup_and_check(rng) dfm = editor.dataModel assert dfm.rowCount() == 20 assert dfm.columnCount() == 1 header = editor.table_header.model() assert header.headerData(0, Qt.Horizontal, Qt.DisplayRole) == "0" if rng_name == "Index": assert data(dfm, 0, 0) == 'A' assert data(dfm, 1, 0) == 'B' assert data(dfm, 2, 0) == 'C' assert data(dfm, 19, 0) == 'T' elif rng_name == "RangeIndex": assert data(dfm, 0, 0) == '0' assert data(dfm, 1, 0) == '1' assert data(dfm, 2, 0) == '2' assert data(dfm, 19, 0) == '19' elif rng_name == "Float64Index": assert data(dfm, 0, 0) == '0' assert data(dfm, 1, 0) == '0.1' assert data(dfm, 2, 0) == '0.2' assert data(dfm, 19, 0) == '1.9' elif rng_name == "DatetimeIndex": assert data(dfm, 0, 0) == '2017-01-01 00:00:00' assert data(dfm, 1, 0) == '2017-01-02 00:00:00' assert data(dfm, 2, 0) == '2017-01-03 00:00:00' assert data(dfm, 19, 0) == '2017-01-20 00:00:00' elif rng_name == "MultiIndex": assert data(dfm, 0, 0) == "('A', 'foo')" assert data(dfm, 1, 0) == "('A', 'bar')" assert data(dfm, 2, 0) == "('B', 'foo')" assert data(dfm, 19, 0) == "('J', 'bar')" elif rng_name == "CategoricalIndex": assert data(dfm, 0, 0) == 'a' assert data(dfm, 1, 0) == 'b' assert data(dfm, 2, 0) == 'c' assert data(dfm, 19, 0) == 'b'
def test_dataframeeditor_edit_complex(qtbot, monkeypatch): """ Test that editing complex dtypes is handled gracefully in df editor. Integration regression test for issue #6115 . """ MockQMessageBox = Mock() attr_to_patch = ('spyder.plugins.variableexplorer.widgets' + '.dataframeeditor.QMessageBox') monkeypatch.setattr(attr_to_patch, MockQMessageBox) test_params = [(1, numpy.complex128), (2, numpy.complex64), (3, complex)] for count, complex_type in test_params: test_df = DataFrame(numpy.arange(10, 15), dtype=complex_type) dialog = DataFrameEditor() assert dialog.setup_and_check(test_df, 'Test Dataframe') dialog.show() qtbot.waitForWindowShown(dialog) view = dialog.dataTable qtbot.keyClick(view, Qt.Key_Right) qtbot.keyClick(view, Qt.Key_Down) qtbot.keyClick(view, Qt.Key_Space) qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace) qtbot.keyClicks(view.focusWidget(), "42") qtbot.keyClick(view.focusWidget(), Qt.Key_Down) MockQMessageBox.critical.assert_called_with(ANY, "Error", ANY) assert MockQMessageBox.critical.call_count == count * 2 - 1 qtbot.keyClick(view, Qt.Key_Down) qtbot.keyClick(view, '1') qtbot.keyClick(view.focusWidget(), Qt.Key_Down) MockQMessageBox.critical.assert_called_with( ANY, "Error", ("Editing dtype {0!s} not yet supported." .format(type(test_df.iloc[1, 0]).__name__))) assert MockQMessageBox.critical.call_count == count * 2 qtbot.wait(200) dialog.accept() qtbot.wait(500) try: assert numpy.sum(test_df[0].values == dialog.get_value().values) == len(test_df) except AttributeError: assert numpy.sum(test_df[0].as_matrix() == dialog.get_value().as_matrix()) == len(test_df)
def test_dataframeeditor_edit_bool(qtbot, monkeypatch): """Test that bools are editible in df and false-y strs are detected.""" MockQMessageBox = Mock() attr_to_patch = ('spyder.plugins.variableexplorer.widgets' + '.dataframeeditor.QMessageBox') monkeypatch.setattr(attr_to_patch, MockQMessageBox) test_params = [numpy.bool_, numpy.bool, bool] test_strs = ['foo', 'false', 'f', '0', '0.', '0.0', '', ' '] expected_df = DataFrame([1, 0, 0, 0, 0, 0, 0, 0, 0], dtype=bool) for bool_type in test_params: test_df = DataFrame([0, 1, 1, 1, 1, 1, 1, 1, 0], dtype=bool_type) dialog = DataFrameEditor() assert dialog.setup_and_check(test_df, 'Test Dataframe') dialog.show() qtbot.waitForWindowShown(dialog) view = dialog.dataTable qtbot.keyClick(view, Qt.Key_Right) for test_str in test_strs: qtbot.keyClick(view, Qt.Key_Space) qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace) qtbot.keyClicks(view.focusWidget(), test_str) qtbot.keyClick(view.focusWidget(), Qt.Key_Down) assert not MockQMessageBox.critical.called qtbot.wait(200) dialog.accept() qtbot.wait(500) try: assert (numpy.sum(expected_df[0].values == dialog.get_value().values[:, 0]) == len(expected_df)) except AttributeError: assert (numpy.sum(expected_df[0].as_matrix() == dialog.get_value().as_matrix()[:, 0]) == len(expected_df))
def createEditor(self, parent, option, index, object_explorer=False): """Overriding method createEditor""" val_type = index.sibling(index.row(), 1).data() self.sig_editor_creation_started.emit() if index.column() < 3: return None if self.show_warning(index): answer = QMessageBox.warning( self.parent(), _("Warning"), _("Opening this variable can be slow\n\n" "Do you want to continue anyway?"), QMessageBox.Yes | QMessageBox.No) if answer == QMessageBox.No: self.sig_editor_shown.emit() return None try: value = self.get_value(index) if value is None: return None except ImportError as msg: self.sig_editor_shown.emit() module = str(msg).split("'")[1] if module in ['pandas', 'numpy']: if module == 'numpy': val_type = 'array' else: val_type = 'dataframe, series' message = _("Spyder is unable to show the {val_type} or object" " you're trying to view because <tt>{module}</tt>" " is not installed. ") if running_in_mac_app(): message += _("Please consider using the full version of " "the Spyder MacOS application.<br>") else: message += _("Please install this package in your Spyder " "environment.<br>") QMessageBox.critical( self.parent(), _("Error"), message.format(val_type=val_type, module=module)) return else: if running_in_mac_app() or is_pynsist(): message = _("Spyder is unable to show the variable you're" " trying to view because the module " "<tt>{module}</tt> is not supported in the " "Spyder Lite application.<br>") else: message = _("Spyder is unable to show the variable you're" " trying to view because the module " "<tt>{module}</tt> is not found in your " "Spyder environment. Please install this " "package in this environment.<br>") QMessageBox.critical(self.parent(), _("Error"), message.format(module=module)) return except Exception as msg: QMessageBox.critical( self.parent(), _("Error"), _("Spyder was unable to retrieve the value of " "this variable from the console.<br><br>" "The error message was:<br>" "%s") % to_text_string(msg)) return key = index.model().get_key(index) readonly = (isinstance(value, (tuple, set)) or self.parent().readonly or not is_known_type(value)) # We can't edit Numpy void objects because they could be anything, so # this might cause a crash. # Fixes spyder-ide/spyder#10603 if isinstance(value, np.void): self.sig_editor_shown.emit() return None # CollectionsEditor for a list, tuple, dict, etc. elif isinstance(value, (list, set, tuple, dict)) and not object_explorer: from spyder.widgets.collectionseditor import CollectionsEditor editor = CollectionsEditor(parent=parent) editor.setup(value, key, icon=self.parent().windowIcon(), readonly=readonly) self.create_dialog(editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # ArrayEditor for a Numpy array elif (isinstance(value, (np.ndarray, np.ma.MaskedArray)) and np.ndarray is not FakeObject and not object_explorer): # We need to leave this import here for tests to pass. from .arrayeditor import ArrayEditor editor = ArrayEditor(parent=parent) if not editor.setup_and_check(value, title=key, readonly=readonly): return self.create_dialog(editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # ArrayEditor for an images elif (isinstance(value, PIL.Image.Image) and np.ndarray is not FakeObject and PIL.Image is not FakeObject and not object_explorer): # Sometimes the ArrayEditor import above is not seen (don't know # why), so we need to reimport it here. # Fixes spyder-ide/spyder#16731 from .arrayeditor import ArrayEditor arr = np.array(value) editor = ArrayEditor(parent=parent) if not editor.setup_and_check(arr, title=key, readonly=readonly): return conv_func = lambda arr: PIL.Image.fromarray(arr, mode=value.mode) self.create_dialog(editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly, conv=conv_func)) return None # DataFrameEditor for a pandas dataframe, series or index elif (isinstance(value, (pd.DataFrame, pd.Index, pd.Series)) and pd.DataFrame is not FakeObject and not object_explorer): # We need to leave this import here for tests to pass. from .dataframeeditor import DataFrameEditor editor = DataFrameEditor(parent=parent) if not editor.setup_and_check(value, title=key): self.sig_editor_shown.emit() return self.create_dialog(editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # QDateEdit and QDateTimeEdit for a dates or datetime respectively elif isinstance(value, datetime.date) and not object_explorer: if readonly: self.sig_editor_shown.emit() return None else: if isinstance(value, datetime.datetime): editor = QDateTimeEdit(value, parent=parent) # Needed to handle NaT values # See spyder-ide/spyder#8329 try: value.time() except ValueError: self.sig_editor_shown.emit() return None else: editor = QDateEdit(value, parent=parent) editor.setCalendarPopup(True) editor.setFont(get_font(font_size_delta=DEFAULT_SMALL_DELTA)) self.sig_editor_shown.emit() return editor # TextEditor for a long string elif is_text_string(value) and len(value) > 40 and not object_explorer: te = TextEditor(None, parent=parent) if te.setup_and_check(value): editor = TextEditor(value, key, readonly=readonly, parent=parent) self.create_dialog(editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # QLineEdit for an individual value (int, float, short string, etc) elif is_editable_type(value) and not object_explorer: if readonly: self.sig_editor_shown.emit() return None else: editor = QLineEdit(parent=parent) editor.setFont(get_font(font_size_delta=DEFAULT_SMALL_DELTA)) editor.setAlignment(Qt.AlignLeft) # This is making Spyder crash because the QLineEdit that it's # been modified is removed and a new one is created after # evaluation. So the object on which this method is trying to # act doesn't exist anymore. # editor.returnPressed.connect(self.commitAndCloseEditor) self.sig_editor_shown.emit() return editor # ObjectExplorer for an arbitrary Python object else: from spyder.plugins.variableexplorer.widgets.objectexplorer \ import ObjectExplorer editor = ObjectExplorer( value, name=key, parent=parent, readonly=readonly) self.create_dialog(editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None
def createEditor(self, parent, option, index): """Overriding method createEditor""" if self.show_warning(index): answer = QMessageBox.warning( self.parent(), _("Warning"), _("Opening this variable can be slow\n\n" "Do you want to continue anyway?"), QMessageBox.Yes | QMessageBox.No) if answer == QMessageBox.No: return None try: value = self.get_value(index) try: self.old_obj = value.copy() except AttributeError: self.old_obj = copy.deepcopy(value) if value is None: return None except Exception as msg: QMessageBox.critical( self.parent(), _("Error"), _("Spyder was unable to retrieve the value of " "this variable from the console.<br><br>" "The error message was:<br>" "<i>%s</i>") % to_text_string(msg)) return self.current_index = index key = index.model().get_key(index).obj_name readonly = (isinstance(value, (tuple, set)) or self.parent().readonly or not is_known_type(value)) # CollectionsEditor for a list, tuple, dict, etc. if isinstance(value, (list, set, tuple, dict)): from spyder.widgets.collectionseditor import CollectionsEditor editor = CollectionsEditor(parent=parent) editor.setup(value, key, icon=self.parent().windowIcon(), readonly=readonly) self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # ArrayEditor for a Numpy array elif (isinstance(value, (np.ndarray, np.ma.MaskedArray)) and np.ndarray is not FakeObject): editor = ArrayEditor(parent=parent) if not editor.setup_and_check(value, title=key, readonly=readonly): return self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # ArrayEditor for an images elif (isinstance(value, PIL.Image.Image) and np.ndarray is not FakeObject and PIL.Image is not FakeObject): arr = np.array(value) editor = ArrayEditor(parent=parent) if not editor.setup_and_check(arr, title=key, readonly=readonly): return conv_func = lambda arr: PIL.Image.fromarray(arr, mode=value.mode) self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly, conv=conv_func)) return None # DataFrameEditor for a pandas dataframe, series or index elif (isinstance(value, (pd.DataFrame, pd.Index, pd.Series)) and pd.DataFrame is not FakeObject): editor = DataFrameEditor(parent=parent) if not editor.setup_and_check(value, title=key): return self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # QDateEdit and QDateTimeEdit for a dates or datetime respectively elif isinstance(value, datetime.date): if readonly: return None else: if isinstance(value, datetime.datetime): editor = QDateTimeEdit(value, parent=parent) else: editor = QDateEdit(value, parent=parent) editor.setCalendarPopup(True) editor.setFont(get_font(font_size_delta=DEFAULT_SMALL_DELTA)) return editor # TextEditor for a long string elif is_text_string(value) and len(value) > 40: te = TextEditor(None, parent=parent) if te.setup_and_check(value): editor = TextEditor(value, key, readonly=readonly, parent=parent) self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # QLineEdit for an individual value (int, float, short string, etc) elif is_editable_type(value): if readonly: return None else: editor = QLineEdit(parent=parent) editor.setFont(get_font(font_size_delta=DEFAULT_SMALL_DELTA)) editor.setAlignment(Qt.AlignLeft) # This is making Spyder crash because the QLineEdit that it's # been modified is removed and a new one is created after # evaluation. So the object on which this method is trying to # act doesn't exist anymore. # editor.returnPressed.connect(self.commitAndCloseEditor) return editor # An arbitrary Python object. # Since we are already in the Object Explorer no editor is needed else: return None
def createEditor(self, parent, option, index, object_explorer=False): """Overriding method createEditor""" self.sig_open_editor.emit() if index.column() < 3: return None if self.show_warning(index): answer = QMessageBox.warning( self.parent(), _("Warning"), _("Opening this variable can be slow\n\n" "Do you want to continue anyway?"), QMessageBox.Yes | QMessageBox.No) if answer == QMessageBox.No: return None try: value = self.get_value(index) if value is None: return None except Exception as msg: QMessageBox.critical( self.parent(), _("Error"), _("Spyder was unable to retrieve the value of " "this variable from the console.<br><br>" "The error message was:<br>" "%s") % to_text_string(msg)) return key = index.model().get_key(index) readonly = (isinstance(value, (tuple, set)) or self.parent().readonly or not is_known_type(value)) # CollectionsEditor for a list, tuple, dict, etc. if isinstance(value, (list, set, tuple, dict)) and not object_explorer: from spyder.plugins.variableexplorer.widgets.collectionseditor \ import CollectionsEditor editor = CollectionsEditor(parent=parent) editor.setup(value, key, icon=self.parent().windowIcon(), readonly=readonly) self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # ArrayEditor for a Numpy array elif (isinstance(value, (ndarray, MaskedArray)) and ndarray is not FakeObject and not object_explorer): editor = ArrayEditor(parent=parent) if not editor.setup_and_check(value, title=key, readonly=readonly): return self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # ArrayEditor for an images elif (isinstance(value, Image) and ndarray is not FakeObject and Image is not FakeObject and not object_explorer): arr = array(value) editor = ArrayEditor(parent=parent) if not editor.setup_and_check(arr, title=key, readonly=readonly): return conv_func = lambda arr: Image.fromarray(arr, mode=value.mode) self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly, conv=conv_func)) return None # DataFrameEditor for a pandas dataframe, series or index elif (isinstance(value, (DataFrame, Index, Series)) and DataFrame is not FakeObject and not object_explorer): editor = DataFrameEditor(parent=parent) if not editor.setup_and_check(value, title=key): return editor.dataModel.set_format(index.model().dataframe_format) editor.sig_option_changed.connect(self.change_option) self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # QDateEdit and QDateTimeEdit for a dates or datetime respectively elif isinstance(value, datetime.date) and not object_explorer: # Needed to handle NaT values # See spyder-ide/spyder#8329 try: value.time() except ValueError: self.sig_editor_shown.emit() return None if readonly: self.sig_editor_shown.emit() return None else: if isinstance(value, datetime.datetime): editor = QDateTimeEdit(value, parent=parent) else: editor = QDateEdit(value, parent=parent) editor.setCalendarPopup(True) editor.setFont(get_font(font_size_delta=DEFAULT_SMALL_DELTA)) self.sig_editor_shown.emit() return editor # TextEditor for a long string elif is_text_string(value) and len(value) > 40 and not object_explorer: te = TextEditor(None, parent=parent) if te.setup_and_check(value): editor = TextEditor(value, key, readonly=readonly, parent=parent) self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None # QLineEdit for an individual value (int, float, short string, etc) elif is_editable_type(value) and not object_explorer: if readonly: self.sig_editor_shown.emit() return None else: editor = QLineEdit(parent=parent) editor.setFont(get_font(font_size_delta=DEFAULT_SMALL_DELTA)) editor.setAlignment(Qt.AlignLeft) # This is making Spyder crash because the QLineEdit that it's # been modified is removed and a new one is created after # evaluation. So the object on which this method is trying to # act doesn't exist anymore. # editor.returnPressed.connect(self.commitAndCloseEditor) self.sig_editor_shown.emit() return editor # ObjectExplorer for an arbitrary Python object else: show_callable_attributes = index.model().show_callable_attributes show_special_attributes = index.model().show_special_attributes dataframe_format = index.model().dataframe_format if show_callable_attributes is None: show_callable_attributes = False if show_special_attributes is None: show_special_attributes = False from spyder.plugins.variableexplorer.widgets.objectexplorer \ import ObjectExplorer editor = ObjectExplorer( value, name=key, parent=parent, show_callable_attributes=show_callable_attributes, show_special_attributes=show_special_attributes, dataframe_format=dataframe_format, readonly=readonly) editor.sig_option_changed.connect(self.change_option) self.create_dialog( editor, dict(model=index.model(), editor=editor, key=key, readonly=readonly)) return None