-Ph)hddlmZddlmZddlmZdZdZdZdZ dZ dd Z Gd d e Z d S)) exceptions) graph_objs)utilsz#3D9970z#FF4136c n|||fD]G}tt|D](}||||krtjd)H|||fD]G}tt|D](}||||krtjd)Hd}||vrtjddS)a ohlc and candlestick specific validations Specifically, this checks that the high value is the greatest value and the low value is the lowest value in each unit. See FigureFactory.create_ohlc() or FigureFactory.create_candlestick() for params :raises: (PlotlyError) If the high value is not the greatest value in each unit. :raises: (PlotlyError) If the low value is not the lowest value in each unit. :raises: (PlotlyError) If direction is not 'increasing' or 'decreasing' zOops! Looks like some of your high values are less the corresponding open, low, or close values. Double check that your data is entered in O-H-L-C orderzOops! Looks like some of your low values are greater than the corresponding high, open, or close values. Double check that your data is entered in O-H-L-C order) increasing decreasingbothzBdirection must be defined as 'increasing', 'decreasing', or 'both'N)rangelenr PlotlyError) openhighlowclose directionkwargslstindexdirection_optss [/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/plotly/figure_factory/_ohlc.py validate_ohlcr s  c5!  3t99%%  EE{SZ'' ,2( dE"  3s88__  E5zCJ&& ,2' :N&&$ V   '&c :t|||||\}}}d|vrd} n|ddd} |dttd|d|td d ||d | d |} | S) a Makes increasing ohlc sticks _make_increasing_ohlc() and _make_decreasing_ohlc separate the increasing trace from the decreasing trace so kwargs (such as color) can be passed separately to increasing or decreasing traces when direction is set to 'increasing' or 'decreasing' in FigureFactory.create_candlestick() :param (list) open: opening values :param (list) high: high values :param (list) low: low values :param (list) close: closing values :param (list) dates: list of datetime objects. Default: None :param kwargs: kwargs to be passed to increasing trace via plotly.graph_objs.Scatter. :rtype (trace) ohlc_incr_data: Scatter trace of all increasing ohlc sticks. nameT IncreasingFlinecolorwidthtextscatterlines)typexymode showlegend)_OHLC get_increase setdefaultdict_DEFAULT_INCREASING_COLOR) r rrrdatesrflat_increase_xflat_increase_y text_increaser( ohlc_incrs rmake_increasing_ohlcr4:s*9> dC99lnn6_o} &,///  fd)B!LLLMMM fm,,,       I rc Rt|||||\}}}|dttd|d||dd|ddtd d ||d d |} | S) a Makes decreasing ohlc sticks :param (list) open: opening values :param (list) high: high values :param (list) low: low values :param (list) close: closing values :param (list) dates: list of datetime objects. Default: None :param kwargs: kwargs to be passed to increasing trace via plotly.graph_objs.Scatter. :rtype (trace) ohlc_decr_data: Scatter trace of all decreasing ohlc sticks. rrrr!r(Fr Decreasingr"r#)r$r%r&r'r))r* get_decreaser,r-_DEFAULT_DECREASING_COLOR) r rrrr/rflat_decrease_xflat_decrease_y text_decrease ohlc_decrs rmake_decreasing_ohlcr=gs9> dC99lnn6_o} fd)B!LLLMMM fm,,, lE*** fl+++ /_7NTI rNr c |tj|||||ntj||||t|||||fi||dkrt|||||fi|}|g}nA|dkrt |||||fi|} | g}n&t|||||fi|}t |||||fi|} || g}t jtdd} t j|| S) aU **deprecated**, use instead the plotly.graph_objects trace :class:`plotly.graph_objects.Ohlc` :param (list) open: opening values :param (list) high: high values :param (list) low: low values :param (list) close: closing :param (list) dates: list of datetime objects. Default: None :param (string) direction: direction can be 'increasing', 'decreasing', or 'both'. When the direction is 'increasing', the returned figure consists of all units where the close value is greater than the corresponding open value, and when the direction is 'decreasing', the returned figure consists of all units where the close value is less than or equal to the corresponding open value. When the direction is 'both', both increasing and decreasing units are returned. Default: 'both' :param kwargs: kwargs passed through plotly.graph_objs.Scatter. These kwargs describe other attributes about the ohlc Scatter trace such as the color or the legend name. For more information on valid kwargs call help(plotly.graph_objs.Scatter) :rtype (dict): returns a representation of an ohlc chart figure. Example 1: Simple OHLC chart from a Pandas DataFrame >>> from plotly.figure_factory import create_ohlc >>> from datetime import datetime >>> import pandas as pd >>> df = pd.read_csv('https://raw.githubusercontent.com/plotly/datasets/master/finance-charts-apple.csv') >>> fig = create_ohlc(df['AAPL.Open'], df['AAPL.High'], df['AAPL.Low'], df['AAPL.Close'], dates=df.index) >>> fig.show() NrrF)zerolineclosest)xaxis hovermode)datalayout) rvalidate_equal_lengthrr4r=rLayoutr-Figure) r rrrr/rrr3rCr<rDs r create_ohlcrHs8F  #D$UEBBBB #D$U;;;$c5)>>v>>>L  (tS%QQ&QQ { l " "(tS%QQ&QQ {(tS%QQ&QQ (tS%QQ&QQ 9%  T5%9%9%9Y O O OF  $v 6 6 66rc0eZdZdZdZdZdZdZdZdS)r*zF Refer to FigureFactory.create_ohlc_increase() for docstring. c ||_||_||_||_dgt |z|_||_g|_g|_g|_ g|_ g|_ g|_ | |dSN)r rrrr emptyr/all_xall_y increase_x increase_y decrease_x decrease_y get_all_xyseparate_increase_decrease)selfr rrrr/rs r__init__z_OHLC.__init__s   Vc$ii'      '')))))rc tt|j|j|j|j|j|j|j|_|jg}tt|jdz D]3}| |j|dz|j|z 4t|dz fd|jD|_ dSdtt|jD|_ dS)a Zip data to create OHLC shape OHLC shape: low to high vertical bar with horizontal branches for open and close values. If dates were added, the smallest date difference is calculated and multiplied by .2 to get the length of the open and close branches. If no date data was provided, the x-axis is a list of integers and the length of the open and close branches is .2. Nrc .g|]}|z |||||zdgSrKr)).0r% date_dif_mins r z$_OHLC.get_all_xy..sB\!1aAq.s<9:S!Q1a#gt4r)listzipr rrrrLrNr/r r appendminrM)rUdate_difir[s @rrSz_OHLC.get_all_xys              : !H3tz??Q.// C C 1q5 1DJqM ABBBBMMQ.LDJJJ >CC NN>S>SDJJJrctt|jD]}|j||j||j|krK|j|j||j|j|w|j |j||j |j|dS)z Separate data into two groups: increase and decrease (1) Increase, where close > open and (2) Decrease, where close <= open N) r r r rrOr`rMrPrNrQrR)rUrs rrTz _OHLC.separate_increase_decreases3ty>>** : :Ez% (E"TYu%555&&tz%'8999&&tz%'89999&&tz%'8999&&tz%'89999 : :rctj|j}tj|j}dt |jz}|||fS)a Flatten increase data and get increase text :rtype (list, list, list): flat_increase_x: x-values for the increasing trace, flat_increase_y: y=values for the increasing trace and text_increase: hovertext for the increasing trace OpenrgHighLowCloserj)rflattenrOrPr )rUr0r1r2s rr+z_OHLC.get_increase L -88-88M   >>rctj|j}tj|j}dt |jz}|||fS)a Flatten decrease data and get decrease text :rtype (list, list, list): flat_decrease_x: x-values for the decreasing trace, flat_decrease_y: y=values for the decreasing trace and text_decrease: hovertext for the decreasing trace rf)rrlrQrRr )rUr9r:r;s rr7z_OHLC.get_decreasermrN) __name__ __module__ __qualname____doc__rVrSrTr+r7r)rrr*r*sj***""""H:::"??? ?????rr*)Nr )plotlyrplotly.graph_objsrplotly.figure_factoryrr.r8rr4r=rHobjectr*r)rrrws((((((''''''&%, , , ^***Z<67676767ri?i?i?i?i?Fi?i?i?i?i?r