관리-도구
편집 파일: provision.cpython-38.pyc
U -?�f� � @ sf d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd lm Z dd lm Z dd lm Z dd lmZ ddlmZ ddlmZ ddlmZ e�d�dd� �Ze �d�dd� �Ze�d�dd� �Ze�d�dd� �Ze �d�dd� �Z e �d�d d � �Z e�d�d d � �Ze�d�dd d!�d"d#��Zd$d%gZe �d�d&d'� �ZdS )(� N� )�exc)�inspect)�text)�warn_test_suite)� create_db)�#drop_all_schema_objects_post_tables)�"drop_all_schema_objects_pre_tables)�drop_db)�log)�post_configure_engine)�prepare_for_drop_tables)� set_default_schema_on_connection)�temp_table_keyword_args)�upsert� postgresqlc C s� | j j}|jdd��� ��}|s,|�d��� }d}z|�d||f � W q� tjk r� } z>|d7 }|dkrl� dt|�kr�t � d ||j |� t � d � W 5 d }~X Y q0 � Y q0X q�q0W 5 Q R X d S ) N� AUTOCOMMIT�Zisolation_levelzselect current_database()r z CREATE DATABASE %s TEMPLATE %s� r zaccessed by other userszFWaiting to create %s, URI %r, template DB %s is in use sleeping for .5g �?)�optionsZpostgresql_templatedb�execution_options�begin�exec_driver_sqlZscalarr ZOperationalError�strr �info�url�time�sleep)�cfg�eng�identZ template_db�conn�attempt�err� r$ �X/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/postgresql/provision.py� _pg_create_db s4 � � �r& c C sX |� � jdd��>}|�� �* |�td�t|d�� |�d| � W 5 Q R X W 5 Q R X d S )Nr r z�select pg_terminate_backend(pid) from pg_stat_activity where usename=current_user and pid != pg_backend_pid() and datname=:dname)ZdnamezDROP DATABASE %s)�connectr r �executer �dictr )r r r r! r$ r$ r% � _pg_drop_db> s ��r* c C s ddgiS )N�prefixes� TEMPORARYr$ )r r r$ r$ r% �#_postgresql_temp_table_keyword_argsM s r- c C s4 |j }d|_ |�� }|�d| � |�� ||_ d S )NTz SET SESSION search_path='%s')Z autocommit�cursorr( �close)r Zdbapi_connectionZ schema_nameZexisting_autocommitr. r$ r$ r% �,_postgresql_set_default_schema_on_connectionR s r0 c C sB |� � jdd��(}|�d��� D ]}|�d| � q W 5 Q R X d S )Nr r z!select gid from pg_prepared_xactszROLLBACK PREPARED '%s')r' r r Zscalars)r r r! �xidr$ r$ r% r ^ s � r c C s^ ddl m} t|�}|�� �8}|�d�D ]&}|�|�|j|d |d d��� q(W 5 Q R X d S )Nr )r �*�name�schema)r3 r4 )Zsqlalchemy.dialectsr r r Z get_enumsr( Z DropEnumTypeZENUM)r r r Z inspectorr! �enumr$ r$ r% r g s ��r c C s6 |� d�}|�� }|r2tdd�dd� |D �� � dS )z;Ensure there are no locks on the current username/database.z�select pid, state, wait_event_type, query from pg_stat_activity where usename=current_user and datname=current_database() and state='idle in transaction' and pid != pg_backend_pid()zHPostgreSQL may not be able to DROP tables due to idle in transaction: %sz; c s s | ]}|j d V qdS )�queryN)�_mapping)�.0�rowr$ r$ r% � <genexpr>� s z*prepare_for_drop_tables.<locals>.<genexpr>N)r �allr �join)�config� connection�result�rowsr$ r$ r% r u s ���r F)� set_lambda�sort_by_parameter_orderc C sV ddl m} ||�}t|�j}|r:|j|j||j�d�}n|�� }|j|d|i�}|S )Nr )�insert)Zindex_elements�set_rB ) Z sqlalchemy.dialects.postgresqlrC r Z selectableZon_conflict_do_updateZ primary_key�excludedZon_conflict_do_nothing� returning)r �tablerF rA rB rC �stmtZtable_pkr$ r$ r% �_upsert� s ���rI )Zcitext�� )ZhstorerJ c C sN |� � �<}tD ]0\}}|jj|kr|�td|� ��� |�� qW 5 Q R X d S )Nz CREATE EXTENSION IF NOT EXISTS )r' � _extensions�dialectZserver_version_infor( r �commit)r ZengineZfollower_identr! � extension� min_versionr$ r$ r% �_create_citext_extension� s �rQ )r � r r r Ztestingr Ztesting.provisionr r r r r r r r r r Zfor_dbr& r* r- r0 rI rL rQ r$ r$ r$ r% �<module> sJ " ��