o
    voi                     @  s   d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ ddl	m
Z
 erBd d	lmZ dd
lmZ ddlmZ ddlmZ eeZdddZdddZdS )    )annotationsN)Optional)Set)TYPE_CHECKING)inspect   )PriorityDispatchResult)	Inspector)AutogenContext)
UpgradeOps)Pluginautogen_contextr
   upgrade_opsr   returnr   c                   s    j }|d us	J  jdd}t|}|jj}|r0t| }|d || |	d  nd h} fdd|D } jd usCJ  j
jd jjd || tjS )Ninclude_schemasFinformation_schemac                   s   h | ]}  |d i r|qS )schema)run_name_filters).0sr    W/home/ubuntu/.local/lib/python3.10/site-packages/alembic/autogenerate/compare/schema.py	<setcomp>.   s
    z'_produce_net_changes.<locals>.<setcomp>r   )	qualifier)
connectionoptsgetr   dialectdefault_schema_namesetget_schema_namesdiscardaddcomparatorsdispatchnamer   CONTINUE)r   r   r   r   	inspectordefault_schemaschemasr   r   r   _produce_net_changes   s*   


r+   pluginr   Nonec                 C  s   |  td d S )Nautogenerate)add_autogenerate_comparatorr+   )r,   r   r   r   setup:   s   r0   )r   r
   r   r   r   r   )r,   r   r   r-   )
__future__r   loggingtypingr   r   r   
sqlalchemyr   utilr   sqlalchemy.engine.reflectionr	   autogenerate.apir
   operations.opsr   runtime.pluginsr   	getLogger__name__logr+   r0   r   r   r   r   <module>   s   

!