
    q-Ph/7                    F   U d dl mZ d dlmZmZmZmZ d dlmZ d dl	m
Z
mZmZmZmZmZmZmZmZ erd dlZd dlZd dlmZmZmZmZ d dlmZ d dlmZmZ d d	lmZ d d
l m!Z!m"Z"m#Z#m$Z$ d dl%m&Z&m'Z'm(Z(m)Z) d dl*m+Z, d dl*m-Z. d dl*m/Z0 d dl*m1Z1 d dl2m3Z3 d dl4m5Z5  ej6        e7          5  d dl8m9Z9 ddd           n# 1 swxY w Y   ej:        dk    rd dl	m;Z; nd dl<m;Z;  G d de          Z= G d de          Z>ed         Z?de@d<   eeAd         df         ZBde@d<   eeAd         df         ZCde@d<   ee?ee?         f         ZDde@d <   eeAeE         eAeF         eAeG         eAeH         eAd!         eAd"         eAd#         eAd$         eAeIe                  eAeJed%f                  eAeK         eAeL         eAd&         eAd         f         ZMde@d'<   eeeHee?eMdf         f         eeeHeJeHee?eMdf         f         f                  f         ZNde@d(<   eeHe?f         ZOde@d)<   eeEeFd&f         ZPde@d*<   ed+         ZQde@d,<   eePeQeHeGeKf         ZRde@d-<   eeReIe         f         ZSde@d.<   ed/d0eHf         ZTde@d1<   eeSeTdf         ZUde@d2<   ed3         ZVde@d4<   d5ZWde@d6<   eeHeWf         ZXde@d7<   ed8         ZYde@d9<   ed:         ZZde@d;<   ed<         Z[de@d=<   ed>         Z\de@d?<   ed@         Z]de@dA<   edB         Z^de@dC<   edD         Z_de@dE<   edF         Z`de@dG<   edH         Zade@dI<   edJ         Zbde@dK<   edL         Zcde@dM<   edN         Zdde@dO<   edP         Zede@dQ<   edR         Zfde@dS<   edT         Zgde@dU<   edV         Zhde@dW<   edX         Zide@dY<   edZ         Zjde@d[<   ed\         Zkde@d]<   ed^         Zlde@d_<   ed`         Zmde@da<   edb         Znde@dc<   edd         Zode@de<   edf         Zpde@dg<   edh         Zqde@di<   edj         Zrde@dk<   edl         Zsde@dm<   edn         Ztde@do<   edp         Zude@dq<   edr         Zvde@ds<   edt         Zwde@du<   edv         Zxde@dw<   edx         Zyde@dy<   edz         Zzde@d{<   ed|         Z{de@d}<   ed~         Z|de@d<   ed         Z}ed         Z~de@d<   ed         Zde@d<   ed         Zde@d<   ed         Zde@d<   ed         Zed         Zde@d<   ed         Zde@d<   ed         Zde@d<   ed         Zde@d<   ed         Zde@d<   ed         Zde@d<   ed         Zde@d<   eeeHeeeL         eeHeeL         f         d0f         f         ee         ddddddf         Zde@d<   eeeXeJeXd%f         f         eeHeeHeHf         f         f         Zde@d<   eeeXeeH         f         eeHeeeHef         eeeHeeHef         f                  f         f         f         Zde@d<   eeeeXeJeX         f         eHf         eeH         eGf         Zde@d<   eeeXeeJeHd%f         eEf         f         eEf         Zde@d<   eeeHeeHeeH         f         f         eeH         eGf         Zde@d<   ed         Zde@d<    edddd0d/          Z eddd          ZeJeEeEeEf         Zde@d<   ed         Zde@d<    G d de          Z G d de          Z G d de          Z G d de          Zed         Zde@d<   eeeeef         Zde@d<   eEZde@d<   eeeeeE         d0df         Zde@d<   eHZde@d<   eeeeH         d0df         Zde@d<   eeeG         d0df         Zde@d<   eeef         Zde@d<   eeeef         Zde@d<   eed         df         Zde@d<   ed         Zde@d<   eeHee
eK         eKeIeH         eIe         eIe
eK                  eIeK         f         Zde@d<   eeegeKf         eegeHf         f         Zed         Zde@d<    G dń dƦ          Zg dǢZ G dȄ dɦ          ZeegeeHeHf         f         Zde@d<   eeeHeHf         ef         Zde@d<   dS )    )annotations)
CollectionIterableMappingSequence)Path)	IOTYPE_CHECKINGAnyCallableLiteralProtocol	TypedDictTypeVarUnionN)datedatetimetime	timedelta)Decimal)
ConnectionEngine)Session)	DataFrameExpr	LazyFrameSeries)DataTypeDataTypeClassIntegerTypeTemporalType)numpy)pandas)pyarrow)torch)	GPUEngine)_selector_proxy_)PyPartitioning)   
   )	TypeAliasc                      e Zd ZdZ	 dd	dZdS )
