o i@sddlmZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z eeddlmZWdn1s@wYerMdd lmZddd dddZdS)) annotationsN)Path)IO TYPE_CHECKING)normalize_filepath)wrap_df)parse_columns_arg) PyDataFrame) DataFrame)columnsn_rowssourcestr | Path | IO[bytes] | bytesr list[int] | list[str] | Noner int | Nonereturnr cCs:t|ttfr t|}t|\}}t||||}t|S)a Read into a DataFrame from Apache Avro format. Parameters ---------- source Path to a file or a file-like object (by "file-like object" we refer to objects that have a `read()` method, such as a file handler like the builtin `open` function, or a `BytesIO` instance). For file-like objects, the stream position may not be updated accordingly after reading. columns Columns to select. Accepts a list of column indices (starting at zero) or a list of column names. n_rows Stop reading from Apache Avro file after reading `n_rows`. Returns ------- DataFrame ) isinstancestrrrrr read_avror)r r r Z projectionZ column_namesZpydfrE/home/app/Keep/.python/lib/python3.10/site-packages/polars/io/avro.pyrs  r)r rr rr rrr ) __future__r contextlibpathlibrtypingrrZpolars._utils.variousrZpolars._utils.wraprZpolars.io._utilsrsuppress ImportErrorZ polars._plrr Zpolarsr rrrrrs