o ̳iu@sUddlZddlZddlmZddlmZddlZddlm Z de de dej fdd Z id ej d ej d ejd ejdejdejdejde de de de de de dejdejdejdejejejejdZee efed<ddidd id!Zee ee e eBffed"<d#Zdd$d%e de d&ede dBdej eee Bej fBf d'd(Zd)e de dBdefd*d+Zd,ede de dBdefd-d.Zd)e d&edefd/d0ZdS)1N)Callable)Any)dataset_download sql_querypathreturncCs>t|}t||}|WdS1swYdSN)sqlite3connectpdread_sql_query)rrconndfrM/home/ubuntu/.local/lib/python3.10/site-packages/kagglehub/pandas_datasets.pywrapped_read_sql_querys  $rz.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_kwargshandlerc Cs|durin|}tj|d}t||}t||}z|t|||it||}W|StyA}z d|} t| |d}~ww)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_messagerrrload_pandas_dataset<s      r)r#cCsL|tvrd|ddt}t|dt|}|tur$|s$tt|S)NzUnsupported file extension: 'z"'. Supported file extensions are: z, )rjoinkeysr"rMISSING_SQL_QUERY_ERROR_MESSAGE)r#rextension_error_messager$rrrrgs   rr$cCs|tkr|gS||gSr)r)r$rrrrrrvsrcCs |tvrint|}i||Sr)r)r#r static_kwargsrrrr {s r )rr collections.abcrtypingrpandasr kagglehub.datasetsrstr DataFramerread_csv read_jsonread_xml read_parquet read_feather read_excelrdict__annotations__rboolr,intr)rlistrr rrrrs~        . +