o
    h                     @   sB   d dl mZ d dlmZ d dlmZ d dlmZ G dd dZdS )    
connection)	Migration)MigrationLoader)MigrateStreamDatac                   @   s0   e Zd ZdZg ZdZdddZ		dddZdS )MigrationTestMixinNc                 C   s4   t dd}t| j| jjd|p| j|d}|g|_|S )Ntest_migration*wagtail_streamfield_migration_toolkit_testcontent)app_name
model_name
field_nameoperations_and_block_pathsrevisions_from)r   r   r   model__name__ default_operation_and_block_path
operations)selfr   r   	migrationmigration_operation r   m/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/test/streamfield_migrations/testutils.pyinit_migration   s   z!MigrationTestMixin.init_migrationc                 C   sF   | j ||d}ttd}|  | }tj|jd}||| d S )N)r   r   r   )atomic)r   r   r   build_graphproject_stateschema_editorr   apply)r   r   r   r   loaderr   r   r   r   r   apply_migration   s   
z"MigrationTestMixin.apply_migration)NN)r   
__module____qualname__r   r   r   r   r    r   r   r   r   r      s    
r   N)		django.dbr   django.db.migrationsr   django.db.migrations.loaderr   +wagtail.blocks.migrations.migrate_operationr   r   r   r   r   r   <module>   s
    