File: //lib/python3/dist-packages/django/db/backends/postgresql/__pycache__/creation.cpython-310.pyc
o
3�a
� @ s@ d dl Z d dlmZ d dlmZ d dlmZ G dd� de�ZdS )� N)�
errorcodes)�BaseDatabaseCreation)� strip_quotesc sJ e Zd Zdd� Zddd�Zdd� Zdd � Zd� fd d � Zdd
d�Z� Z S )�DatabaseCreationc C s | j j�|�S )N)�
connection�ops�
quote_name)�self�name� r �H/usr/lib/python3/dist-packages/django/db/backends/postgresql/creation.py� _quote_name s z DatabaseCreation._quote_nameNc C s: d}|r |d� |�7 }|r|d� | �|��7 }|o d| S )N� z ENCODING '{}'z TEMPLATE {}�WITH)�formatr
)r �encoding�template�suffixr r r �_get_database_create_suffix s z,DatabaseCreation._get_database_create_suffixc C s8 | j jd }|d d u sJ d��| j|d |�d�d�S )N�TEST� COLLATIONzHPostgreSQL does not support collation setting at database creation time.�CHARSET�TEMPLATE)r r )r �
settings_dictr �get)r �
test_settingsr r r �sql_table_creation_suffix s ��z*DatabaseCreation.sql_table_creation_suffixc C s |� dt|�g� |�� d uS )Nz7SELECT 1 FROM pg_catalog.pg_database WHERE datname = %s)�executer �fetchone)r �cursor�
database_namer r r �_database_exists s z!DatabaseCreation._database_existsFc
s� z|r| � ||d �rW d S t� �|||� W d S tyL } z(t|jdd�tjkr7| �d| � t �
d� n
|s:� W Y d }~d S W Y d }~d S d }~ww )N�dbname�pgcoder z+Got an error creating the test database: %s� ) r! �super�_execute_create_test_db� Exception�getattr� __cause__r �DUPLICATE_DATABASE�log�sys�exit)r r �
parameters�keepdb�e�� __class__r r r& $ s ����z(DatabaseCreation._execute_create_test_dbc C s | j �� | j jd }| �|�d }| �|�| j|d�d�}| �� �c}z | �|||� W nH tyt z |dkrD| � d| �
||�f � |� d| � | �|||� W n tyq } z| � d| � t �
d� W Y d }~nd }~ww Y n w W d � d S W d � d S 1 s�w Y d S ) N�NAME)r )r" r � z,Destroying old test database for alias %s...zDROP DATABASE %(dbname)sz*Got an error cloning the test database: %sr$ )r �closer �get_test_db_clone_settingsr
r � _nodb_cursorr& r' r+ �_get_database_display_strr r, r- ) r r � verbosityr/ �source_database_name�target_database_name�test_db_paramsr r0 r r r �_clone_test_db5 s8
�
��������" z DatabaseCreation._clone_test_db)NN)F)
�__name__�
__module__� __qualname__r
r r r! r& r= �
__classcell__r r r1 r r s
r )r, �psycopg2r � django.db.backends.base.creationr �django.db.backends.utilsr r r r r r �<module> s