B eJd@sddlmZddlmZddlZddlmZmZddlm Z ddl m Z ddl mZedZddd gZd d Zd d ZdddZdddZdddZdddZdS))absolute_import)NumberN) exceptionsoptional_imports)utilsZpandasTaskStartFinishcCs||||g||||gfS)z Returns the corner points of a scatter rectangle :param x0: x-start :param y0: y-lower :param x1: x-end :param y1: y-upper :return: ([x], [y]), tuple of lists containing the x and y values )x0y0x1y1r r N/var/www/html/venv/lib/python3.7/site-packages/plotly/figure_factory/_gantt.py_get_corner_pointss rcCstrt|tjjjrx*tD]"}||krtdd tqWt |j }g}x>t |D]2}i}x|D]}|j ||||<qfW||qXW|St|tstdt |dkrtdt|dtstd|S)z2 Validates the inputted dataframe or list zBThe columns in your dataframe must include the following keys: {0}z, zszgantt..csg|] }|qSr r )r\r])marker_data_dictr rr_s)datalayout)rrrrreversercopydeepcopystrrrsortedgoFigure)r!colorsrU bar_width showgrid_x showgrid_yrVrWtasks task_namesra group_tasksshow_hover_fill show_colorbarrtaskr-scatter_data_templatemarker_data_templatetnZ color_indexgroupIDcolor_idxsysr6rbfigr )r`r^rganttFs                     r|c&s0 | dkr g} | dkrg} | dkr$g} d}x`tt|D]P}t||d||d||dd}d||kr|||d|d<| |q6Wtt|rd }nd }ggd d d|d d}ggdgtd dddd dd d}g}x6tt| D]&}||||kr||||qW|t|d|trt|dkrHtdxt||<||d<||d<t||||d <t |t j}t|ddkrˆ|d|dd|ddt| |d| |d| |d| |d\}}|d|7<|d|7<|krPt||<||d d!<||d<|d| |d|d| |d|d||d|d| |kr|d"| |d|d"| |d| |d=n$|d"d|d"dqW|d#krJd#}xD]} || d'<q4Wt|||| gd(t|| ttt| dt| dgddd)t|dtttd*d+d,d-d.tdd/d0d-d.td1d2d0d-d.tdd3d4d5d.tdd6d4d-d.td7d8gd9d:d;d<}$fd=d>tD} | fd?d>tD7} tj| |$d@}%|%S)Az= Refer to FigureFactory.create_gantt() for docstring NFrr r)r r r#r$r%r#r&r'r(r/)r)r*r+r,r6r-r8r.r0r)r1r2r3)r)r*r+r4r5r#r6r8zYou must use at least 2 colors in 'colors' if you are using a colorscale. However only the first two colors given will be used for the lower and upper bounds on the colormap.r rgY@r7r8r)r9r*r r r5r1r4Td)Z colorscaleZ showscaleZcmaxZcminztError. The number of colors in 'colors' must be no less than the number of unique index values in your group column.r6r:)r;r<r=rr>r?r@rArBrC)rDrErFrGrHrIrJrKrLrMrNrOrP)rF)rQrR)r;r?rSrT)rUr6rVrWrXrYrZr[csg|] }|qSr r )r\r])r^r rr_Msz$gantt_colorscale..csg|] }|qSr r )r\r])r`r rr_Ns)rarb)rrrrsortrrrrrcrclrsZ color_parserZ unlabel_rgbZfind_intermediate_colorZ label_rgbrdrerfrrkeysupdatergrhri)&r!rjrU index_colrrrkrlrmrVrWrnrorarprqr6rrsr-rtru index_valsrowrvrwZlowcolorZ highcolorZintermedZintermed_colorrxryrzr]Zindex_vals_dictZc_indexr rbr{r )r`r^rgantt_colorscales                                            rc s| dkr g} | dkrg} | dkr$g} d}x`tt|D]P}t||d||d||dd}d||kr|||d|d<| |q6Wtt|rd }nd }ggd d |d d}ggdgtd dddd dd}g}x6tt| D]&}||||kr||||qW|x"|D]}||krtdqWxr?r@rArBrC)rDrErFrGrHrIrJrKrLrMrNrOrP)rF)rQrR)r;r?rSrT)rUr6rVrWrXrYrZr[csg|] }|qSr r )r\r])r^r rr_szgantt_dict..csg|] }|qSr r )r\r])r`r rr_s)rarb)rrrrrrrrcrrdrerrfrrgrhri) r!rjrUrrrrkrlrmrVrWrnrorarprqr6rrsr-rtrurrr rvrwrxryrzZ index_valuerbr{r )r`r^r gantt_dictWs                         r Gantt Chart皙?XcCs"t|}|rL||dkr"tdg}x|D]}|||q,Wt|t|trdt |d}n t |d}|dkr| |st|trtdt ||||||| | ddd|||d}|St|tst ||||||||| | ddd||d}|St||||||||| | ddd||d}|SdS) a **deprecated**, use instead :func:`plotly.express.timeline`. Returns figure for a gantt chart :param (array|list) df: input data for gantt chart. Must be either a a dataframe or a list. If dataframe, the columns must include 'Task', 'Start' and 'Finish'. Other columns can be included and used for indexing. If a list, its elements must be dictionaries with the same required column headers: 'Task', 'Start' and 'Finish'. :param (str|list|dict|tuple) colors: either a plotly scale name, an rgb or hex color, a color tuple or a list of colors. An rgb color is of the form 'rgb(x, y, z)' where x, y, z belong to the interval [0, 255] and a color tuple is a tuple of the form (a, b, c) where a, b and c belong to [0, 1]. If colors is a list, it must contain the valid color types aforementioned as its members. If a dictionary, all values of the indexing column must be keys in colors. :param (str|float) index_col: the column header (if df is a data frame) that will function as the indexing column. If df is a list, index_col must be one of the keys in all the items of df. :param (bool) show_colorbar: determines if colorbar will be visible. Only applies if values in the index column are numeric. :param (bool) show_hover_fill: enables/disables the hovertext for the filled area of the chart. :param (bool) reverse_colors: reverses the order of selected colors :param (str) title: the title of the chart :param (float) bar_width: the width of the horizontal bars in the plot :param (bool) showgrid_x: show/hide the x-axis grid :param (bool) showgrid_y: show/hide the y-axis grid :param (float) height: the height of the chart :param (float) width: the width of the chart Example 1: Simple Gantt Chart >>> from plotly.figure_factory import create_gantt >>> # Make data for chart >>> df = [dict(Task="Job A", Start='2009-01-01', Finish='2009-02-30'), ... dict(Task="Job B", Start='2009-03-05', Finish='2009-04-15'), ... dict(Task="Job C", Start='2009-02-20', Finish='2009-05-30')] >>> # Create a figure >>> fig = create_gantt(df) >>> fig.show() Example 2: Index by Column with Numerical Entries >>> from plotly.figure_factory import create_gantt >>> # Make data for chart >>> df = [dict(Task="Job A", Start='2009-01-01', ... Finish='2009-02-30', Complete=10), ... dict(Task="Job B", Start='2009-03-05', ... Finish='2009-04-15', Complete=60), ... dict(Task="Job C", Start='2009-02-20', ... Finish='2009-05-30', Complete=95)] >>> # Create a figure with Plotly colorscale >>> fig = create_gantt(df, colors='Blues', index_col='Complete', ... show_colorbar=True, bar_width=0.5, ... showgrid_x=True, showgrid_y=True) >>> fig.show() Example 3: Index by Column with String Entries >>> from plotly.figure_factory import create_gantt >>> # Make data for chart >>> df = [dict(Task="Job A", Start='2009-01-01', ... Finish='2009-02-30', Resource='Apple'), ... dict(Task="Job B", Start='2009-03-05', ... Finish='2009-04-15', Resource='Grape'), ... dict(Task="Job C", Start='2009-02-20', ... Finish='2009-05-30', Resource='Banana')] >>> # Create a figure with Plotly colorscale >>> fig = create_gantt(df, colors=['rgb(200, 50, 25)', (1, 0, 1), '#6c4774'], ... index_col='Resource', reverse_colors=True, ... show_colorbar=True) >>> fig.show() Example 4: Use a dictionary for colors >>> from plotly.figure_factory import create_gantt >>> # Make data for chart >>> df = [dict(Task="Job A", Start='2009-01-01', ... Finish='2009-02-30', Resource='Apple'), ... dict(Task="Job B", Start='2009-03-05', ... Finish='2009-04-15', Resource='Grape'), ... dict(Task="Job C", Start='2009-02-20', ... Finish='2009-05-30', Resource='Banana')] >>> # Make a dictionary of colors >>> colors = {'Apple': 'rgb(255, 0, 0)', ... 'Grape': 'rgb(170, 14, 200)', ... 'Banana': (1, 1, 0.2)} >>> # Create a figure with Plotly colorscale >>> fig = create_gantt(df, colors=colors, index_col='Resource', ... show_colorbar=True) >>> fig.show() Example 5: Use a pandas dataframe >>> from plotly.figure_factory import create_gantt >>> import pandas as pd >>> # Make data as a dataframe >>> df = pd.DataFrame([['Run', '2010-01-01', '2011-02-02', 10], ... ['Fast', '2011-01-01', '2012-06-05', 55], ... ['Eat', '2012-01-05', '2013-07-05', 94]], ... columns=['Task', 'Start', 'Finish', 'Complete']) >>> # Create a figure with Plotly colorscale >>> fig = create_gantt(df, colors='Blues', index_col='Complete', ... show_colorbar=True, bar_width=0.5, ... showgrid_x=True, showgrid_y=True) >>> fig.show() rzIn order to use an indexing column and assign colors to the values of the index, you must choose an actual column name in the dataframe or key if a list of dictionaries is being used.rgbTzError. You have set colors to a dictionary but have not picked an index. An index is required if you are assigning colors to particular values in a dictionary.N)rnrorarprqrr)rnrorarprq)r"rrrrZvalidate_indexrrrZvalidate_colors_dictZvalidate_colorsrcr|rr)rrjrrrZreverse_colorsrUrkrlrmrVrWrnrorarprqr!Z index_list dictionaryr{r r r create_gantt#s       r)NNNFTT)NNNFT)NNNFT)NNFFrrFFrNNNNFT) __future__rnumbersrrdZplotlyrrZ plotly.colorsrjrZplotly.figure_factoryrZplotly.graph_objectsZ graph_objectsrhZ get_modulerrrr"r|rrrr r r rsZ       0 9 T @