o
    6i                     @   s@   d dl Z d dlmZ d dlmZ d dlmZ G dd deZdS )    N)
errorcodes)BaseDatabaseCreation)strip_quotesc                       sJ   e Zd Zdd ZdddZdd Zdd	 Zd fdd	ZdddZ  Z	S )DatabaseCreationc                 C   s   | j j|S )N)
connectionops
quote_name)selfname r   k/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-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 jd }|d d u sJ 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   | dt|g | d uS )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 rW d S t ||| W d S  tyL } z(t|jddtjkr7| d|  t	
d n
|s: W Y d }~d S W Y d }~d S d }~ww )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 jd }| |d }| || j|dd}|  c}z	| ||| W nH tyt   z |dkrD| 	d| 
||f  |d|  | ||| W n tyq } z| 	d|  td W Y d }~nd }~ww Y n	w W d    d S W d    d S 1 sw   Y  d S )	NNAME)r   )r"   r      z,Destroying old test database for alias %s...zDROP 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
    