o
    ^"<f  ã                   @   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   ú]/var/www/html/kck/venv/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 sˆw   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
    