o d@sddlmZddlmZddlZddlmZmZddlm Z ddl m Z ddl mZedZgdZdd Zd d Z    dddZ    dddZ    dddZ          dddZdS))absolute_import)NumberN) exceptionsoptional_imports)utilspandas)TaskStartFinishcCs||||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 a/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/plotly/figure_factory/_gantt.py_get_corner_pointss rcCstrCt|tjjjrCtD]}||vrtdd tq t |j }g}t |D]}i}|D] }|j ||||<q/||q)|St|tsMtdt |dkrXtdt|dtsdtd|S)z2 Validates the inputted dataframe or list zBThe columns in your dataframe must include the following keys: {0}z, zheightwidthshapes hovermodeyaxisxaxiscg|]}|qSr r .0kscatter_data_dictr r zgantt..crjr r rkmarker_data_dictr rrprqdatalayout)rrr!rreversercopydeepcopystrrr sortedgoFigure)r%colorsrc bar_width showgrid_x showgrid_yrdretasks task_namesru group_tasksshow_hover_fill show_colorbarrtaskr3scatter_data_templatemarker_data_templatetn color_indexgroupIDcolor_idxsysr>rvfigr rsrorganttFs                           "rc&s| durg} | dur g} | durg} d}tt|D](}t||d||d||dd}d||vr=||d|d<| |qtt|rNd }nd }ggd d d|d d}ggdgtd dddd dd d}g}tt| D]}||||vr||||qr|t|d|tr>t|dkrtdtt| D]}| |d }| r|| vr| |q| r| tt| D]Q}| |d }| |d =|}| r| |}||| |d<||| |d<t |t j }|d}|d}|||d}t |||}t |t j}|| |d<| |d}|vr)t||<||d<t||||d <||d<t |t j}t|ddkrk|d|dd|ddt| |d| |d| |d| |d\}}|d|7<|d|7<|vrt||<||d d!<||d<|d| |d|d| |d|d||d|d| |vr|d"| |d|d"| |d| |d=qň|d"d|d"dq|d#ur>td} | d td|dgd|dggd#d$dd%t|d|tr g}tt| D]}||||vrd||||qP|t|t|krxtd&i}!d}"|D]}#|"t|dkrd}"||"|!|#<|"d7}"q~tt| D]}| |d }| r|| vr| |q| r| tt| D]5}| |d }| |d =|}| r| |}||| |d<||| |d<|!|||| |d<| |d}|vrt||<||d<||d<t||||d <t |t j}t|ddkrI|d|dd|ddt| |d| |d| |d| |d\}}|d|7<|d|7<|vrt||<||d d!<||d<|d| |d|d| |d|d||d|d| |vr|d"| |d|d"| |d| |d=q|d"d|d"dq|d#ur 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 NFr r rr(r*r+r)r,r-r.r5)r/r0r1r2r>r3r@r4r6rr7)r/r0r1r<r=r)r>r@zYou 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@r?r@r/rAr0r rr=r8r<Td) colorscale showscalecmaxcminztError. The number of colors in 'colors' must be no less than the number of unique index values in your group column.r>rBrCrIrJrKrLrMrRrSrTrUrVrWrXrYrZr[r\r^r_rbcrjr r rkrnr rrpMrqz$gantt_colorscale..crjr r rkrrr rrpNrqrt)rrr!rsortrrrrrwrclrs color_parser unlabel_rgbfind_intermediate_color label_rgbrxryrzrr keysupdater{r|r})&r%r~rc index_colrrrrrdrerrrurrr>rrr3rr index_valsrowrrlowcolor highcolorintermedintermed_colorrrrrmindex_vals_dictc_indexr#rvrr rrgantt_colorscales                                                    "rc s| durg} | dur g} | durg} d}tt|D](}t||d||d||dd}d||vr=||d|d<| |qtt|rNd }nd }ggd d |d d}ggdgtd dddd dd}g}tt| D]}||||vr||||qp||D] }||vrtdqtt| D]}| |d }| r|| vr| |q| r| tt| D]}| |d }| |d =|}| r| |}||| |d<||| |d<||||| |d<| |d}|vrt ||<||d<||d<t|ddkr0|d|dd|ddt | |d| |d| |d| |d\}}|d|7<|d|7<|vrwt ||<||dd<||d<|d| |d|d| |d|d||d|d| |vrʈ|d | |d|d | |d| |d=q|d d|d dq|d!urd!}|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);rNFr r rr(r*r+r)r,r-r.r5)r/r0r1r2r3r@r4r6rr7r;zdIf you are using colors as a dictionary, all of its keys must be all the values in the index column.r rr?r@r/rAr0r rr=r8r<TrBrCrIrJrKrLrMrRrSrTrUrVrWrXrYrZr[r\r^r_rbcrjr r rkrnr rrprqzgantt_dict..crjr r rkrrr rrprqrt)rrr!rrrrrwrrxryrrzr r{r|r}) r%r~rcrrrrrrdrerrrurrr>rrr3rrrrr#rrrrr index_valuervrr rr gantt_dictWs                            "r Gantt Chart皙?XcCst|}|r$||dvrtdg}|D] }|||qt|t|tr0t |d}nt |d}|dur>| |s^t|trJtdt ||||||| | ddd|||d}|St|tsxt ||||||||| | ddd||d}|St||||||||| | ddd||d}|S) 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)rrrurrr)rrrurr)r'rrrrvalidate_indexrr!rvalidate_colors_dictvalidate_colorsrwrrr)r"r~rrreverse_colorsrcrrrrdrerrrurrr% index_list dictionaryrr r r create_gantt#s      r)NNNFTT)NNNFT)NNFFrrFFrNNNNFT) __future__rnumbersrrxplotlyrr plotly.colorsr~rplotly.figure_factoryrplotly.graph_objects graph_objectsr| get_modulerrrr'rrrrr r r rsd     0 G c O