OwlCyberSecurity - MANAGER
Edit File: questioner.cpython-39.pyc
a ���e4 � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl m Z d dl m Z ddlmZ G d d � d �ZG d d � d e�ZG d d� de�ZdS )� N)�apps)� OutputWrapper)� NOT_PROVIDED)�timezone)�get_docs_version� )�MigrationLoaderc @ sZ e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�MigrationQuestionerz� Give the autodetector responses to questions it might have. This base class has a built-in noninteractive mode, but the interactive subclass is what the command-line arguments will use. Nc C s |pi | _ |pt� | _|| _d S )N)�defaults�set�specified_apps�dry_run)�selfr r r � r �`/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/migrations/questioner.py�__init__ s z MigrationQuestioner.__init__c C s� || j v rdS zt�|�}W n ty< | j�dd� Y S 0 t�|j�\}}|du rd| j�dd�S zt � |�}W n t y� | j�dd� Y S 0 t |dd�r�t �t j�|j��}n0t|d�r�t|j�dkr�dS t �t|j�d �}td d � |D �� S dS ) z2Should we create an initial migration for the app?T� ask_initialFN�__file__�__path__r r c s s | ]}|d kr|� d�V qdS )z __init__.pyz.pyN)�endswith)�.0�xr r r � <genexpr>7 � z2MigrationQuestioner.ask_initial.<locals>.<genexpr>)r r Zget_app_config� LookupErrorr �getr �migrations_module�label� importlib� import_module� ImportError�getattr�os�listdir�path�dirnamer �hasattr�lenr �list�any)r � app_labelZ app_configZmigrations_import_path�_r � filenamesr r r r s( z MigrationQuestioner.ask_initialc C s dS )�#Adding a NOT NULL field to a model.Nr �r � field_name� model_namer r r �ask_not_null_addition9 s z)MigrationQuestioner.ask_not_null_additionc C s dS )�"Changing a NULL field to NOT NULL.Nr r. r r r �ask_not_null_alteration> s z+MigrationQuestioner.ask_not_null_alterationc C s | j �dd�S )� Was this field really renamed?� ask_renameF�r r )r r0 �old_name�new_name�field_instancer r r r5 C s z MigrationQuestioner.ask_renamec C s | j �dd�S )� Was this model really renamed?�ask_rename_modelFr6 )r �old_model_state�new_model_stater r r r; G s z$MigrationQuestioner.ask_rename_modelc C s | j �dd�S )z)Should these migrations really be merged?� ask_mergeFr6 �r r* r r r r> K s z MigrationQuestioner.ask_mergec C s dS )�(Adding an auto_now_add field to a model.Nr r. r r r �ask_auto_now_add_additionO s z-MigrationQuestioner.ask_auto_now_add_additionc C s dS )�.Adding a unique field with a callable default.Nr r. r r r �$ask_unique_callable_default_additionT s z8MigrationQuestioner.ask_unique_callable_default_addition)NNN) �__name__� __module__� __qualname__�__doc__r r r1 r3 r5 r; r> rA rC r r r r r s r c sr e Zd Zd� fdd� Zddd�Zdd� Zdd d �Zd d � Zd d� Zdd� Z dd� Z dd� Z dd� Z dd� Z � ZS ) � InteractiveMigrationQuestionerNc s& t � j|||d� |p ttj�| _d S �N)r r r )�superr r �sys�stdout� prompt_output)r r r r rM �� __class__r r r [ s �z'InteractiveMigrationQuestioner.__init__c C sh | j j|� d�dd� t� }|s,|d ur,|S |r@|d �� dvrX| j jddd� t� }q,|d �� dkS )N� � ��endingr ZynzPlease answer yes or no: �y)rM �write�input�lower)r �question�default�resultr r r �_boolean_inputc s z-InteractiveMigrationQuestioner._boolean_inputc C s� | j �|� � t|�D ] \}}| j �d|d |f � q| j jddd� t� }z t|�}W n tyl Y n"0 d| k r�t|�kr�n n|S | j jddd� t� }qNd S )Nz %s) %sr zSelect an option: rQ rR r z Please select a valid option: )rM rU � enumeraterV �int� ValueErrorr' )r rX �choices�i�choicerZ �valuer r r � _choice_inputm s z,InteractiveMigrationQuestioner._choice_inputrQ c C s� | j �d� |r$| j �d|� d�� | j �d� | j �d� |rLd�|�}nd}| j j|dd � t� }|sr|rr|}|s�| j �d � q<|d kr�t�d � q<zt|i ttd ��W S t t fy� } z | j �d| � W Y d}~q<d}~0 0 q<dS )z� Prompt for a default value. The ``default`` argument allows providing a custom default value (as a string) which will be shown to the user and used as the return value if the user doesn't provide any other input. z/Please enter the default value as valid Python.zAccept the default 'z/' by pressing 'Enter' or provide another value.zxThe datetime and django.utils.timezone modules are available, so it is possible to provide e.g. timezone.now as a value.z Type 'exit' to exit this promptz[default: {}] >>> z>>> rQ rR z;Please enter some code, or 'exit' (without quotes) to exit.�exitr )�datetimer zInvalid input: %sN) rM rU �formatrV rK rd �evalre r � SyntaxError� NameError)r rY �prompt�code�er r r � _ask_default~ s4 �� � z+InteractiveMigrationQuestioner._ask_defaultc C sD | j s@| �d|� d|� d�ddg�}|dkr8t�d� n| �� S dS ) r- z.It is impossible to add a non-nullable field 'z' to z{ without specifying a default. This is because the database needs something to populate existing rows. Please select a fix:�bProvide a one-off default now (will be set on all existing rows with a null value for this column)�6Quit and manually define a default value in models.py.� � N�r rc rK rd rm �r r/ r0 ra r r r r1 � s ��� z4InteractiveMigrationQuestioner.ask_not_null_additionc C sP | j sL| �d|� d|� d�g d��}|dkr0tS |dkrDt�d� n| �� S dS )r2 z-It is impossible to change a nullable field 'z' on z� to non-nullable without providing a default. This is because the database needs something to populate existing rows. Please select a fix:)rn z�Ignore for now. Existing rows that contain NULL values will have to be handled manually, for example with a RunPython or RunSQL operation.ro rp rq N)r rc r rK rd rm rs r r r r3 � s �� z6InteractiveMigrationQuestioner.ask_not_null_alterationc C s" d}| � ||||||jjf d�S )r4 z(Was %s.%s renamed to %s.%s (a %s)? [y/N]F)r[ rO rD )r r0 r7 r8 r9 �msgr r r r5 � s ���z)InteractiveMigrationQuestioner.ask_renamec C s d}| � ||j|j|jf d�S )r: z(Was the model %s.%s renamed to %s? [y/N]F)r[ r* �name)r r< r= rt r r r r; � s ��z/InteractiveMigrationQuestioner.ask_rename_modelc C s | � dd�S )Nz� Merging will only work if the operations printed above do not conflict with each other (working on different fields or models) Should these migration branches be merged? [y/N]F)r[ r? r r r r> � s �z(InteractiveMigrationQuestioner.ask_mergec C sH | j sD| �d|� d|� d�ddg�}|dkr8t�d� n | jdd �S d S ) r@ z#It is impossible to add the field 'z ' with 'auto_now_add=True' to zf without providing a default. This is because the database needs something to populate existing rows. zDProvide a one-off default now which will be set on all existing rowsro rp rq z timezone.now)rY Nrr rs r r r rA � s ��� z8InteractiveMigrationQuestioner.ask_auto_now_add_additionc C sH | j sDt� }| �d|� d|� d�d|� d�dg�}|dkrDt�d� d S ) rB z!Callable default on unique field �.zO will not generate unique values upon migrating. Please choose how to proceed: z�Continue making this migration as the first step in writing a manual migration to generate unique values described here: https://docs.djangoproject.com/en/z=/howto/writing-migrations/#migrations-that-add-unique-fields.z)Quit and edit field options in models.py.rp rq N)r r rc rK rd )r r/ r0 �versionra r r r rC s ��� zCInteractiveMigrationQuestioner.ask_unique_callable_default_addition)NNNN)N)rQ )rD rE rF r r[ rc rm r1 r3 r5 r; r> rA rC � __classcell__r r rN r rH Z s � ( rH c s> e Zd Zd � fdd� Zdd� Zdd� Zd d � Zd d � Z� ZS )�!NonInteractiveMigrationQuestionerNr c s" || _ || _t� j|||d� d S rI )� verbosity�logrJ r )r r r r rz r{ rN r r r $ s �z*NonInteractiveMigrationQuestioner.__init__c C s, | j dkr(| �d|� d|� d|� d�� d S )Nr �Field '� ' on model 'z' not migrated: rv )rz r{ )r r/ r0 �reasonr r r �log_lack_of_migration4 s ��z7NonInteractiveMigrationQuestioner.log_lack_of_migrationc C s | � ||d� t�d� d S )NzIit is impossible to add a non-nullable field without specifying a defaultrq �r rK rd r. r r r r1 ; s �z7NonInteractiveMigrationQuestioner.ask_not_null_additionc C s | � d|� d|� d�� tS )Nr| r} z8' given a default of NOT PROVIDED and must be corrected.)r{ r r. r r r r3 E s �z9NonInteractiveMigrationQuestioner.ask_not_null_alterationc C s | � ||d� t�d� d S )NzUit is impossible to add a field with 'auto_now_add=True' without specifying a defaultrq r� r. r r r rA M s �z;NonInteractiveMigrationQuestioner.ask_auto_now_add_addition)NNNr N) rD rE rF r r r1 r3 rA rx r r rN r ry # s � ry )re r r" rK Z django.appsr Zdjango.core.management.baser Zdjango.db.modelsr Z django.utilsr Zdjango.utils.versionr �loaderr r rH ry r r r r �<module> s K J