o
    QDih                     @   sh   d dl mZ d dlmZ d dlmZ d dlmZ edddgedZed	d
dgedZ	dd Z
dd ZdS )    )connections)DatabaseWrapper)Gauge)	NAMESPACE!django_migrations_unapplied_totalz4Count of unapplied migrations by database connection
connection)	namespacedjango_migrations_applied_totalz2Count of applied migrations by database connectionc                 C   sB   | |jj }t| t| t| t|jj d S )N)	migration_planloadergraph
leaf_nodesunapplied_migrationslabelssetlenapplied_migrations)aliasexecutorplan r   U/var/www/Datamplify/venv/lib/python3.10/site-packages/django_prometheus/migrations.pyExportMigrationsForDatabase   s   r   c                  C   sL   ddl m}  dtv rttd trdS tjD ]}| t| }t|| qdS )zExports counts of unapplied migrations.

    This is meant to be called during app startup, ideally by
    django_prometheus.apps.AppConfig.
    r   )MigrationExecutordefaultN)django.db.migrations.executorr   r   
isinstancer   	databasesr   )r   r   r   r   r   r   ExportMigrations   s   
r   N)	django.dbr   django.db.backends.dummy.baser   prometheus_clientr   django_prometheus.confr   r   r   r   r   r   r   r   r   <module>   s$    