
    h                       S SK Jr  S SKrS SKJrJr  S SKJr  S SKJ	r	J
r
  \R                  " \5         S SKJr  SSS5        \(       a  S SKJr  S SKJrJr  SS	 jrSSS
S.         SS jjrg! , (       d  f       N== f)    )annotationsN)TYPE_CHECKINGAny)dataframe_to_pydf)wrap_dfwrap_s)PySeries)	DataFrame)SchemaDefinition
SchemaDictc                @    [        U S5      =(       d    [        U S5      $ )z1Check if object supports the PyCapsule interface.__arrow_c_stream____arrow_c_array__)hasattr)objs    Q/home/james-whalen/.local/lib/python3.13/site-packages/polars/_utils/pycapsule.pyis_pycapsuler      s    3,-R>Q1RR    F)schemaschema_overridesrechunkc                  [        U S5      (       aB  Sn[        [        R                  " U 5      5      nUR	                  U5      R                  U5      nOe[        U S5      (       aB  Sn[        [        R                  " U 5      5      nUR	                  U5      R                  U5      nOSU < S3n[        U5      eU(       a  UR                  5       nU(       d  U(       a  [        [        XaUS95      nU$ )z&Convert PyCapsule object to DataFrame.r    r   z3object does not support PyCapsule interface; found  )r   r   )r   r   r	   from_arrow_c_arrayto_frameunnestfrom_arrow_c_stream	TypeErrorr   r   r   )r   r   r   r   tmp_col_namesdfmsgs           r   pycapsule_to_framer$      s     s'(( 8..s34ZZ%,,\:	*	+	+ 8//45ZZ%,,\:CC7!LnZZ\!bBRS
 Ir   )r   r   returnbool)
r   r   r   zSchemaDefinition | Noner   zSchemaDict | Noner   r&   r%   r
   )
__future__r   
contextlibtypingr   r   $polars._utils.construction.dataframer   polars._utils.wrapr   r   suppressImportErrorpolars._plrr	   polarsr
   polars._typingr   r   r   r$    r   r   <module>r2      s    "  % B .%$ &  ;S '+*.	 $ (	
   &%s   A11
A?