OwlCyberSecurity - MANAGER
Edit File: creation.cpython-39.pyc
a ���ebQ � @ s\ d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z dZ G dd � d e�Z dS ) � N)�settings)� DatabaseError)�BaseDatabaseCreation)�get_random_string)�cached_propertyZtest_c @ s e Zd Zedd� �ZdDdd�Zdd� Zd d � Zd d � ZdEd d�Z dFdd�Z dGdd�Z dd� Z dd� Z dHdd�Zdd� Zdd � ZdId d �Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#d S )J�DatabaseCreationc C s^ t j| jj }|�d�p |d }|�d�p0|d }i |�||d��}t| j�}||| jjd�S )a> This is analogous to other backends' `_nodb_connection` property, which allows access to an "administrative" connection which can be used to manage the test databases. For Oracle, the only connection that can be used for that purpose is the main (non-test) connection. � SAVED_USER�USER�SAVED_PASSWORD�PASSWORD)r r )�alias)r � DATABASES� connectionr �get�type)�self� settings_dict�user�passwordZDatabaseWrapper� r �c/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/backends/oracle/creation.py�_maindb_connection s z#DatabaseCreation._maindb_connection� Fc C s^ | � � }| j�� ��}| �� �r�z| �||||� W �n� t�y� } �z�dt|�vrn| �d| � t� d� |s�t d|d �}|s�|dk�r�|dkr�| �d| j j � z| � |||� W n� t�y } zBd t|�v r�| �||||� n| �d | � t� d� W Y d }~nFd }~0 t�yT } z$| �d | � t� d� W Y d }~n d }~0 0 z| �||||� W n> t�y� } z$| �d | � t� d� W Y d }~n d }~0 0 n| �d � t� d� W Y d }~n d }~0 0 | �� �r |dk�r�| �d � z| �||||� W �n t�y } z�dt|�v�rB| �d| � t� d� |�sXt d|d �}|�sh|dk�r�zJ|dk�r~| �d� | �|||� |dk�r�| �d � | �||||� W n> t�y� } z$| �d| � t� d� W Y d }~n d }~0 0 n| �d � t� d� W Y d }~n d }~0 0 W d � n1 �s40 Y | j�� | �|� | j jd S )N� ORA-01543z+Got an error creating the test database: %s� z^It appears the test database, %s, already exists. Type 'yes' to delete it, or 'no' to cancel: r �yesr �.Destroying old test database for alias '%s'...z ORA-29857z1Got an error destroying the old test database: %sz-Got an error recreating the test database: %szTests cancelled.zCreating test user...� ORA-01920z'Got an error creating the test user: %szZIt appears the test user, %s, already exists. Type 'yes' to delete it, or 'no' to cancel: �Destroying old test user...z)Got an error recreating the test user: %s�NAME)�_get_test_db_paramsr �cursor�_test_database_create�_execute_test_db_creation� Exception�str�log�sys�exit�inputr r � _execute_test_db_destructionr �)_handle_objects_preventing_db_destruction�_test_user_create�_create_test_user�_destroy_test_user�close�_switch_to_test_userr )r � verbosity� autoclobber�keepdb� parametersr! �e�confirmr r r �_create_test_db s� � ����� ��� � ��" �� �" @ z DatabaseCreation._create_test_dbc C s� t j| jj }| jjd |d< | jjd< | jjd |d< | jjd< |d }| jjd }|d |d< |d< |d< | jjd< |d |d< | jjd< dS ) a� Switch to the user that's used for creating the test database. Oracle doesn't have the concept of separate databases under the same user, so a separate user is used; see _create_test_db(). The main user is also needed for cleanup when testing is completed, so save its credentials in the SAVED_USER/SAVED_PASSWORD key in the settings dict. r r r r �TESTr r N)r r r r r )r r4 Z real_settingsZreal_test_settingsZ test_settingsr r r r0 ~ s( �� �� �� ��z%DatabaseCreation._switch_to_test_userc C s$ |d | j jd< |d | j jd< dS )z� Set this database up to be used in testing as a mirror of a primary database whose settings are given. r r N�r r )r Zprimary_settings_dictr r r �set_as_test_mirror� s z#DatabaseCreation.set_as_test_mirrorc C sF | � d� | �� �r |s(td|d �}|s6|dk�rz$|dkrJ| � d� | �|||� W n< ty� } z$| � d| � t�d� W Y d }~n d }~0 0 z,|dkr�| � d | jj � | � |||� W n> t�y } z$| � d | � t�d� W Y d }~n d }~0 0 n| � d � t�d� n&| � d |d � | � d � t�d� d S ) Na There are objects in the old test database which prevent its destruction. If they belong to the test user, deleting the user will allow the test database to be recreated. Otherwise, you will need to find and remove each of these objects, or use a different tablespace. z Type 'yes' to delete user %s: r r r r z)Got an error destroying the test user: %sr r z-Got an error destroying the test database: %sz5Tests cancelled -- test database cannot be recreated.z[Django is configured to use pre-existing test user '%s', and will not attempt to delete it.) r& r, r) r. r$ r' r( r r r* )r r! r4 r1 r2 r6 r5 r r r r+ � sD � ��" �� z:DatabaseCreation._handle_objects_preventing_db_destructionc C s� | j jd | j jd< | j jd | j jd< | j �� | �� }| j�� �`}| �� rn|dkr`| �d� | �|||� | � � r�|dkr�| �d� | � |||� W d � n1 s�0 Y | j�� dS ) z� Destroy a test database, prompting the user for confirmation if the database already exists. Return the name of the test database created. r r r r r zDestroying test user...z"Destroying test database tables...N) r r r/ r r r! r, r& r. r" r* )r Ztest_database_namer1 r4 r! r r r �_destroy_test_db� s$ � � ,z!DatabaseCreation._destroy_test_dbc C sV |dkr| � d|d � | �� r,ddg}nddg}|r<dnd }| �|||||� d S ) Nr z _create_test_db(): dbname = %sr z� CREATE TABLESPACE %(tblspace)s DATAFILE SIZE %(size)s AUTOEXTEND ON NEXT %(extsize)s MAXSIZE %(maxsize)s z� CREATE TEMPORARY TABLESPACE %(tblspace_temp)s TEMPFILE SIZE %(size_tmp)s AUTOEXTEND ON NEXT %(extsize_tmp)s MAXSIZE %(maxsize_tmp)s z� CREATE TABLESPACE %(tblspace)s DATAFILE '%(datafile)s' SIZE %(size)s REUSE AUTOEXTEND ON NEXT %(extsize)s MAXSIZE %(maxsize)s z� CREATE TEMPORARY TABLESPACE %(tblspace_temp)s TEMPFILE '%(datafile_tmp)s' SIZE %(size_tmp)s REUSE AUTOEXTEND ON NEXT %(extsize_tmp)s MAXSIZE %(maxsize_tmp)s r )r& �#_test_database_oracle_managed_files� _execute_allow_fail_statements)r r! r4 r1 r3 � statements�acceptable_ora_errr r r r# � s �� �z*DatabaseCreation._execute_test_db_creationc C s� |dkr| � d|d � ddg}|r*dnd }| �|||||�}|sh| �d�d u rhd}| �||g||� d D ]>} d } | |d < | �|| g||d �}|sl|dkrl| � d | � qld S )Nr z"_create_test_user(): username = %sr z�CREATE USER %(user)s IDENTIFIED BY "%(password)s" DEFAULT TABLESPACE %(tblspace)s TEMPORARY TABLESPACE %(tblspace_temp)s QUOTA UNLIMITED ON %(tblspace)s z�GRANT CREATE SESSION, CREATE TABLE, CREATE SEQUENCE, CREATE PROCEDURE, CREATE TRIGGER TO %(user)sr r z0ALTER USER %(user)s IDENTIFIED BY "%(password)s")ZVIEWzMATERIALIZED VIEWz(GRANT CREATE %(object_type)s TO %(user)s� object_typez ORA-01031zBFailed to grant CREATE %s permission to test user. This may be ok.)r& r= �_test_settings_get�_execute_statements) r r! r4 r1 r3 r> r? �success� set_passwordr@ �extrar r r r- s0 � � � ��z"DatabaseCreation._create_test_userc C s6 |dkr| � d|d � ddg}| �||||� d S )Nr z)_execute_test_db_destruction(): dbname=%sr zQDROP TABLESPACE %(tblspace)s INCLUDING CONTENTS AND DATAFILES CASCADE CONSTRAINTSzVDROP TABLESPACE %(tblspace_temp)s INCLUDING CONTENTS AND DATAFILES CASCADE CONSTRAINTS�r& rB �r r! r4 r1 r> r r r r* . s �z-DatabaseCreation._execute_test_db_destructionc C s> |dkr$| � d|d � | � d� dg}| �||||� d S )Nr z _destroy_test_user(): user=%sr z&Be patient. This can take some time...zDROP USER %(user)s CASCADErF rG r r r r. 9 s �z#DatabaseCreation._destroy_test_userc C sv |D ]l}|| }|dkr t |� z|�|� W q tyn } z(|rJ|dkrX| �d| � � W Y d }~qd }~0 0 qd S )Nr z Failed (%s))�print�executer$ r& ) r r! r> r4 r1 �allow_quiet_fail�template�stmt�errr r r rB B s z$DatabaseCreation._execute_statementsc C st z.|duot |�dk}| j|||||d� W dS tyn } z(t|�}|du sV||vrX� W Y d}~dS d}~0 0 dS )z� Execute statements which are allowed to fail silently if the Oracle error code given by `acceptable_ora_err` is raised. Return True if the statements execute without an exception, or False otherwise. Nr )rJ TF)�lenrB r r% ) r r! r> r4 r1 r? rJ rM � descriptionr r r r= P s ��z/DatabaseCreation._execute_allow_fail_statementsc C sT | � � | �� | �� | �� | �� | �� | �� | �� | �� | � � | � � | � � | � � d� S )N) Zdbnamer r �tblspaceZ tblspace_tempZdatafileZ datafile_tmp�maxsizeZ maxsize_tmp�sizeZsize_tmpZextsizeZ extsize_tmp) �_test_database_name�_test_database_user�_test_database_passwd�_test_database_tblspace�_test_database_tblspace_tmp� _test_database_tblspace_datafile�$_test_database_tblspace_tmp_datafile� _test_database_tblspace_maxsize�#_test_database_tblspace_tmp_maxsize� _test_database_tblspace_size� _test_database_tblspace_tmp_size� _test_database_tblspace_extsize�#_test_database_tblspace_tmp_extsize�r r r r r k s �z$DatabaseCreation._get_test_db_paramsNc C s4 | j j}|d �||�}|du r0|r0t|| }|S )z� Return a value from the test settings dict, or a given default, or a prefixed entry from the main settings dict. r8 N�r r r �TEST_DATABASE_PREFIX)r �key�default�prefixedr �valr r r rA | s z#DatabaseCreation._test_settings_getc C s | j ddd�S )Nr �re �rA r` r r r rS � s z$DatabaseCreation._test_database_namec C s | j ddd�S )NZ CREATE_DBT�rd rh r` r r r r" � s z&DatabaseCreation._test_database_createc C s | j ddd�S )NZ CREATE_USERTri rh r` r r r r, � s z"DatabaseCreation._test_user_createc C s | j ddd�S )Nr rg rh r` r r r rT � s z$DatabaseCreation._test_database_userc C s& | � d�}|d u r"| �� r"td�}|S )Nr � )rA r, r )r r r r r rU � s z&DatabaseCreation._test_database_passwdc C s | j ddd�S )NZTBLSPACEr rg rh r` r r r rV � s z(DatabaseCreation._test_database_tblspacec C s$ | j j}|d �dt|d d �S )Nr8 Z TBLSPACE_TMPr �_tempra �r r r r r rW � s �z,DatabaseCreation._test_database_tblspace_tmpc C s d| � � }| jd|d�S )N�%s.dbfZDATAFILEri )rV rA �r rP r r r rX � s z1DatabaseCreation._test_database_tblspace_datafilec C s d| � � }| jd|d�S )Nrm Z DATAFILE_TMPri )rW rA rn r r r rY � s z5DatabaseCreation._test_database_tblspace_tmp_datafilec C s | j ddd�S )NZDATAFILE_MAXSIZE�500Mri rh r` r r r rZ � s z0DatabaseCreation._test_database_tblspace_maxsizec C s | j ddd�S )NZDATAFILE_TMP_MAXSIZEro ri rh r` r r r r[ � s z4DatabaseCreation._test_database_tblspace_tmp_maxsizec C s | j ddd�S )NZ DATAFILE_SIZE�50Mri rh r` r r r r\ � s z-DatabaseCreation._test_database_tblspace_sizec C s | j ddd�S )NZDATAFILE_TMP_SIZErp ri rh r` r r r r] � s z1DatabaseCreation._test_database_tblspace_tmp_sizec C s | j ddd�S )NZDATAFILE_EXTSIZE�25Mri rh r` r r r r^ � s z0DatabaseCreation._test_database_tblspace_extsizec C s | j ddd�S )NZDATAFILE_TMP_EXTSIZErq ri rh r` r r r r_ � s z4DatabaseCreation._test_database_tblspace_tmp_extsizec C s | j ddd�S )NZORACLE_MANAGED_FILESFri rh r` r r r r< � s z4DatabaseCreation._test_database_oracle_managed_filesc C s | j jd S )z� Return the 'production' DB name to get the test DB creation machinery to work. This isn't a great deal in this case because DB names as handled by Django don't have real counterparts in Oracle. r r9 r` r r r �_get_test_db_name� s z"DatabaseCreation._get_test_db_namec C s* | j j}|d |d |d |d | �� fS )NZHOSTZPORTZENGINEr )r r rT rl r r r �test_db_signature� s �z"DatabaseCreation.test_db_signature)r FF)r )F)F)F)NN)$�__name__� __module__� __qualname__r r r7 r0 r: r+ r; r# r- r* r. rB r= r rA rS r" r, rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r< rr rs r r r r r sD a, # ( � r ) r' Z django.confr Z django.dbr Z django.db.backends.base.creationr Zdjango.utils.cryptor Zdjango.utils.functionalr rb r r r r r �<module> s