Yc@`s2dZddlmZmZmZddlmZmZddlmZddl Z ddl Z ddl m Z m Z ddlmZddlZddlmZddlZdd lmZdd lmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'dd l(m)Z)d Z*edefde!fde$fdefde fdefdefde%fdefdefde#fde"fdefg Z+dZ,e'dZ-dZ.dZ/dZ0e1d Z2d!Z3d"Z4d#d$Z5ie4d%6e5d&6Z6d'Z7d(Z8d)d*Z9d+e:fd,YZ;d-Z<d.Z=dd/d0d1d2Z?d3Z@d4ZAdd/d0d5ZBd6ZCd7d8d9d:ZDd;ZEd<ZFd=ZGddd>ZHdddd?ZIeJd@ZKdAZLdBdCdDZMdEZNdS(Fsn This is the utils module that collects convenience functions and code that are useful for charts ecosystem. i(tabsolute_importtdivisiontprint_function(t OrderedDictt defaultdict(tcopyN(tcostsin(t hsv_to_rgb(tjson_normalize(t iteritems( tAsterisktCirclet CircleCrosstCircleXtCrosstDiamondt DiamondCrosstInvertedTriangletSquaret SquareCrosstSquareXtTriangletX(tDEFAULT_PALETTE(tColumnDataSourcetabcdefghijklmnopqrstuvwxyztcircletsquarettriangletdiamondtinverted_triangletasterisktcrosstxt circle_crosstcircle_xtsquare_xt square_crosst diamond_crosscC`stj||S(s/Return first n items of the iterable as a list.(t itertoolstislice(tntiterable((s-lib/python2.7/site-packages/bkcharts/utils.pyttake1scC`sIg}tj|}x-tt|D]}|jt|q(W|S(s Build a color list just cycling through a given palette. Args: chuck (seq): the chunk of elements to generate the color list palette (seq[color]) : a palette of colors to cycle through Returns: colors (R(tcycletrangetlentappendtnext(tchunktpalettetcolorstgti((s-lib/python2.7/site-packages/bkcharts/utils.pyt cycle_colors6s cC`s]d}g}xAt|||D]-\}}}|j||||dq"Wt|S(sTranslate polar coordinates to cartesian. Args: r (float): radial coordinate start_angles (list(float)): list of start angles end_angles (list(float)): list of end_angles angles Returns: x, y points cS`s|t||t|fS(N(RR(trtalpha((s-lib/python2.7/site-packages/bkcharts/utils.pytUsi(tzipR0(R8t start_anglest end_anglest cartesiantpointststarttend((s-lib/python2.7/site-packages/bkcharts/utils.pytpolar_to_cartesianJs ""cC`sGi}g}x4|D],}||krd||<|j|qqW|S(sCreates an ordered list from strings, tuples or other hashable items. Returns: list of unique and ordered values i(R0(R+tmmaptord_settitem((s-lib/python2.7/site-packages/bkcharts/utils.pyt ordered_set]s   cK`sTdt|D}ttjjg|jD]}|j^q/}t|S(sCollect list of unique and ordered columns across attribute specifications. Args: specs (dict): attribute name, :class:`AttrSpec` mapping Returns: list of columns in order as they appear in attr spec and without duplicates cS`s(i|]\}}|jr||qS((tcolumns(t.0t spec_nametspec((s-lib/python2.7/site-packages/bkcharts/utils.pys zs (R tlistR(tchaint from_iterabletvaluesRGRF(tspecstselected_specsRJt spec_cols((s-lib/python2.7/site-packages/bkcharts/utils.pytcollect_attribute_columnsos %cK`sd}t|tr<t|}tj|}WdQXnt|trZt|}nMt|trx;t |D]*\}}t|trvt|}qvqvWn|r|dk rt |}n|S(sAttempt to produce :class:`pandas.DataFrame` from hierarchical json-like data. This utility wraps the :func:`pandas.io.json.json_normalize` function and by default will try to rename the columns produced by it. Args: data (str or list(dict) or dict(list(dict))): a path to json data or loaded json data. This function will look into the data and try to parse it correctly based on common structures of json data. rename (bool, optional: try to rename column hierarchy to the base name. So medals.bronze would end up being bronze. This will only rename to the base column name if the name is unique, and only if the pandas json parser produced columns that have a '.' in the column name. **kwargs: any kwarg supported by :func:`pandas.io.json.json_normalize` Returns: a parsed pandas dataframe from the json data, unless the path does not exist, the input data is nether a list or dict. In that case, it will return `None`. N( tNonet isinstancetstrtopentjsontloadRKR tdictR tdenormalize_column_names(tdatatrenametkwargstparsedt data_filetktv((s-lib/python2.7/site-packages/bkcharts/utils.pyt df_from_jsonscC`s|jj}tt}x;|D]3}d|kr"||j|jddq"q"Wi}x=t|D]/\}}t|dkrl|d||Adds a ChartHelp object to the help attribute of the function.c`st|_|S(N(Rthelp(tf(R(s-lib/python2.7/site-packages/bkcharts/utils.pytadd_help>s((RR((Rs-lib/python2.7/site-packages/bkcharts/utils.pyR<scC`st|dks$|dks$|dkr3d}d}n7|dkrjt|tr[|d}n|}d}n||fS(s:Produces consistent aggregation spec from optional column specification. This utility provides some consistency to the flexible inputs that can be provided to charts, such as not specifying dimensions to aggregate on, not specifying an aggregation, and/or not specifying a column to aggregate on. RlitcountN(RSRTRK(tdim_colstagg_coltagg((s-lib/python2.7/site-packages/bkcharts/utils.pytderive_aggregationEs$    tmeang?g{Gz?cC`st|||||}t|||d|d|dd|dsgw@(tformatRR(thuet saturationRot golden_ratio((s-lib/python2.7/site-packages/bkcharts/utils.pytcolor_in_equal_spaceus   cC`s|jjj}t|dtrQg|D]}|djdd^q,}n"t|dtrm|}ng}xN|D]F}||krqznd|krqzn|jj|j||qzW|S(sg Args: renderer (GlyphRenderer): renderer for the glyph to be modified. tooltips (bool, list(str), list(tuple)): valid tooltips string as defined in the builder class. group (DataGroup): group of data containing missing columns. Returns: renderer (GlyphRenderer): renderer with missing columns added iiRRwt$( t data_sourceR[RfRTRtreplaceRUtaddt get_values(trendererRtgrouptcurrent_columnstpairttooltips_columnstcolumn((s-lib/python2.7/site-packages/bkcharts/utils.pytadd_tooltips_columnss ,     (ORt __future__RRRt collectionsRRRR(RWtmathRRtcolorsysRtpandasRtpandas.io.jsonR tnumpyRtsixR tbokeh.models.glyphsR R R RRRRRRRRRRtbokeh.plotting.helpersRtbokeh.models.sourcesRRxt marker_typesR,R7RBRFRRtTrueRbRZRmRqtspecial_columnsRvR~RtobjectRRRRSRRRRRRRRRRRtFalseRRRR(((s-lib/python2.7/site-packages/bkcharts/utils.pyts~    X                  (            K