ɯeifddlZddlmZmZmZ ddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZdd lmZdd lmZddlZdd lmZdd lmZmZddlm Z GddZ!y#e$rZed edZ[wwxYw)N)ListOptionalUnion)Root)Database) NotFoundError) Procedure)Schema)Table TableColumn)UserDefinedFunction)Viewz.Missing optional dependency: 'snowflake.core'.)convert_sp_to_sf_type)lit parse_json)DataTypec eZdZdZd:dZ d;deeeefdeeee e e e e fdefdZ d;deeee fdeeee e e e fdefd Zd eee e fd eeedefd Zde#Z?e$Z@e%ZAe&ZBe(ZCe)ZDe*ZEe+ZFe,ZGe-ZHe.ZIe/ZJe0ZKe1ZLy)=CatalogzThe Catalog class provides methods to interact with and manage the Snowflake objects. It allows users to list, get, and drop various database objects such as databases, schemas, tables, views, functions, etc. returnNc@||_t||_d|_yN)_sessionr_root_python_regex_udf)selfsessions \/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/snowflake/snowpark/catalog.py__init__zCatalog.__init__#s '] !%database model_objcVt|ttttt fr|j }|J|St|tr|r|St|tr |jS|s)|jj}| td|Stdt|d)Nz9No database detected. Please provide database to proceed.z0Unexpected type. Expected str or Database, got '') isinstancer r rr r database_namestrrnamerget_current_database ValueErrortype)rr r!db_namecurrent_databases r_parse_databasezCatalog._parse_database(s i&%yBU!V W--G& &&N h $O h )== #}}AAC ' O$ #>tH~>Na P  rschemacLt|ttttfr|j }|J|St|t r|r|St|tr |jS|s)|jj}| td|Stdt|d)Nz5No schema detected. Please provide schema to proceed.z.Unexpected type. Expected str or Schema, got 'r#) r$r rr r schema_namer&r r'rget_current_schemar)r*)rr.r!r0current_schemas r _parse_schemazCatalog._parse_schemaCs i%y:M!N O#//K* **  fc "vM ff %;; !]]==?N% K" !arg_types must be provided when function/procedure is a stringz, ()c34K|]}|jywr)datatype).0args r z7Catalog._parse_function_or_procedure..ms!G3#,,!Gs)r$r&r)joinr argumentsr')rr4r5arg_type arg_types_strs r_parse_function_or_procedurez$Catalog._parse_function_or_procedure^s b#   T!IIAJKX&x0KMT=/+ + !G",,!GG ''!M?!,, LsBc|jj5|j dddydtdtdtfd}|jj j ||_dddy#1swYyxYw)Npatterninputrc@ttj||Sr)boolrematch)rDrEs rpython_regex_filterz:Catalog._initialize_regex_udf..python_regex_filterusBHHWe455r)r_lockrr&rGudfregister)rrJs r_initialize_regex_udfzCatalog._initialize_regex_udfps} ]]  U%%1 U U 6S 6 6 6&*]]%6%6%?%?@S%TD " U U UsA7AA77B)r r.rDlike object_namerDrOc |j|}|j|}|rd|dnd} |jjd|d| d|d|d } |rT|j |j J| j |j t|td d } ttfd | jS) NzLIKE 'r#zSHOW AS RESOURCE  z IN .z -- catalog apiz "As Resource"r'c,j|dS)Nr) from_json)row object_classs rz'Catalog._list_objects..sL$:$:3q6$Br) r-r3rsqlrNrfilterrrlistmapcollect) rrPrXr r.rDrOr+r0like_strdfs ` r _list_objectszCatalog._list_objectszs&&x0((0 '+VD6# ]]   }AhZtG9Ak]Ra b    & & ())5 55 &&L*_"=f"EB CBBJJLQRRr)rDrOc|jjj|}rtfd|}t |S)zList databases in the current session. Args: pattern: the python regex pattern of name to match. Defaults to None. like: the sql style pattern for name to match. Default to None. rOcDtj|jSrrHrIr'xrDs rrYz(Catalog.list_databases..BHHWaff$=r)r databasesiterr[r\)rrDrOrjs ` rlist_databaseszCatalog.list_databasess<zz##((d(3 =tDDDzr)r rDrOc|j|}|jj|jj |}rt fd|}t |S)aList schemas in the current session. If database is provided, list schemas in the database, otherwise list schemas in the current database. Args: database: database name or ``Database`` object. Defaults to None. pattern: the python regex pattern of name to match. Defaults to None. like: the sql style pattern for name to match. Default to None. rccDtj|jSrrerfs rrYz&Catalog.list_schemas..rhr)r-rrischemasrjr[r\)rr rDrOr+rjs ` r list_schemaszCatalog.list_schemassW&&x0zz##G,4499t9D =tDDDzrc8|jdt||||S)aList tables in the current session. If database or schema are provided, list tables in the given database or schema, otherwise list tables in the current database/schema. Args: database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. pattern: the python regex pattern of name to match. Defaults to None. like: the sql style pattern for name to match. Default to None. TABLESrPrXr r.rDrO)rar rr r.rDrOs r list_tableszCatalog.list_tabless."!!  "  rc8|jdt||||S)aList views in the current session. If database or schema are provided, list views in the given database or schema, otherwise list views in the current database/schema. Args: database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. pattern: the python regex pattern of name to match. Defaults to None. like: the sql style pattern for name to match. Default to None. VIEWSrr)rarrss r list_viewszCatalog.list_viewss."!! "  rr r. table_namectt|tr|j|||}n|}|j}|J|S)zList columns in the given table. Args: table_name: table name. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. rx)r$r& get_tablecolumns)rryr r.tablecolss r list_columnszCatalog.list_columnssA j# &NN:NPEE}} rc8|jdt||||S)aList of procedures in the given database and schema. If database or schema are not provided, list procedures in the current database and schema. Args: database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. pattern: the python regex pattern of name to match. Defaults to None. like: the sql style pattern for name to match. Default to None. PROCEDURESrr)rar rss rlist_procedureszCatalog.list_proceduress."!!$" "  rc8|jdt||||S)aList of user defined functions in the given database and schema. If database or schema are not provided, list user defined functions in the current database and schema. Args: database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. pattern: the python regex pattern of name to match. Defaults to None. like: the sql style pattern for name to match. Default to None. zUSER FUNCTIONSrr)rar rss rlist_user_defined_functionsz#Catalog.list_user_defined_functions(s. !!(, "  rc6|jjS)zGet the current database.)rr(rs rr(zCatalog.get_current_databaseBs}}1133rc6|jjS)zGet the current schema.)rr1rs rr1zCatalog.get_current_schemaFs}}//11rcP|jj|jS)zName of the database to get)rrifetch)rr s r get_databasezCatalog.get_databaseJs zz##H-3355r)r c|j|}|jj|j|j S)zName of the schema to get.)r-rrirnr)rr.r r+s r get_schemazCatalog.get_schemaNs=&&x0zz##G,44V<BBDDrc|j|}|j|}|jj|j|j |j S)anGet the table by name in given database and schema. If database or schema are not provided, get the table in the current database and schema. Args: table_name: name of the table. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. )r-r3rrirntablesr)rryr r.r+r0s rr{zCatalog.get_tableUs\&&x0((0 JJ  ) W[ " VJ UW  r view_namec|j|}|j|}|jj|j|j |j S)ajGet the view by name in given database and schema. If database or schema are not provided, get the view in the current database and schema. Args: view_name: name of the view. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. )r-r3rrirnviewsr)rrr r.r+r0s rget_viewzCatalog.get_viewmsZ&&x0((0 JJ  ) 1 1+ > D DY O U U W rprocedure_namec|j|}|j|}|j||}|jj|j |j |jS)aGet the procedure by name and argument types in given database and schema. If database or schema are not provided, get the procedure in the current database and schema. Args: procedure_name: name of the procedure. arg_types: list of argument types to uniquely identify the procedure. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. )r-r3rBrrirn proceduresr)rrr5r r.r+r0 procedure_ids r get_procedurezCatalog.get_proceduresn"&&x0((0 88S JJ  ) W[ " Z  &UW  rudf_namec|j|}|j|}|j||}|jj|j |j |jS)aGet the user defined function by name and argument types in given database and schema. If database or schema are not provided, get the user defined function in the current database and schema. Args: udf_name: name of the user defined function. arg_types: list of argument types to uniquely identify the user defined function. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. )r-r3rBrrirnuser_defined_functionsr)rrr5r r.r+r0 function_ids rget_user_defined_functionz!Catalog.get_user_defined_functionso$&&x0((0 77)L JJ  ) W[ " # #K 1UW  rc\|j|}|jj|y)zSet the current default database for the session. Args: database: database name or ``Database`` object. N)r-r use_databaserr r+s rset_current_databasezCatalog.set_current_databases& &&x0 ""7+rc\|j|}|jj|y)z}Set the current default schema for the session. Args: schema: schema name or ``Schema`` object. N)r3r use_schema)rr.r0s rset_current_schemazCatalog.set_current_schemas& ((0    -rc|j|} |jj|jy#t$rYywxYw)zwCheck if the given database exists. Args: database: database name or ``Database`` object. TF)r-rrirrrs rdatabase_existszCatalog.database_existssI &&x0  JJ  ) / / 1  s'; AAc|j||}|j|} |jj|j|j y#t $rYywxYw)a,Check if the given schema exists in the given database. If database is not provided, check if the schema exists in the current database. Args: schema: schema name or ``Schema`` object. database: database name or ``Database`` object. Defaults to None. TF)r-r3rrirnrrrr.r r+r0s r schema_existszCatalog.schema_existssf&&x8((0   JJ  ) 1 1+ > D D F  s4A A&%A&r}c*|j||}|j||}t|tr|n |j} |j j |j|j|jy#t$rYywxYw)aCheck if the given table exists in the given database and schema. If database or schema are not provided, check if the table exists in the current database and schema. Args: table: table name or ``Table`` object. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. TF) r-r3r$r&r'rrirnrrrrr}r r.r+r0rys r table_existszCatalog.table_existss&&x7((7 (4U%**   JJ  ) 1 1+ > E E eg  AB BBviewc*|j||}|j||}t|tr|n |j} |j j |j|j|jy#t$rYywxYw)aCheck if the given view exists in the given database and schema. If database or schema are not provided, check if the view exists in the current database and schema. Args: view: view name or ``View`` object. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. TF) r-r3r$r&r'rrirnrrrrrr r.r+r0rs r view_existszCatalog.view_exists s&&x6((6 &tS1Dtyy   JJ  ) 1 1+ > D DY O U U W  r procedurec|j||}|j||}|j||} |jj|j |j |jy#t$rYywxYw)aCheck if the given procedure exists in the given database and schema. If database or schema are not provided, check if the procedure exists in the current database and schema. Args: procedure: procedure name or ``Procedure`` object. arg_types: list of argument types to uniquely identify the procedure. Defaults to None. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. TF) r-r3rBrrirnrrr)rrr5r r.r+r0rs rprocedure_existszCatalog.procedure_exists!s"&&x;((; 88IN   JJ  ) 1 1+ > I I eg  AA:: BBrLc|j||}|j||}|j||} |jj|j |j |jy#t$rYywxYw)aTCheck if the given user defined function exists in the given database and schema. If database or schema are not provided, check if the user defined function exists in the current database and schema. Args: udf: user defined function name or ``UserDefinedFunction`` object. arg_types: list of argument types to uniquely identify the user defined function. Defaults to None. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. TF) r-r3rBrrirnrrr)rrLr5r r.r+r0rs ruser_defined_function_existsz$Catalog.user_defined_function_exists>s&&&x5((5 77YG   JJ  ) 1 1+ > U U eg  rct|j|}|jj|jy)zlDrop the given database. Args: database: database name or ``Database`` object. N)r-rridroprs r drop_databasezCatalog.drop_database^s/ &&x0 W%**,rc|j||}|j|}|jj|j|j y)aDrop the given schema in the given database. If database is not provided, drop the schema in the current database. Args: schema: schema name or ``Schema`` object. database: database name or ``Database`` object. Defaults to None. N)r-r3rrirnrrs r drop_schemazCatalog.drop_schemagsL&&x8((0  W%--k:??Arc |j||}|j||}t|tr|n |j}|j j |j|j|jy)azDrop the given table in the given database and schema. If database or schema are not provided, drop the table in the current database and schema. Args: table: table name or ``Table`` object. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. N) r-r3r$r&r'rrirnrrrs r drop_tablezCatalog.drop_tablexsm&&x7((7 (4U%**  W%--k:AA*MRRTrc |j||}|j||}t|tr|n |j}|j j |j|j|jy)auDrop the given view in the given database and schema. If database or schema are not provided, drop the view in the current database and schema. Args: view: view name or ``View`` object. database: database name or ``Database`` object. Defaults to None. schema: schema name or ``Schema`` object. Defaults to None. N) r-r3r$r&r'rrirnrrrs r drop_viewzCatalog.drop_viewsm&&x6((6 &tS1Dtyy  W%--k:@@KPPRr)rz"snowflake.snowpark.session.SessionrNr)rN)M__name__ __module__ __qualname____doc__rrrr&rr r rr r r-r3rrrBrNrarkrortrwr rrrr(r1rrr{rrrrrrGrrrrrrrrrr listDatabases listSchemas listTables listViews listColumnslistProcedureslistUserDefinedFunctionsgetCurrentDatabasegetCurrentSchema getDatabase getSchemagetTablegetView getProceduregetUserDefinedFunctionsetCurrentDatabasesetCurrentSchemadatabaseExists schemaExists tableExists viewExistsprocedureExistsuserDefinedFunctionExists dropDatabase dropSchema dropTabledropViewrrrrs  &   5h/0  #vudI7JJ K   @   sF{+,  #udI/BB C   6- #y"55 6-DN+- -$U48/3!%" S S 5h/0 S sF{+, S# Ssm SL"&" #sm  h *48!%" 5h/0#  sm  f 048/3!%"  5h/0 sF{+,  #  sm   e :48/3!%"  5h/0 sF{+,  #  sm   d <48/3 #u*%5h/0  sF{+,  k  248/3!%"  5h/0 sF{+,  #  sm   i :48/3!%"  5h/0 sF{+,  #  sm   ! " 44hsm42HSM26S6X6 JNEE(0sH}1E(FE E48/3   5h/0  sF{+,    848/3   5h/0  sF{+,    448/3   > 5h/0  sF{+,    @48/3   > 5h/0  sF{+,    :,U3=-A,d,.sF{);.. c8m(<  "48 c6k"5h/0   248/3 S%Z 5h/0  sF{+,   <48/3 CI5h/0  sF{+,   6/3 48/3 i(DN+ 5h/0  sF{+,  @/3 48/3  3++ ,DN+ 5h/0  sF{+,  @-eCM&:-t-48 Bc6k"B5h/0 B  B*48/3 US%Z U5h/0 U sF{+, U  U248/3 SCIS5h/0 S sF{+, S  S,#MKJIK$N:-)KIHG L6-)$N LKJ&O < LJIHrr)"rHtypingrrrsnowflake.corersnowflake.core.databasersnowflake.core.exceptionsrsnowflake.core.procedurer snowflake.core.schemar snowflake.core.tabler r $snowflake.core.user_defined_functionr snowflake.core.viewr ImportErroresnowflake.snowpark snowflake'snowflake.snowpark._internal.type_utilsrsnowflake.snowpark.functionsrrsnowflake.snowpark.typesrrrrrrsl  ((  #072,7H(I8-d d  8    s2A%%A8* A33A8