PiuhUddlZddlZddlmZddlmZddlZddlm Z de de dej fdZ id ej d ej d ejd ejd ejdejdejde de de de de de dejdejdejdejejejejdZee efed<ddiddid Zee ee e ezffed!<d"Zdd#d$e de d%ede dzdej eee zej fzf d&Zd'e de dzdefd(Zd)ede de dzdefd*Zd'e d%edefd+ZdS),N)Callable)Any)dataset_download sql_querypathreturnctj|5}tj||}|cdddS#1swxYwYdSN)sqlite3connectpdread_sql_query)rrconndfs m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/kagglehub/pandas_datasets.pywrapped_read_sql_queryrs   $  y$ / /s 9==z.csv.tsvz.json.jsonlz.xmlz.parquetz.featherz.sqlitez.sqlite3z.dbz.db3z.s3dbz.dl3z.xlsz.xlsxz.xlsmz.xlsb)z.odfz.odsz.odt%SUPPORTED_READ_FUNCTIONS_BY_EXTENSIONsep linesT)rrSTATIC_KWARGS_BY_EXTENSIONz/Loading from a SQLite file requires a SQL query) pandas_kwargshandlerc8|in|}tj|d}t||}t ||} |t |||it ||}n'#t$r}d|} t| |d}~wwxYw|S)aCreates pandas DataFrame(s) from a file in the dataset Args: handle: (string) The dataset handle path: (string) Path to a file within the dataset pandas_kwargs: (dict) Optional set of kwargs to pass to the pandas `read_*` method while constructing the DataFrame(s) sql_query: (string) Argument to be used for SQLite files. Required when reading a SQLite file. See pandas documentation for details: https://pandas.pydata.org/docs/reference/api/pandas.read_sql_query.html Returns: - dict[int | str, DataFrame] for Excel-like files with multiple sheets - A pandas DataFrame for all others Raises: ValueError: If the file extension is not supported or the file fails to read NzError reading file: ) osrsplitext_validate_read_functionr _build_args _build_kwargs Exception ValueError) rrrrfile_extension read_functionfilepathresulteread_error_messages rload_pandas_datasetr+<s2(/BB]MW%%d++A.N+NIFFM --H4 ) < < NM::   4447A77+,,!34 Ms %A33 B=BBr%c|tvrBd|ddt}t|dt|}|tur|stt |S)NzUnsupported file extension: 'z"'. Supported file extensions are: z, )rjoinkeysr$rMISSING_SQL_QUERY_ERROR_MESSAGE)r%rextension_error_messager&s rr r gsBBB hN h h.2ii8]8b8b8d8d.e.e h h 011t;9.IM...y.8999 r&c&|tkr|gn||gSr )r)r&rrs rr!r!vs"&<<|tvrin t|}i||Sr )r)r%r static_kwargss rr"r"{s2$>>>D^_mDn .m -} --r1)rr collections.abcrtypingrpandasr kagglehub.datasetsrstr DataFramerread_csv read_jsonread_xml read_parquet read_feather read_excelrdict__annotations__rboolr/intr+r listr!r"r1rrGs $$$$$$////// c> BK> BK> R\> bl >  BK >  >>%>&> !> "> #>  "!>& BM'>( R])>* R]+>, R]->. M M M3>>>%tCM'::JOPT bikoap?q?qDd3d ?&;!;<qqq"S ((( ( ( ( Tz ( \DsBL011 ((((V C C$J 8    TxTsTsTzTdTTTT .#.c.d......r1