o i4@sddlmZddlZddlmZmZddlmZddlm Z m Z e e ddl mZWdn1s6wYerKddlmZddlmZmZdd dZdddddddZdS)) annotationsN) TYPE_CHECKINGAny)dataframe_to_pydf)wrap_dfwrap_s)PySeries) DataFrame)SchemaDefinition SchemaDictobjrreturnboolcstfdddDS)z?Check if object looks like it supports the PyCapsule interface.c3s |] }tt|dVqdS)N)callablegetattr).0attrr N/home/app/Keep/.python/lib/python3.10/site-packages/polars/_utils/pycapsule.py s  zis_pycapsule..)__arrow_c_stream____arrow_c_array__)anyrrrr is_pycapsules rF)schemaschema_overridesrechunkrSchemaDefinition | NonerSchemaDict | Nonerr cCst|drd}tt|}|||}n!t|dr.d}tt|}|||}n d|d}t||r>|}|sB|rKt t |||d}|S)z&Convert PyCapsule object to DataFrame.rrz3object does not support PyCapsule interface; found  )rr) hasattrrrZfrom_arrow_c_arrayZto_frameZunnestZfrom_arrow_c_stream TypeErrorrrr)r rrrZ tmp_col_namesdfmsgrrrpycapsule_to_frames"    r')r rr r) r rrrrrrrr r ) __future__r contextlibtypingrrZ$polars._utils.construction.dataframerZpolars._utils.wraprrsuppress ImportErrorZ polars._plrrZpolarsr Zpolars._typingr r rr'rrrrs