U SZfч@sddlmZddlZddlmZmZddlmZddl m Z ddl m Z edZdddgZd d Zd d ZdddZdddZdddZdddZdS))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 @/tmp/pip-unpacked-wheel-5ksk5baj/plotly/figure_factory/_gantt.py_get_corner_pointss rcCstrt|tjjjrtD]"}||krtdd tqt |j }g}t |D].}i}|D]}|j ||||<q^||qR|St|tstdt |dkrtdt|dtstd|S)z2 Validates the inputted dataframe or list zBThe columns in your dataframe must include the following keys: {0}z, zZ rangeselectortypetitler7heightwidthZshapesZ hovermodeZyaxisZxaxiscsg|] }|qSr r .0kscatter_data_dictr r szgantt..csg|] }|qSr r rYmarker_data_dictr rr^sdatalayout)rrrrreversercopydeepcopystrrrsortedgoFigure)r colorsrV bar_width showgrid_x showgrid_yrWrXtasks task_namesrb group_tasksshow_hover_fill show_colorbarrtaskr-scatter_data_templatemarker_data_templatetnZ color_indexgroupIDcolor_idxsysr7rcfigr r`r]rganttDs                            "r~c&s | dkr g} | dkrg} | dkr$g} d}tt|D]P}t||d||d||dd}d||krz||d|d<| |q4tt|rd }nd }ggd d d|d d}ggdgtd dddd dd d}g}tt| D]&}||||kr||||q|t|d|trt|dkr@tdtt| D],}| |d }| rl|| krL| |qL| r| tt| D]}| |d }| |d =|}| r| |}||| |d<||| |d<t |t j }|d}|d}|||d}t |||}t |t j}|| |d<| |d}|kr^t||<||d<t||||d <||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"dq|d#krtd} | d td|dgd|dggd#d$dd%t|d|tr"g}tt| D]*}||||kr||||q|t|t|krtd&i}!d}"|D]0}#|"t|dkr$d}"||"|!|#<|"d7}"q tt| D],}| |d }| rh|| krH| |qH| r| tt| D]h}| |d }| |d =|}| r| |}||| |d<||| |d<|!|||| |d<| |d}|krt||<||d<||d<t||||d <t |t j}t|ddkr|d|dd|ddt| |d| |d| |d| |d\}}|d|7<|d|7<|kr.t||<||d d!<||d<|d| |d|d| |d|d||d|d| |krԈ|d"| |d|d"| |d| |d=n$|d"d|d"dq|d#kr"d#}D]} || d'<qt|||| 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)A= Refer to FigureFactory.create_gantt() for docstring NFrrrr"r$r%r#r&r'r(r/)r)r*r+r,r7r-r9r.r0rr1)r)r*r+r5r6r#r7r9zYou 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 r gY@r8r9r)r:r*r r r6r2r5Td)Z colorscaleZ showscaleZcmaxZcminztError. The number of colors in 'colors' must be no less than the number of unique index values in your group column.r7r;r<r?r@rArBrCrGrHrIrJrKrLrMrNrOrPrQrRrSrUcsg|] }|qSr r rYr\r rr^Ksz$gantt_colorscale..csg|] }|qSr r rYr_r rr^Lsra)rrrrsortrrrrrdrclrsZ color_parserZ unlabel_rgbZfind_intermediate_colorZ label_rgbrerfrgrrkeysupdaterhrirj)&r rkrV index_colrsrlrmrnrWrXrorprbrqrrr7rrtr-rurv index_valsrowrwrxZlowcolorZ highcolorZintermedZintermed_colorryrzr{r[Zindex_vals_dictZc_indexrrcr|r r}rgantt_colorscales                                                 "rc s| dkr g} | dkrg} | dkr$g} d}tt|D]P}t||d||d||dd}d||krz||d|d<| |q4tt|rd }nd }ggd d |d d}ggdgtd dddd dd}g}tt| D]&}||||kr||||q||D]}||krtdqtt| D],}| |d }| r\|| kr<| |q<| rx| tt| D]B}| |d }| |d =|}| r| |}||| |d<||| |d<||||| |d<| |d}|krt ||<||d<||d<t|ddkrn|d|dd|ddt | |d| |d| |d| |d\}}|d|7<|d|7<|krt ||<||dd<||d<|d| |d|d| |d|d||d|d| |kr|d | |d|d | |d| |d=n$|d d|d dq|d!krd!}|D]}t |||d <qt|||| gd"t|| t tt| dt| dgddd#t|dtt td$d%d&d'd(tdd)d*d'd(td+d,d*d'd(tdd-d.d/d(tdd0d.d'd(td1d2gd3d4d5d6}fd7d8tD} | fd9d8tD7} tj| |d:}|S);rNFrrrr"r$r%r#r&r'r(r/)r)r*r+r,r-r9r.r0rr1r4zdIf you are using colors as a dictionary, all of its keys must be all the values in the index column.r r r8r9r)r:r*r r r6r2r5Tr;r<r?r@rArBrCrGrHrIrJrKrLrMrNrOrPrQrRrSrUcsg|] }|qSr r rYr\r rr^szgantt_dict..csg|] }|qSr r rYr_r rr^sra)rrrrrrrrdrrerfrrgrrhrirj) r rkrVrrsrlrmrnrWrXrorprbrqrrr7rrtr-rurvrrrrwrxryrzr{Z index_valuercr|r r}r gantt_dictUs                            "r Gantt Chart皙?XcCst|}|rH||dkr"tdg}|D]}|||q*t|t|tr`t |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.ZrgbTzError. 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)rorprbrqrrrs)rorprbrqrr)r!rrrrZvalidate_indexrrrZvalidate_colors_dictZvalidate_colorsrdr~rr)rrkrrsZreverse_colorsrVrlrmrnrWrXrorprbrqrrr Z index_list dictionaryr|r r r create_gantt!s      r)NNNFTT)NNNFT)NNNFT)NNFFrrFFrNNNNFT)ZnumbersrreZplotlyrrZ plotly.colorsrkrZplotly.figure_factoryrZplotly.graph_objectsZ graph_objectsriZ get_modulerrrr!r~rrrr r r rs`      0 G c O