ArrowArrayExportablezGType protocol for Arrow C Data Interface via Arrow PyCapsule Interface.Nrequested_schemaobject | Nonereturntuple[object, object]c                    d S N selfr.   s     N/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/_typing.py__arrow_c_array__z&ArrowArrayExportable.__arrow_c_array__/   s     #    r3   )r.   r/   r0   r1   )__name__
__module____qualname____doc__r8   r4   r9   r7   r-   r-   ,   s8        QQ 15$ $ $ $ $ $ $r9   r-   c                      e Zd ZdZdd	dZdS )
ArrowStreamExportablezIType protocol for Arrow C Stream Interface via Arrow PyCapsule Interface.Nr.   r/   r0   objectc                    d S r3   r4   r5   s     r7   __arrow_c_stream__z(ArrowStreamExportable.__arrow_c_stream__7         r9   r3   )r.   r/   r0   r@   )r:   r;   r<   r=   rB   r4   r9   r7   r?   r?   4   s%        SSWWWWWWWr9   r?   )r   r   r+   PolarsDataTyper!   PolarsTemporalTyper    PolarsIntegerTypeOneOrMoreDataTypesr   r   r   r   .r   PythonDataTypeSchemaDefinition
SchemaDictNumericLiteral)r   r   r   r   TemporalLiteralNonNestedLiteralPythonLiteralr   r   IntoExprColumnIntoExpr)eqneqgtltgt_eqlt_eqComparisonOperatorr'   SelectorTypeColumnNameOrSelector)earliestlatestraisenull	Ambiguous)uncompressedsnappydeflateAvroCompression)	necessaryalwaysnon_numericneverCsvQuoteStyle)physicallexicalCategoricalOrdering)utf8z
utf8-lossyCsvEncoding)forwardbackwardminmaxmeanzerooneFillNullStrategy)fullmixedFloatFmt)cfortran
IndexOrder)r_   lz4zstdIpcCompression)zm:mzm:1z1:mz1:1JoinValidation)leftright	datapointLabel)noner   r   
left_right
right_leftMaintainOrderJoin)r\   r]   NonExistent)ignoredropNullBehavior)autocolumns
row_groupsprefilteredr   ParallelStrategy)r{   r_   r`   gziplzobrotlir|   ParquetCompression)ro   rp   firstlastsumrq   medianlenPivotAgg)nearesthigherlowermidpointlinearequiprobableQuantileMethod)averagero   rp   denseordinalrandom
RankMethod)r\   rm   rn   Roll)half_to_evenhalf_away_from_zero	RoundMode)binaryjsonSerializationFormat)littlebig
Endianness)
bkbmbgbtbbytes	kilobytes	megabytes	gigabytes	terabytesSizeUnit)	windowr   mondaytuesday	wednesdaythursdayfridaysaturdaysundayStartBy)dataallSyncOnCloseMethod)nsusmsTimeUnit)NFCNFKCNFDNFKDUnicodeForm)r   r   anyr   UniqueKeepStrategy)vertical
horizontalUnstackDirection)thread_local	threadingMapElementsStrategy)rn   rm   r   AsofJoinStrategy)r   r   bothr   ClosedInterval)r   r   InterpolationMethod)innerr   r   ru   semianticrossouterJoinStrategy)first_non_null	max_widthListToStructWidthStrategy)
r   vertical_relaxeddiagonaldiagonal_relaxedr   align
align_fullalign_inner
align_leftalign_right)pearsonspearmanCorrelationMethod)adbc
connectorxDbReadEngine)
sqlalchemyr   DbWriteEngine)replaceappendfailDbWriteMode)r   r   r   sd)arraydictJaxExportType)colrowOrientation)r   r   r   SearchSortedSide)tensordatasetr   TorchExportType)hexbase64TransferEncoding)group_to_rowsjoinexplodeWindowMappingStrategy)plaintreeExplainFormatznp.ndarray[Any, Any]zpa.Tablezpd.DataFrameztorch.TensorFrameInitTypesColumnFormatDictConditionalFormatDictColumnTotalsDefinitionColumnWidthsDefinitionRowTotalsDefinition)fastbalanced	expensiveParametricProfileNames
PolarsTyper   r   	FrameType
BufferInfo)calamineopenpyxlxlsx2csvExcelSpreadsheetEnginec                  2    e Zd ZU dZded<   ded<   ded<   dS )SeriesBufferszUnderlying buffers of a Series.r   valueszSeries | NonevalidityoffsetsN)r:   r;   r<   r=   __annotations__r4   r9   r7   r!  r!     s9         ))NNNr9   r!  c                      e Zd ZddZdS )BasicConnectionargsr   kwargsr0   c                    dS )zReturn a cursor object.Nr4   r6   r(  r)  s      r7   cursorzBasicConnection.cursor
  rC   r9   Nr(  r   r)  r   r0   r   )r:   r;   r<   r,  r4   r9   r7   r'  r'  	  s(        & & & & & &r9   r'  c                      e Zd ZddZdS )BasicCursorr(  r   r)  r0   c                    dS )zExecute a query.Nr4   r+  s      r7   executezBasicCursor.execute  rC   r9   Nr-  )r:   r;   r<   r1  r4   r9   r7   r/  r/    s(             r9   r/  c                      e Zd ZddZddZdS )	Cursorr(  r   r)  r0   c                    dS )zFetch all results.Nr4   r+  s      r7   fetchallzCursor.fetchall  rC   r9   c                    dS )zFetch results in batches.Nr4   r+  s      r7   	fetchmanyzCursor.fetchmany  rC   r9   Nr-  )r:   r;   r<   r5  r7  r4   r9   r7   r3  r3    s<        ! ! ! !( ( ( ( ( (r9   r3  )r   r   r   AlchemyConnectionConnectionOrCursorSingleIndexSelectorMultiIndexSelectorSingleNameSelectorMultiNameSelectorBooleanMaskSingleColSelectorMultiColSelector)r   z	in-memory	streaminggpur&   
EngineType)irrh   	PlanStage
FileSource)functionrenamed_parameterstreaming_parameternonkeyword_argumentsparameter_as_multi_positionalDeprecationTypec                  .    e Zd Zd	dZed
d            ZdS )PartitioningSchemepy_partitioningr(   r0   Nonec                    || _         d S r3   )_py_partitioning)r6   rO  s     r7   __init__zPartitioningScheme.__init__V  s     !0r9   
str | Nonec                    | j         j        S r3   )rR  	base_path)r6   s    r7   
_base_pathzPartitioningScheme._base_path\  s    $..r9   N)rO  r(   r0   rP  )r0   rT  )r:   r;   r<   rS  propertyrW  r4   r9   r7   rN  rN  U  sH        0 0 0 0 / / / X/ / /r9   rN  )\r^   r-   r?   r   rb   r>  r  rj   r   r  rY   r  r  rW   ConcatMethodr  r9  r   rl   rg   r3  r   r   r   rL  r   rC  EpochTimeUnitr  r  rF  rt   rw   r  r  rz   r   rP   rO   r}   JSONEncoderr   r   r~   r   r   r   r   r@  r;  r=  r   rM   r   rK   rG   r   r   r  r   rN  r   rD   rF   rE   r  rH   rN   r   r   r   r  rI   rJ   r  rX   r   r!  r?  r:  r<  r   r   r   rL   r   r  r  r   r   r   r  c                  &    e Zd ZU dZddZded<   dS )	ParquetMetadataContextz
    The context given when writing file-level parquet metadata.

    .. warning::
        This functionality is considered **experimental**. It may be removed or
        changed at any point without it being considered a breaking change.
    arrow_schemastrr0   rP  c                   || _         d S r3   )r^  )r6   r^  s     r7   rS  zParquetMetadataContext.__init__  s    (r9   N)r^  r_  r0   rP  )r:   r;   r<   r=   rS  r%  r4   r9   r7   r]  r]    s>          ) ) ) ) r9   r]  ParquetMetadataFnParquetMetadata)
__future__r   collections.abcr   r   r   r   pathlibr   typingr	   r
   r   r   r   r   r   r   r   
contextlibsysr   r   r   r   decimalr   sqlalchemy.enginer   r   sqlalchemy.ormr   polarsr   r   r   r   polars.datatypesr   r   r    r!   polars.dependenciesr"   npr#   pdr$   par%   polars.lazyframe.engine_configr&   polars.selectorsr'   suppressImportErrorpolars.polarsr(   version_infor+   typing_extensionsr-   r?   rD   r%  typerE   rF   rG   intfloatboolr_  listtupler   r@   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rW   rX   rY   r^   rb   rg   rj   rl   rt   rw   rz   r}   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rY  r   r   r   r   rZ  r   r   r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r!  r'  r/  r3  r8  r9  r:  sliceranger;  r<  r=  r>  r?  r@  rC  rE  rF  r[  rL  rN  __all__r]  r   ra  rb  r4   r9   r7   <module>r     sz   " " " " " " " C C C C C C C C C C C C      
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  0JJJ88888888888844444444&&&&&&999999999999SSSSSSSSSSSS//////000000111111))))))888888111111		[	)	) 1 10000001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 7""$$$$$$$//////$ $ $ $ $8 $ $ $X X X X XH X X X ""=> > > > > %d>&:N&J K  K K K K$T-%8-%GH  H H H H %nh~6N&N O  O O O O!IKJILLcOsCxKLOJ    " $C~~t;<<=U3c5)M#NNOOPQS       ^ 34
 4 4 4 4!#ui"78 8 8 8 8"#JK K K K K#NOS$PU$UV  V V V V !149!<= = = = =!&(C"78 8 8 8 8M>4?@ @ @ @ @ '(Q R  R R R R - , , , ,"'\(9":  : : : : DE	 E E E E$%HI I I I I"#PQ Q Q Q Q!()>!?  ? ? ? ? !56 6 6 6 6%>      o. . . . ./
 / / / /#$AB B B B B#$>? ? ? ? ?78 8 8 8 8&7       !1 1 1 1 1!"23 3 3 3 3%:      !(D!      A     $F       UV
 V V V V89 9 9 9 9DE	 E E E E!()9!:  : : : :0
 0 0 0 0	     
 
 
 
 
  '}5  5 5 5 5./ / / / / !=> > > > > '(F G  G G G G%&>?  ? ? ? ?!()D!E  E E E E &&FG  G G G G#$CD D D D D!()<!=  = = = =!F     (//L'M  M M M M 	  ''<=  = = = =!"67 7 7 7 7"#78 8 8 8 8 !<= = = = =23"?3 3 3 3 3 . . . . .%&<=  = = = =$%@A A A A A%o6  6 6 6 6#*+M#N  N N N N"?3 3 3 3 3 "Cx'hv6F1F)GQRRSSM	 	 	 	 	 &	
&:C&? @
@A	#wsCx 
 !#     
 $+	

3
/0	#uWS#X&sGCH<M7M1N(OOP
PQS$     
 %*E&.B(CCDcIJSM
%      %* %c3h(<"==>C%      "'CsJsO+,,-sO
"      %,,K$L  L L L L W\;XvNN
GKk::	c3m,
 , , , , %,,N$O  O O O O    I   & & & & &h & & &
    (   
( ( ( ( ([ ( ( (  %%FG  G G G G %[&*;;!     
 "%  $ $ $ $ %		SM	!      !$  # # # #$	SM       TN    
  %%8:L%LM  M M M M#$68I;$VW  W W W W 34kA
     /0	 0 0 0 0uI	IJEOK	
 	 	 	 	 HcUE\*HcUCZ,@@A$%    	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/] ] ]@         ()?(@$sCx.(PQ  Q Q Q Q"4S>3D#DE E E E E E Es   &B99B= B=