o PDi@sFddlmZddlmZddlmZddlmZGdddedZdS) ) ArrowArray) DataFrameImpl) BaseMetaClass)errorsc@seZdZdZddZeddZeddZdd Zdd d Z d e e fd dZ d e e fddZded e fddZde d e fddZd efddZd efddZdS) DataFrameNcCsttjdSN)r _raise_errERR_INTERNAL_CREATION_REQUIREDselfr K/var/www/Datamplify/venv/lib/python3.10/site-packages/oracledb/dataframe.py__init__)szDataFrame.__init__cCs||}|t||Sr)__new__ _initializerfrom_arrow_stream)clsobjdfr r r _from_arrow,s zDataFrame._from_arrowcCs||}|||Sr)rr)rimplrr r r _from_impl2s  zDataFrame._from_implcCs<||_dd|D|_i|_|jD]}||j|j<qdS)zB Initializes the object given the implementation. cSsg|]}t|qSr )rr.0ar r r =sz)DataFrame._initialize..N)_impl get_arrays_arrays_arrays_by_namename)r rarrayr r r r8s  zDataFrame._initializecCs|durtd|jS)z Returns the ArrowArrayStream PyCapsule which allows direct conversion to foreign data frames that support this interface. Nrequested_schema)NotImplementedErrorrget_stream_capsule)r r"r r r __arrow_c_stream__Bs zDataFrame.__arrow_c_stream__returncCs|jS)ze Returns a list of ArrowArray objects, each containing a select list column. rr r r r column_arraysKszDataFrame.column_arrayscCsdd|jDS)zG Returns a list of the column names in the data frame. cSsg|]}|jqSr )r rr r r rVsz*DataFrame.column_names..r'r r r r column_namesRszDataFrame.column_namesicCs8|dks ||krtd|d|d|j|S)z Returns an :ref:`ArrowArray ` object for the column at the given index ``i``. If the index is out of range, an IndexError exception is raised. z Column index z% is out of bounds for DataFrame with z columns) num_columns IndexErrorr)r r*r r r get_columnXs zDataFrame.get_columnr cCs,z|j|WStytd|dw)z Returns an :ref:`ArrowArray ` object for the column with the given name ``name``. If the column name is not found, a KeyError exception is raised. zColumn z not found in DataFrame)rKeyError)r r r r r get_column_by_namees   zDataFrame.get_column_by_namecCs t|jS)zB Returns the number of columns in the data frame. lenrr r r r r,ps zDataFrame.num_columnscCst|jdS)z? Returns the number of rows in the data frame. r+r1r r r r num_rowsvszDataFrame.num_rowsr)__name__ __module__ __qualname__rr classmethodrrrr%listrr(strr)intr.r0r,r3r r r r r&s      r) metaclassN) arrow_arrayr arrow_implrbaserrrr r r r s