o i#@sddlZddlZddlmZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZdd Zd d Zd d ZdddZddZddZdS)N)copy)Dialect create_engine)make_url)OperationalErrorProgrammingError)object_session)UnmappedInstanceErrorcCsNt|dr |j}nzt|j}Wn ty|}Ynwt|ds%td|S)a' Return the bind for given SQLAlchemy Engine / Connection / declarative model object. :param obj: SQLAlchemy Engine / Connection / declarative model object :: from sqlalchemy_utils import get_bind get_bind(session) # Connection object get_bind(user) bindexecutezSThis method accepts only Session, Engine, Connection and declarative model objects.)hasattrr rr TypeError)objconnr[/var/www/html/LicenciasMIG/venv/lib/python3.10/site-packages/spyne/store/relational/util.pyget_bind.s   rcCs*t|tr|}nt|j}|||S)a Conditionally quote an identifier. :: from sqlalchemy_utils import quote engine = create_engine('sqlite:///:memory:') quote(engine, 'order') # '"order"' quote(engine, 'some_other_identifier') # 'some_other_identifier' :param mixed: SQLAlchemy Session / Connection / Engine / Dialect object. :param ident: identifier to conditionally quote ) isinstancerrdialectpreparerquote)mixedidentrrrrrHs  rc Cstt|}|j}|jdrd|_n |jdsd|_t|}|jjdkr3d|}t| | S|jjdkrFd|}t| | S|jjdkrV|dkpUt j |Sd }z||_t|}| |Wd SttfysYd Sw) aCheck if a database exists. :param url: A SQLAlchemy engine URL. Performs backend-specific testing to quickly determine if a database exists on the server. :: database_exists('postgresql://postgres@localhost/name') #=> False create_database('postgresql://postgres@localhost/name') database_exists('postgresql://postgres@localhost/name') #=> True Supports checking against a constructed URL as well. :: engine = create_engine('postgresql://postgres@localhost/name') database_exists(engine.url) #=> False create_database(engine.url) database_exists(engine.url) #=> True postgressqliteN postgresqlz,SELECT 1 FROM pg_database WHERE datname='%s'mysqlzLSELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = '%s':memory:zSELECT 1TF)rrdatabase drivername startswithrrnameboolr scalarospathexistsrr)urlrenginetextrrrdatabase_exists\s4       r*utf8 template1cCstt|}|j}|jdrd|_n |jdsd|_t|}|jjdkrL|jdkr8ddl m }| |d t|||t||}||dS|jjd krbd t|||}||dS|jjdkru|d krut|d dSd t||}||dS)aIssue the appropriate CREATE DATABASE statement. :param url: A SQLAlchemy engine URL. :param encoding: The encoding to create the database as. :param psql_template: The name of the template from which to create the new database, only supported by PostgreSQL driver. As per Postgresql docs, defaults to "template1". To create a database, you can pass a simple URL that would have been passed to ``create_engine``. :: create_database('postgresql://postgres@localhost/name') You may also pass the url from an existing engine. :: create_database(engine.url) Has full support for mysql, postgres, and sqlite. In theory, other database engines should be supported. rrNrpsycopg2rISOLATION_LEVEL_AUTOCOMMITz/CREATE DATABASE {0} ENCODING '{1}' TEMPLATE {2}rz)CREATE DATABASE {0} CHARACTER SET = '{1}'rwzCREATE DATABASE {0})rrrrr rrr!driverpsycopg2.extensionsr/raw_connectionset_isolation_levelformatrr openclose)r'encoding psql_templaterr(r/r)rrrcreate_databases:       r:cCs,tt|}|j}|jdrd|_n |jdsd|_t|}|jjdkr3|jdkr3t |jdS|jjdkr|j dkrddl m }| |ttt|d dd }|dd krj|d d krjdnd}d||d}||dt||}||dSdt||}||dS)a8Issue the appropriate DROP DATABASE statement. :param url: A SQLAlchemy engine URL. Works similar to the :ref:`create_database` method in that both url text and a constructed url are accepted. :: drop_database('postgresql://postgres@localhost/name') drop_database(engine.url) rr,rNrr-rr.zSHOW server_version. pidprocpidz SELECT pg_terminate_backend(pg_stat_activity.%(pid_column)s) FROM pg_stat_activity WHERE pg_stat_activity.datname = '%(database)s' AND %(pid_column)s <> pg_backend_pid(); ) pid_columnrzDROP DATABASE {0})rrrrr rrr!r$remover1r2r/r3r4listmapintr firstsplitr5r)r'rr(r/versionrAr)rrr drop_databases8     rIcsfdd}|S)Nc sHt}z||i|}||W|jddS|jddw)Ntime)sort)cProfileProfileenabledisable print_stats)argskwargsprofileresultfuncrr profiled_func sz"do_cprofile..profiled_funcr)rVrWrrUr do_cprofile s  rX)r+r,)r$rLrsqlalchemy.enginerrsqlalchemy.engine.urlrsqlalchemy.excrrsqlalchemy.ormrsqlalchemy.orm.excr rrr*r:rIrXrrrrs!     6? ;