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
