
    j4                         d dl mZ d dlmZmZ d dlmZ defdZd	dZ	d	dZ
 ej                         r e	        y e
        y)
    )context)create_enginepool)get_settingsreturnc                  r    t               j                  } | j                  d      rd| t        d      d  z   } | S )Nzpostgresql://zpostgresql+psycopg://)r   database_url
startswithlen)urls    (/var/www/avy-guidance/api/alembic/env.py_urlr      s7    
.
%
%C
~~o&%C,@,B(CCJ    Nc                      t        j                  t               d       t        j                         5  t        j                          d d d        y # 1 sw Y   y xY w)NT)r   literal_binds)r   	configurer   begin_transactionrun_migrations r   r   run_migrations_offliner      sA    $&5		"	"	$ ! ! ! !s   AAc                  :   t        t               t        j                        } | j	                         5 }t        j                  |       t        j                         5  t        j                          d d d        d d d        y # 1 sw Y   xY w# 1 sw Y   y xY w)N)	poolclass)
connection)	r   r   r   NullPoolconnectr   r   r   r   )enginer   s     r   run_migrations_onliner      su    46T]];F		 %ZZ0&&( 	%""$	%% %	% 	%% %s#   +BB4BB	
BB)r   N)alembicr   
sqlalchemyr   r   app.settingsr   strr   r   r   is_offline_moder   r   r   <module>r#      s@     * %c !% 7r   