U 祡c{6@stddlmZmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZddlmZddlmZdd lmZmZmZmZdd lmZdd lmZdd l m!Z!dd l"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddl)m*Z*e rddl+m,Z,GdddeZ-e de#fZ.e de-fZ/Gddde0Z1Gddde1Z2GdddZ3GdddeZ4Gddde4Z5Gdd d e4Z6e&Gd!ddZ7e8d"krpdd#l9mZeZe7Z:e:;e7d$d%d&e7dd'd(e7d)d*d+e:d'd8e?e:d9S):)ABCabstractmethod)islice) itemgetter)RLock) TYPE_CHECKINGDictIterableList NamedTupleOptionalSequenceTupleUnion) ratio_resolve)Align)ConsoleConsoleOptionsRenderableType RenderResult)ReprHighlighter)Panel)Pretty)Region)Result rich_repr)Segment) StyleTypeTreec@s*eZdZUdZeed<eeeed<dS) LayoutRenderzAn individual layout render.regionrenderN)__name__ __module__ __qualname____doc__r__annotations__r rr)r)//tmp/pip-unpacked-wheel-qkhzhrww/rich/layout.pyr! s r!Layoutc@seZdZdZdS) LayoutErrorzLayout related error.Nr$r%r&r'r)r)r)r*r,+sr,c@seZdZdZdS) NoSplitterz"Requested splitter does not exist.Nr-r)r)r)r*r./sr.c@s<eZdZdZeZd deddddZee e dd d Z dS) _Placeholderz4An internal renderable used as a Layout placeholder.r+N)layoutstylereturncCs||_||_dSN)r1r2)selfr1r2r)r)r*__init__8sz_Placeholder.__init__consoleoptionsr3ccsx|j}|jp|jj}|j}|jr8|jd|d|dnd|d|d}ttjt|dd|j | |d|dVdS) Nz (z x )(middle)ZverticalZblue)r2titleZ border_styleheight) max_widthr>sizer1namerrcenterrr2 highlighter)r5r8r9widthr>r1r=r)r)r*__rich_console__<sz_Placeholder.__rich_console__)r0) r$r%r&r'rrCrr6rrrrEr)r)r)r*r/3sr/c@sVeZdZUdZdZeed<eedddZee de e e de fdd d Z d S) SplitterzBase class for a splitter.r0rAr3cCsdS)z(Get the icon (emoji) used in layout.treeNr)r5r)r)r* get_tree_iconUszSplitter.get_tree_iconr+childrenr"r3cCsdS)zDivide a region amongst several child layouts. Args: children (Sequence(Layout)): A number of child layouts. region (Region): A rectangular region to divide. Nr))r5rKr"r)r)r*divideYszSplitter.divideN)r$r%r&r'rAstrr(rrIr rr rrLr)r)r)r*rFPs  rFc@sDeZdZdZdZedddZedee e defddd Z d S) RowSplitterz!Split a layout region in to rows.rowrGcCsdS)Nu[layout.tree.row]⬌r)rHr)r)r*rIjszRowSplitter.get_tree_iconr+rJc csV|\}}}}t||}d}t} t||D](\} } | | |||| |fV|| 7}q(dSNrrrzip) r5rKr"xyrDr>Z render_widthsoffset_RegionchildZ child_widthr)r)r*rLms  zRowSplitter.divideN r$r%r&r'rArMrIr rr rrLr)r)r)r*rNesrNc@sDeZdZdZdZedddZedee e defddd Z d S) ColumnSplitterz$Split a layout region in to columns.columnrGcCsdS)Nu[layout.tree.column]⬍r)rHr)r)r*rI~szColumnSplitter.get_tree_iconr+rJc csV|\}}}}t||}d}t} t||D](\} } | | ||||| fV|| 7}q(dSrPrQ) r5rKr"rSrTrDr>Zrender_heightsrUrVrWZ child_heightr)r)r*rLs  zColumnSplitter.divideNrXr)r)r)r*rYysrYc @seZdZdZeedZd8ddddddeeee ee e e e dddd Z e d d d Zeed d dZeedd ddZeed ddZe eddddZe ddddZedd ddZddedefeee fddddZedefdd d!d"Zedefdd d#d$Zedefdd d%d&Zdd d'd(Zedd)d*d+Zd,e dd-d.d/Z e e e!d0d1d2Z"e#e$ed3d4d5Z%e#e$e&d3d6d7Z'dS)9r+aBA renderable to divide a fixed height in to rows or columns. Args: renderable (RenderableType, optional): Renderable content, or None for placeholder. Defaults to None. name (str, optional): Optional identifier for Layout. Defaults to None. size (int, optional): Optional fixed size of layout. Defaults to None. minimum_size (int, optional): Minimum size of layout. Defaults to 1. ratio (int, optional): Optional ratio for flexible layout. Defaults to 1. visible (bool, optional): Visibility of layout. Defaults to True. )rOrZNrT)rAr@ minimum_sizeratiovisible) renderablerAr@r[r\r]r3cCsR|p t||_||_||_||_||_||_|jd|_g|_ i|_ t |_ dS)NrZ) r/ _renderabler@r[r\rAr] splitterssplitter _children _render_mapr_lock)r5r^rAr@r[r\r]r)r)r*r6s zLayout.__init__rGccs<d|jdfVd|jdfVd|jdfVd|jdfVdS)NrAr@r[rr\)rAr@r[r\rHr)r)r* __rich_repr__szLayout.__rich_repr__cCs|jr |S|jS)zLayout renderable.)rbr_rHr)r)r*r^szLayout.renderablecCsdd|jDS)zGets (visible) layout children.cSsg|]}|jr|qSr))r]).0rWr)r)r* sz#Layout.children..rbrHr)r)r*rKszLayout.childrencCs|jS)zGet a map of the last render.)rcrHr)r)r*mapsz Layout.map)rAr3cCs8|j|kr|S|jD]}||}|dk r|SqdS)zGet a named layout, or None if it doesn't exist. Args: name (str): Name of layout. Returns: Optional[Layout]: Layout instance or None if no layout was found. N)rArbget)r5rArWZ named_layoutr)r)r*rjs    z Layout.getcCs$||}|dkr td||S)NzNo layout with name )rjKeyError)r5rAr1r)r)r* __getitem__s zLayout.__getitem__r csddlmddlmddlm}ddfdd |}||d |jjd d }d dd dfdd |||S)z/Get a tree renderable to show layout structure.r)Styled)Tablerr+)r1r3csF|j}jdd}|jr$t|n t|d}||||}|S)N)rrrr)paddingZdim)rarIZgridr]rZadd_row)r1ZicontabletextZ_summary)rmrnr)r*summarys   zLayout.tree..summary layout.tree.T) guide_styleZ highlightr N)treer1r3cs2|jD]&}|j|d|jjd|qdS)Nrs)rt)rbaddrarA)rur1rW)recurserrr)r*rws  zLayout.tree..recurse)Z rich.styledrmZ rich.tablern rich.treer rarA)r5r r1rur))rmrnrwrrr*rus      z Layout.treerZ)ra)layoutsrar3cGsddd|D}z t|tr|n |j||_Wn"tk rPtd|YnX||jdd<dS)zSplit the layout in to multiple sub-layouts. Args: *layouts (Layout): Positional arguments should be (sub) Layout instances. splitter (Union[Splitter, str]): Splitter instance or name of splitter. cSs"g|]}t|tr|nt|qSr) isinstancer+rfr1r)r)r*rgsz Layout.split..zNo splitter called N)r{rFr`rarkr.rb)r5rary_layoutsr)r)r*splits  z Layout.split)ryr3cGsdd|D}|j|dS)zAdd a new layout(s) to existing split. Args: *layouts (Union[Layout, RenderableType]): Positional arguments should be renderables or (sub) Layout instances. css$|]}t|tr|nt|VqdSr4rzr|r)r)r* &sz#Layout.add_split..N)rbextend)r5ryr}r)r)r* add_splitszLayout.add_splitcGs|j|ddidS)zSplit the layout in to a row (layouts side by side). Args: *layouts (Layout): Positional arguments should be (sub) Layout instances. rarONr~r5ryr)r)r* split_row,szLayout.split_rowcGs|j|ddidS)zSplit the layout in to a column (layouts stacked on top of each other). Args: *layouts (Layout): Positional arguments should be (sub) Layout instances. rarZNrrr)r)r* split_column4szLayout.split_columncCs|jdd=dS)zReset splits to initial state.NrhrHr)r)r*unsplit<szLayout.unsplit)r^r3c Cs|j ||_W5QRXdS)zjUpdate renderable. Args: renderable (RenderableType): New renderable object. N)rdr_)r5r^r)r)r*update@sz Layout.updater)r8 layout_namer3c Csl|j\||}|j|\}}|\}}}} |||j|| } t|| |j|<|| ||W5QRXdS)zRefresh a sub-layout. Args: console (Console): Console instance where Layout is to be rendered. layout_name (str): Name of layout. N)rdrc render_linesr9update_dimensionsr!Zupdate_screen_lines) r5r8rr1r"_linesrSrTrDr>linesr)r)r*refresh_screenIs  zLayout.refresh_screen)rDr>r3c Cs|tdd||fg}|j}|j}g}|j}|rl|||d\}} |j} | r*|j| | D] } || q\q*ddt|tddD} | S)z,Create a dict that maps layout on to Region.rcSsi|]\}}||qSr)r)rfr1r"r)r)r* isz+Layout._make_region_map..r)key)rappendpoprKrarLsortedr) r5rDr>stackpushrlayout_regionsZappend_layout_regionr1r"rKZchild_and_region region_mapr)r)r*_make_region_mapZs    zLayout._make_region_mapr7c Csv|j}|jp|j}|||}dd|D}i}|j}|j} |D],\} } || j| | j| j} t| | || <qD|S)zRender the sub_layouts. Args: console (Console): Console instance. options (ConsoleOptions): Console options. Returns: RenderMap: A dict that maps Layout on to a tuple of Region, lines cSsg|]\}}|js||fqSr))rKrr)r)r*rg|sz!Layout.render..) r?r>ritemsrrr^rDr!) r5r8r9Z render_widthZ render_heightrr render_maprrr1r"rr)r)r*r#os     z Layout.renderc cs|j|jp|j}|jp|j}|||||}||_ddt|D}t}| D]>\}} |\} } } } t ||| | | | D]\}}| |qqXt }|D]}|EdH|VqW5QRXdS)NcSsg|]}gqSr)r))rf_r)r)r*rgsz+Layout.__rich_console__..)rdr?rDr>r#rrcrangervaluesrRrrline)r5r8r9rDr>rZ layout_lines_islicer"r_xrTZ _layout_widthZ layout_heightrOrnew_lineZ layout_rowr)r)r*rEs$     zLayout.__rich_console__)N)(r$r%r&r'rNrYr`r rrMintboolr6rrepropertyr^r rK RenderMaprirjrlrurrFr~rrrrrr RegionMaprrrr#rrEr)r)r)r*r+s^  *     __main__)rheader)rAr@main)r\rA footer)r@rAZside)rAbody)rAr\contents2topr<ZbottomZleft1Zleft2ZfooN)@abcrr itertoolsroperatorr threadingrtypingrrr r r r r rrZ_ratiorZalignrr8rrrrrCrZpanelrprettyrr"rreprrrsegmentrr2rrxr r!rr Exceptionr,r.r/rFrNrYr+r$Z rich.consoler1rrrurprintr)r)r)r*s`   ,