JedZddlZddlmZddlmZddlmZmZddl m Z m Z m Z m Z mZmZmZddlmZmZddlmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"dd l#m$Z%m&Z'dd l(m)Z)dd l*m+Z+dd l,m-Z-dd l.m/Z/ddl0m1Z1m2Z2m3Z3ddl4m5Z5ddl6m7Z7m8Z8m9Z9m:Z:ddl;mZ>e?e@e%je%jfZCeDe%je%jfZGgdZHdZIdZJdZKdZLdZMdZNdZOdZPdZQdZRdZSejZUdZVd ZWd!ZXd"ZYd#ZZGd$d%e Z[Gd&d'ee)Z\Gd(d)e Z]Gd*d+e Z^Gd,d-e>e)Z_d1d.Z`d/Zaebd0k(reayy)2ae Pandas DataFrame Editor Dialog. DataFrameModel is based on the class ArrayModel from array editor and the class DataFrameModel from the pandas project. Present in pandas.sandbox.qtpandas in v0.13.1. DataFrameHeaderModel and DataFrameLevelModel are based on the classes Header4ExtModel and Level4ExtModel from the gtabview project. DataFrameModel is based on the classes ExtDataModel and ExtFrameModel, and DataFrameEditor is based on gtExtTableView from the same project. DataFrameModel originally based on pandas/sandbox/qtpandas.py of the `pandas project `_. The current version is qtpandas/models/DataFrameModel.py of the `QtPandas project `_. Components of gtabview from gtabview/viewer.py and gtabview/models.py of the `gtabview project `_. N) perf_counter)parse) from_qvariant to_qvariant)QAbstractTableModelQEventQItemSelectionModel QModelIndexQtSignalSlot)QColorQCursor) QApplication QCheckBox QGridLayout QHBoxLayout QInputDialog QLineEditQMenu QMessageBox QPushButton QTableView QScrollBar QTableWidgetQFrame QItemDelegate)numpypandas)SpyderConfigurationAccessor)_)DEFAULT_SMALL_DELTA)get_font)is_text_stringis_type_text_stringto_text_string)ima) add_actions create_action keybinding qapplication) get_idx_rect) BaseDialog)falsef0z0.z0.0 z%.6ggAgj@<i(gQ?gQ?gffffff??g333333?g?g?g333333?c6|jtvrd}|S)zl Used to convert bool entrance to false. Needed since any string in bool('') will return True. )lower _bool_false)values Y/usr/lib/python3/dist-packages/spyder/plugins/variableexplorer/widgets/dataframeeditor.pybool_false_checkr;]s  {{} # Lcr|Dcgc]}|| }}t|\}}t|t|fScc}w)z'Returns the global maximum and minimum.)zipmaxmin)col_valsindexxcol_vals_without_Nonemax_colmin_cols r: global_maxrGhsC(0B1AMQBB12GW w<W %%Cs44c:eZdZdZedfdZdZdZdZe dZ e dZ e d Z dd Z d Zd Zd ZdZdZdZdZdZej.fdZdZej4fdZdZej:dfdZdZe fdZ!ddZ"e fdZ#dZ$y)DataFrameModelz DataFrame Table Model. Partly based in ExtDataModel and ExtFrameModel classes of the gtabview project. For more information please see: https://github.com/wavexx/gtabview/blob/master/gtabview/models.py Nctj|||_||_d|_d|_||_d|_g|_|jjd|_ |jjd|_ |j|jz}d|_ |tkr0|jd|_d|_|j#dnd|_d|_|j#d|tkDrt$|_t(|_y|jt,kDr t$|_n|j|_|jt.kDr t(|_y|j|_y)NrTF)r__init__dialogdfdf_columns_list df_index_list_formatcomplex_intrandisplay_error_idxsshape total_rows total_cols max_min_col LARGE_SIZEmax_min_col_updatecolum_avg_enabledbgcolor_enabled colum_avg ROWS_TO_LOAD rows_loaded COLS_TO_LOAD cols_loaded LARGE_NROWS LARGE_COLS)self dataFrameformatparentsizes r:rLzDataFrameModel.__init__zs-$$T* #! ""$''--*''--*0 *   # # %%)D "#'D NN1 %*D "#(D NN1  * +D +D ,#/ #'?? +#/ #'?? r<cd|dk(r|jjS|jjS)z Return the corresponding labels taking into account the axis. The axis could be horizontal (0) or vertical (1). r)rNcolumnsrBrcaxiss r:_axiszDataFrameModel._axiss% #'!)tww>>r<c|dk(rA|j)|jjj|_|jS|j)|jj j|_|jS)z Return the corresponding labels as a list taking into account the axis. The axis could be horizontal (0) or vertical (1). r)rOrNritolistrPrBrjs r: _axis_listzDataFrameModel._axis_listsm 19##+'+ww'='='?$'' '!!)%)WW]]%9%9%;"%% %r<cj|j|}t|dsdSt|jS)z Return the number of levels in the labels taking into account the axis. Get the number of levels for the columns (0) or rows (1). levelsrK)rlhasattrlenrq)rcrkaxs r: _axis_levelszDataFrameModel._axis_levelss. ZZ H-qA3ryy>Ar<c.|jjS)z"Return the shape of the dataframe.)rNrTrcs r:rTzDataFrameModel.shapesww}}r<cF|jd|jdfS)zz*DataFrameModel.colum_avg.. s huor<N)rZ return_maxrGrrs r:r\zDataFrameModel.colum_avgs.!&  ! !EDO(DO r<c|j}|jsy|j|j|}|j|t j |rHtt}t|r|jt|S|jt|St|trt}nt }|j#|j|\}} ||z dk(rd}n||z }t&t(|||z z|z z} t!t| } | dkDrd} tj*| t,t.t0}|S#t$$rYywxYw)z$Background color depending on value.Nrr4rK)columnr[ get_valuerowrWpdisnarBACKGROUND_NONNUMBER_COLORr$ setAlphaFBACKGROUND_STRING_ALPHABACKGROUND_MISC_ALPHA isinstancerrfloatrrBACKGROUND_NUMBER_MINHUEBACKGROUND_NUMBER_HUERANGEfromHsvFBACKGROUND_NUMBER_SATURATIONBACKGROUND_NUMBER_VALUEBACKGROUND_NUMBER_ALPHA) rcrBrr9color color_funcrrvmax_vmin_diffhues r: get_bgcolorzDataFrameModel.get_bgcolor%sO## uyy{F3   F # +rwwu~56Ee$ 78< 9 568 5%!56 " )9)96BJD$  $;!#%(N%)D[N,.H:e,,/.1?/AACC/CQwOOC)E$;$;=E   s'E E E cP |jj||f}|S#tjjj $rA|jj dd|fjtj|}Y|S|jj ||f}Y|SxYw)z"Return the value of the DataFrame.N) rNiatr_libstslibOutOfBoundsDatetimeilocastypestr)rcrrr9s r:rzDataFrameModel.get_valuePs .GGKKV ,E  xx~~11 AGGLLF+2237;;C@E  .GGLLf-E sA$B%B%c<|js tS|tjk(s|tjk(r}|j }|j }|j||}t|tr t|j|zSt|r|S tt|S|tj&k(rt|j)|S|tj*k(rtt-t.S|tj0k(r||j"vr t3dStS#ttf$rtt|zcYSwxYw#t $r|j"j%|YywxYw)z Cell contentzDisplay Error!)font_size_deltazWIt is not possible to display this value because an error ocurred while trying to do it)isValidrr DisplayRoleEditRolerrrrrrQ ValueErrorrDEFAULT_FORMATr%r& ExceptionrSrBackgroundColorRolerFontRoler#r" ToolTipRoler!)rcrBrolerrr9s r:datazDataFrameModel.data\sb}}= 2>> !TR[[%8\\^F))+CNN3/E%'?&t||e';<< %U+  -&~e'<==R++ +t//67 7 R[[ x8KLM M R^^ #///BCC}1#I.?'~'=>>?!-++2259,-s$E *E4 !E10E14$FFcV|jjj|_y)zRecalcuate index information.N)rNrBrnrPrws r:recalculate_indexz DataFrameModel.recalculate_indexs!WW]]113r<c |jK|jjdj|r"tj|j ddy |t jk(}|dk\r8 |jj|jj||dd n-|j!|jj#d| |j)y#t$r9|jj|jj||dd YRt$r7}tj|j dd t|zYd}~d}~wt$r7}tj|j dd t|zYd}~d}~wwxYw#t$$r7}tj|j ddt'|zYd}~yd}~wwxYw)zOverriding sort methodNr)rkErrorzDTypeError error: no ordering relation is defined for complex numbersFT mergesort)by ascendinginplacekind)rirrrzValueError: %szSystemError: %s)rrzTypeError error: %s)rRanyrrcriticalrMr AscendingOrderrN sort_valuesriAttributeErrorsortrr& SystemErrorr sort_indexrrr)rcrorderres r:rzDataFrameModel.sorts    *""&&A&.33F;$$T[['&OP !2!22I{PGG''477??6+B2;T-8(:(&&(""49"E 5&3GGLL)@+4d&1!3"O ((g)9N1)EA)NP !s+A%F7*G'7*G$#G$'I 6A II c|jS)z Return data)rNrws r:get_datazDataFrameModel.get_datas wwr<c |j|jkr |jS|jS#t$rYywxYw)zDataFrame row numberr)rUr^rrs r:rowCountzDataFrameModel.rowCountsA  $"2"22&'''  s$3 3 ??c|r|j|jkDr|j|jz }t|t}|j t |j|j|zdz |xj|z c_|j |r|j|jkDr|j|jz }t|t}|jt |j|j|zdz |xj|z c_|jyyy)zGet more columns and/or rows.rKN) rUr^r@r]beginInsertRowsr endInsertRowsrVr`r_beginInsertColumnsendInsertColumnsrcrowsrireminderitems_to_fetchs r: fetch_morezDataFrameModel.fetch_mores DOOd&6&66)9)99H <8N  0@0@!%!1!1N!BQ!F H    .     t)9)99)9)99H <8N  # #KM43C3C$($4$4~$E$I K    .   ! ! # :7r<c t|jjdk(ry|j|jkr |jS|jS#t $rYywxYw)DataFrame column numberrKr)rsrNrTrVr`rrs r: columnCountzDataFrameModel.columnCountsY 477==!Q&D$4$44&'''  s"A$A A A"!A"cD|j|jyr)beginResetModel endResetModelrws r:rzDataFrameModel.resets  r<)rFF)%r __module__ __qualname____doc__rrLrlrorupropertyrTryr{rrrYrrrr\rrr rrrrrrrrrr rrrrrr<r:rIrIos*8&3P? &B<<"" '$-L   )V  "~~#J4"$!2!2)V4 *,$%N)] $"!,  r<rIceZdZdZeZeZeZdZdZ dZ dZ d dZ dZ dZd Zd Zed Zy ) DataFrameViewz Data Frame view class. Signals ------- sig_sort_by_column(): Raised after more columns are fetched. sig_fetch_more_rows(): Raised after more rows are fetched. variable_explorerctj|||j||j||j ||j tj |jtj dg|_||_ |jjj|j|j|_|j|j d||j#j$j|j&|j)j$j|j*y)z Constructor.Ncopy)rrLsetModelsetHorizontalScrollBarsetVerticalScrollBarsetHorizontalScrollModeScrollPerPixelsetVerticalScrollModesort_old header_classsectionClickedconnect sortByColumn setup_menumenuconfig_shortcutrhorizontalScrollBar valueChanged_load_more_columnsverticalScrollBar_load_more_rows)rcrfmodelrhscrollvscrolls r:rLzDataFrameView.__init__"sD&) e ##G, !!'* $$Z%>%>? "":#<#<= " ((001B1BCOO%  TYY5   "//77  # # %  --55d6J6JKr<cJ |j|dy#t$rYywxYw)zLoad more columns to display.TriNload_more_data NameErrorrcr9s r:r$z DataFrameView._load_more_columns4s,    t  4     ""cJ |j|dy#t$rYywxYw)zLoad more rows to display.TrNr,r/s r:r&zDataFrameView._load_more_rows=s,    D  1   r0c |r[||jjk(r:|jj||jj |r]||j jk(r;|jj||jj yyy#t$rYywxYw)z&Load more rows and columns to display.r2r+N) r%maximumr'rsig_fetch_more_rowsrr"sig_fetch_more_columnsr.)rcr9rris r:r-zDataFrameView.load_more_dataFs !7!7!9!A!A!CC ''T'2((--/5D$<$<$>$F$F$HH '''8++002Iw   sB:B?? C  C c|jdgk(r|jjd|jj}|j j ||skt |jdk7r|jjdy|jj|jd|jdy||jjg|_|jjy)zImplement a column sort.NTrFrrK) rrsetSortIndicatorShownsortIndicatorOrderr'rrssetSortIndicatorsig_sort_by_columnrrcrB sort_orders r:rzDataFrameView.sortByColumnUs ==TF "    3 3D 9&&99; zz|   34==!Q&!!77> !!224==3C37==3CE  1 1 D D FG  $$&r<cv|jj|j|jy)zReimplement Qt method.N)r popup globalPosaccept)rcevents r:contextMenuEventzDataFrameView.contextMenuEventds" )* r<cttdtdtjdj t j}tdtftdtftdtftdtftdtff}|g}|D],\}fd }|t||t j gz }.t}t|||S) zSetup context menu.Copyeditcopy)shortcuticon triggeredcontextzTo boolz To complexzTo intzTo floatzTo strc(jyrr)funcrcsr:slotz&DataFrameView.setup_menu..slotus  &r<)rIrJ)r)r!r*r'rHrr WidgetShortcutrcomplexrrr&rr()rc copy_action functions types_in_menurrNr rMs` @r:rzDataFrameView.setup_menuis#D!F)-7-?),*)=.2ii,.,=,= ?  lD)AlOW+Ek3'!J-)?k>24 % # HJD$ ' mD$6:464E4EGH HM H T{D-( r<c|j}|j}|Dcgc]}|j|d|c}ycc}w)z'A function that changes types of cells.r6rLN)r'selectedIndexesr)rcrMr' index_listis r:rzDataFrameView.change_type~s; ))+ 9CDAq"$ /DDsAcB|jsyt|j\}}}}dx}}|jj}|jt ||dzt ||dzf}t j} |j| d||| j} | jtj } | j#| y#t$r,tj|tdtdYzwxYw)zCopy text to clipboardNTrK )seprBrrzText can't be copied.)rUr,r'rNrsliceioStringIOto_csvUnicodeEncodeErrorrrr!getvaluecloser clipboardsetText) rcrow_minrow_maxcol_mincol_maxrBrrNobjoutputcontentsrbs r:rzDataFrameView.copys##% )$*>*>*@A ' ' ZZ\__ggeGWq[1GWq[123 , JJv4uVJ D??$  **, (#" ,  ' )* , ,sC))2DDNr )rr r r r r;r6r5 CONF_SECTIONrLr$r&r-rrCrrr rrr<r:rrsf #X (&LL$  ' *E  V$ $r<rcteZdZdZdZfdZd dZefdZd dZ e jfdZ dZ d ZxZS) DataFrameHeaderModelz This class is the model for the header or index of the DataFrameEditor. Taken from gtabview project (Header4ExtModel). For more information please see: https://github.com/wavexx/gtabview/blob/master/gtabview/viewer.py crtt| ||_||_||_|jj d|_|jj d|_|j|jz}|tkDrt|_ t|_ n`|jtkDr t|_ n|j|_ |jtkDr t|_ n|j|_ |jdk(rV|jj d|_|jj d|jj df|_y|jj d|_|jj d|jj df|_y)a Header constructor. The 'model' is the QAbstractTableModel of the dataframe, the 'axis' is to acknowledge if is for the header (horizontal - 0) or for the index (vertical - 1) and the palette is the set of colors to use. rrKN)superrmrLr'rk_paletterTrUrVrXr]r^r_r`rbrary_shape)rcr'rkpalettergrs r:rLzDataFrameHeaderModel.__init__sK "D24   ****1-****1-0 * +D +D +#/ #'?? ,#/ #'?? 99>"jj..q1DO::2215tzz7G7G7JKDK"jj..q1DO::++A. 0G0G0JKDKr<c|jdk(rtd|jdS|j|jkr |jS|jS)z!Get number of rows in the header.rrK)rkr?rrrUr^rs r:rzDataFrameHeaderModel.rowCountsJ 99>q$++a.) )$"2"22&'''r<c|jdk(r1|j|jkr |jS|jStd|jdS)rrrK)rkrVr`r?rrrs r:rz DataFrameHeaderModel.columnCountsJ 99>$"2"22&'''q$++a.) )r<c|jdk(r|j|jkDr|j|jz }t|t}|j t |j|j|zdz |xj|z c_|j|jdk(r|j|jkDr|j|jz }t|t}|jt |j|j|zdz |xj|z c_ |jyyy)z)Get more columns or rows (based on axis).rKrN) rkrUr^r@r]rr rrVr`r_rrrs r:rzDataFrameHeaderModel.fetch_mores IINt1A1AA)9)99H <8N  0@0@!%!1!1N!BQ!F H    .     99>doo0@0@@)9)99H <8N  # #KM43C3C$($4$4~$E$I K    .   ! ! # A>r<cx|tjk(}|jj|j|y)zOverriding sort method.)rT)r rr'r COLUMN_INDEX)rcrrrs r:rzDataFrameHeaderModel.sorts/R...  ));r<c>|tjk(rM|tjk(rtjSt tj tj zS|tjk7r|tjk7ry|jdk(r|jddkry|tjk(rdnd}|jj|dkDr|}|S|jj|j|}t|s t|}|S)z)Get the information to put in the header.NrKr)r TextAlignmentRole Horizontal AlignCenterr AlignRight AlignVCenterrrrkrrr'ryrr%r&)rcsection orientationr orient_axisrs r: headerDatazDataFrameHeaderModel.headerDatas 2'' 'bmm+~~%2==2??:;; 2>> !dbnn&< 99>dkk!n1&"--7aQ :: " "; /! 3F ZZ&&tyy':F 'v.'/ r<c2|jr@|j|jdk\s |j|jdk\ry|jdk(r |j|jfn|j|jf\}}|t j k7ry|jdk(r|jddkry|jj|j||}t|s t|}|S)z^ Get the data for the header. This is used when a header has levels. rrKN) rrrrrrkr rr'rr%r&)rcrBrrrrs r:rzDataFrameHeaderModel.datas }} 99;$++a. ( <<>T[[^ +59YY!^UYY[%,,.1,,.%))+6 S 2>> ! 99>dkk!n1""499c37 #6*#F+F r<rr )rr r r rxrLrr rrr rrrr __classcell__rs@r:rmrmsFL$LL(!, *$""$!2!2 4r<rmc>eZdZdZfdZddZddZdZdZxZ S)DataFrameLevelModela Data Frame level class. This class is used to represent index levels in the DataFrameEditor. When using MultiIndex, this model creates labels for the index/header as Index i for each section in the index/header Based on the gtabview project (Level4ExtModel). For more information please see: https://github.com/wavexx/gtabview/blob/master/gtabview/viewer.py cLtt| ||_|j j |_|j jdkDr|j|_ n|j|_ ||_ |jd||_ y)NT)rprrLr'darkr _background lightnesstext _foregroundhighlightedTextrqsetBold_font)rcr'rsfontrs r:rLzDataFrameLevelModel.__init__As !413 "<<>//1    % % '# -&||~D &668D   T r<cHtd|jjdS)z5Get number of rows (number of levels for the header).rKrr?r'ryrs r:rzDataFrameLevelModel.rowCountM1djj--a011r<cHtd|jjdS)z;Get the number of columns (number of levels for the index).rKrrs r:rzDataFrameLevelModel.columnCountQrr<c|tjk(rM|tjk(rtjSt tj tj zS|tjk7r|tjk7ry|jjddkrV|tjk(rC|jjd|r|jjd|StdS|jjddkry|jjddkr|tjk(rytddzt|zS)z Get the text to put in the header of the levels of the indexes. By default it returns 'Index i', where i is the section in the index NrrKIndexr1)r rzr{r|rr}r~rrr'ryrr!Verticalr&)rcrrrs r:rzDataFrameLevelModel.headerDataUs 2'' 'bmm+~~%2==2??:;; 2>> !dbnn&< :: " "1 % *{bmm/Kzzq)zzq11W:  ZZ $ $Q '1 , ZZ $ $Q '1 , 1KzC."999r<c|jsy|tjk(r |jSd}|j |j j ddz k(r4t|j jd|j}n`|j|j j ddz k(r3t|j jd|j }|tjk(r|r|S|tjk(r |jS|tjk(r |jS|tjk(r|jj!Sy)z"Get the information of the levels.Nr6rKr)rr rrrr'ryrrrrForegroundRolerBackgroundRolerrqwindow)rcrBrlabels r:rzDataFrameLevelModel.datals}} 2;; ::  <<>TZZ44Q7!; ; 599;78E YY[DJJ33A6: : 5<<>:;E 2>> !eL R&& &## # R&& &## # R&& &=='') )r<r) rr r r rLrrrrrrs@r:rr4s!  22:.r<rc.eZdZdZdZd$fd Zd%dZeeedZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZd&dZdZd$dZdZdZdZdZdZdZdZ edZ!dZ"dZ#d Z$d!Z%d"Z&ed#Z'xZ(S)'DataFrameEditorz Dialog for displaying and editing DataFrame and related objects. Based on the gtabview project (ExtTableView). For more information please see: https://github.com/wavexx/gtabview/blob/master/gtabview/viewer.py rc~t|||jtjd|_d|_y)NF)rprL setAttributer WA_DeleteOnClose is_serieslayout)rcrfrs r:rLzDataFrameEditor.__init__s4  "--. r<cX d|_d|_t|_|jj d|jj dddd|j |j|r&t|d|jjzz}n"td|jjz}t|tjrd|_|j}n/t|tj rtj"|}|j%|t't(j*|_t't(j.|_|j3|j5|j7t9|||_|j:j<j?|j@|jC|jjE|j,d dd d |jjE|j0dd d d tG|_$d |_%|jLjO||jQjS}|d z|_*|d z|_+|j |j|jYt(jZt]}|j dt_td}|jE||j`j?|jbt_td}|jE||j`j?|jdtgtd}|ji|j:jj|jm|j:jj|jnj?|jp|jE|tgtd|_9|jrji|j:jt|jrjm|j xr|j:jj|jrjnj?|j:jv|jE|jr|jyt_td|_=|jzj}d|jzj`j?|j~|jE|jzt_td|_@|jjd|jjd|jj`j?|j|jE|j|j dddd|jj|ddd d |j|j:|jd|jdz}|j:j|y)z Setup DataFrameEditor: return False if data is not supported, True otherwise. Supported types for data are DataFrame, Series and Index. FNrz - %sz %s editorTrfrrKi, @FormatResizezBackground colorzColumn min/maxzSave and CloseClose%dataframe_format)I_selection_rec_modelrr setSpacingsetContentsMargins setLayoutr&rrr!rrSeriesrto_framer DataFramesetWindowTitlerr r{r(rr)create_table_levelcreate_table_headercreate_table_indexrI dataModelrrsave_and_close_enablecreate_data_table addWidgetset_autosized_cols_max_autosize_ms dataTableinstallEventFilter fontMetricsaverageCharWidth min_trunc max_widthsetWindowFlagsWindowrrclicked change_formatresize_to_contentsr setCheckedr[ setEnabled stateChangedchange_bgcolor_enablebgcolor_globalrZr\ addStretchbtn_save_and_close setDisabledrA btn_closesetAutoDefault setDefaultreject addLayoutrresizeColumnsToContentsget_confr) rcrtitle avg_width btn_layout btn_format btn_resizerres r:setup_and_checkzDataFrameEditor.setup_and_checks $ !m  q! &&r2r15 t{{# "5)Gdnn6M6M,MMEkNT^^%<%<>99:4>>99:$$T%?%?@W%'*:(;< &&t~~'G'GH &&4>>'9(F'+~~'E'E G ((001I1IJT001"-a0@.A"B ++D1 ''// <T445$QwZ0 %%d+ !!$' &&t{{3T^^,%%aQ3 j!Q15 dnn% $$&t}}%788 !!&)r<c|jjd|jjd|jjdy)zAHandle the data change event to enable the save and close button.TN)rrrr)rctop_left bottom_rights r:rz%DataFrameEditor.save_and_close_enables@ **40 ..t4 **40r<ct|_|jjtj|jj t j|jjt j|jjtj|jjjj|j|jj!jj|j"|jj%t'|j(j+|jdd|jj-dddd|jjj.j|j0y)z5Create the QTableView that will hold the level model.rN)r table_levelsetEditTriggersrNoEditTriggerssetHorizontalScrollBarPolicyr ScrollBarAlwaysOffsetVerticalScrollBarPolicy setFrameStylerPlainhorizontalHeadersectionResizedr_index_resizedverticalHeader_header_resizedsetItemDelegaterrrrr sortByIndexrws r:rz"DataFrameEditor.create_table_level sE%< (()D)DE 55b6K6KL 33B4I4IJ &&v||4 ))+::BB8<8K8K M '')88@@8<8L8L N ((9 d..15 ++Aq!Q7 ))+::BB<@ '' 5 **,;;CC8<8L8L N ))-/: d//A6r<ct|_|jjj|jj t j |jjtj|jjtj|jjtj|jj|j|jjt j"|jj%j&j)|j*|jj-t/|j0j3|jdd|jj5ddddy)z5Create the QTableView that will hold the index model.rKrN)r table_indexrrrrrrr rrrrrr)rrrrrr _row_resizedrrrrrrws r:rz"DataFrameEditor.create_table_index)s8%< ))+002 (()D)DE 55b6K6KL 33B4I4IJ ..z/H/HI --dll; &&v||4 '')88@@<@y)z4Create the QTableView that will hold the data model.rKN) rrrrr(r)rrrrr rrrrrrrrrrrrr setFocusProxyr;r _sort_updater6_fetch_more_columnsr5_fetch_more_rowsrws r:rz!DataFrameEditor.create_data_table9sz&tT^^'+'8'8'I'I'K'+||T\\C %%',,. '')..0 33B4I4IJ 11"2G2GH ..z/H/HI ,,Z-F-FG $$V\\2 &&}7 dnna3 4>>* ))11$2C2CD --55d6N6NO **2243H3HIr<c|jjjd|jjj}|jj j |||jy)zImplement a Index sort.TN)rrr8r9rr'rrr<s r:rzDataFrameEditor.sortByIndexLsg ))+AA$G%%668KKM   %%eZ8 r<c|jS)zGet the model of the dataframe.)rrws r:r'zDataFrameEditor.modelSs {{r<c\|jj|||jy)zUpdate the column width.N)rsetColumnWidth_update_layoutrcr old_width new_widths r:rzDataFrameEditor._column_resizedWs" %%c95 r<c\|jj|||jy)zUpdate the row height.N)r setRowHeightr rcr old_height new_heights r:rzDataFrameEditor._row_resized\s" ##C4 r<c\|jj|||jy)z>Resize the corresponding column of the index section selected.N)rrr r s r:rzDataFrameEditor._index_resizedas$ ''Y7 r<c\|jj|||jy)z))::<CCEFJ 1}  --a6!!..q$7 ((4 '' 3;;++A.2 a<((779??AI((??I((44X>?((779??ABI &&y1 &&y1 ,,.r<cN|j}|j||r~yy)z!Set the model in the given table.N)selectionModelr)rctabler' old_sel_models r: _reset_modelzDataFrameEditor._reset_models',,.  u  r<c||_y)z4Set maximum time to calculate size hint for columns.N)r)rclimit_mss r:setAutosizeLimitTimez$DataFrameEditor.setAutosizeLimitTimes (r<c ||_|jj}|jj |j |j |jt||j|j|j |jt|d|j|j |jt|d|j|r|jyy)z9Set the model for the data, header/index and level views.rrKN)rrr&currentColumnChangedr_resizeCurrentColumnToContentsr)rrrsrrrmrr )rcr'relayout sel_models r:rzDataFrameEditor.setModels NN113 &&..33 5 $**,?@D @D -M N $++-AWNN >  HC  ( (sJ G Hr<c||jk(r1|jtjk(r|j y)z+Override eventFilter to catch resize event.F)rrrrr)rcrhrBs r: eventFilterzDataFrameEditor.eventFilters. $.. UZZ\V]]%B  0 0 2r<c|jjjj}|jj |x}}|j j d}|jj |jjjj}|dk(r|n|dz}|jd}n|jtd||z z }||krd}||jvrE|jj|d}|j|j|j|||dz }|r{|jjjj}|jj |}|dk(r|n|dz}||jtd||z z }||kryy)z(Resize the columns that are in the view.rKrnNFT)rrecttopLeftrCcolumnAtrrT bottomRightrr?raddrCr)rc index_columnstartrrendrEresizeds r:rz/DataFrameEditor._resizeVisibleColumnsToContentss~~**,44688: nn--l;; !!!$nn%%dnn&9&9&;&G&G&I&K&K&MNbyecAg  (J..Qe 1DDJCiG$...$$((-,,T->->-0*> 1HC#~~224@@BDDF nn--l;"byecAg)!%!6!6Qe 9L!LJCir<c|j|jvr,|j|jj |yy)z*Resize the current column to its contents.N)rrrrscrollTo)rc new_index old_indexs r:r/z.DataFrameEditor._resizeCurrentColumnToContentss<    T%9%9 9  0 0 2 NN # #I . :r<ct|_|j|j|j|j |j y)z#Resize the columns to its contents.N)rrrFrrrr rws r:rz'DataFrameEditor.resizeColumnsToContentss@"u %%d&6&6&*&6&68M8M O r<c|jj||jj|j xr|dkDy)zV This is implementet so column min/max is only active when bgcolor is rN)rrrrrrs r:rz%DataFrameEditor.change_bgcolor_enable s: u% &&4>>'9'GeaiHr<cRtj|tdtdtj|j j \}}|rt|} |dz|jds;tdj|}tj|td|y|j j||d d}|jd |yy#tdj|}tj|td|YyxYw) zD Ask user for display format for floats and use it. rzFloat formattingg?zFormat ({}) is incorrectrNrz!Format ({}) should start with '%'rKr)rgetTextr!rNormalrrrrerr startswithrset_conf)rcrevalidmsgs r:rzDataFrameEditor.change_formats %,,T1X;-./A-B-6-=-=-1^^-F-F-HJ  [F  $$S);<CCFK$$T1W:s; NN % %f -ABZF MM,f 5  23::6B$$T1W:s;s #C((>771a4= Ir<c|jj|jjj}t d|D]}}||kru|j j |}|jj |}||kDr|jj||`|j j||}yy)z&Update the column width of the header.rN)rrr'rr7rrr)rc column_countrB column_width header_widths r:_update_header_sizez#DataFrameEditor._update_header_size8s 113((..0<<> 1l+ E|##~~99%@ #00< ..0 !!$!/ ..0   "**,r<r)r6)T))rr r r rkrLrr r rrrrrrr'rrrrr r)r,rr3r?rCrFrHrr/rrrrrerrrrrrs@r:rrs'L iV +{#1$1 N" 78 J&    /B)"20 " *H M6/I V6 62 ./. V- -r<rct|}|j||r |j|jSddl}|j dy)zTest subroutiner)rrNrK)rrexec_rsysexit)rrrfdlgros r: test_editrrfsB  (C 4u- }}  r<c ddlm}ttjtdk\r ddlm}m}nddlm}m}t}tjddgddgd d gd d gd dgtjjdddgddgddggdd|||dddg|dg}t|}|||tjddg|dgd}t|j d}|||tjtjjdd}t|}|||tjtj"dd}t|}|||y )!zDataFrame editor testr)nanz2.0.0)assert_frame_equalassert_series_equalTry??rPteststringg(\?rrKrz Unkown typez Large valueduáéíunicodeabczTest global maxdType)rBri)rBri )rN)rrtrr __version__pandas.testingrurvpandas.util.testingr+rrrandomrandrrrrarange)rtrurvappdf1outresultseriess r:rwrwrsN R^^g.JJO .C ,, 6N 9  X  7O J YY^^Aq != 1 C y ! Cc3->Df  C C.CsC YYf~c6] EF CHHQK C$ ,,ryy~~fb1 2C C.CsC YYryy}1 -F F C$r<__main__)r6N)cr r\timerpackaging.versionr qtpy.compatrr qtpy.QtCorerrr r r r r qtpy.QtGuirrqtpy.QtWidgetsrrrrrrrrrrrrrr spyder_kernels.utils.lazymodulesrrrrspyder.api.config.mixinsr spyder.config.baser!spyder.config.fontsr"spyder.config.guir#spyder.py3compatr$r%r&spyder.utils.icon_managerr'spyder.utils.qthelpersr(r)r*r+3spyder.plugins.variableexplorer.widgets.arrayeditorr,2spyder.plugins.variableexplorer.widgets.basedialogr-rrint64int32rrP complex64 complex128rr8rrXrarbr]r_rrrrr lightGrayrBACKGROUND_INDEX_ALPHArrr;rGrIrrmrrrrrwrrr<r:rs, $2   'GA 3&..)>>LIC2884r}}=3       !"\\&a(aH K$J ;K$\P.PfK-K\^-j"=^-H &%R zFr<