from db import db
from workers.base_worker import BaseWorker
from utils.elastic_search.indexer import ElasticSearchIndexer
from models.projects import Project
from models.campaign import Campaign
from models.prs_purchase_order import PRSPurchaseOrder


class PRSElasticIndexerWorker(BaseWorker):
    def execute(self, *args):
        indexer = ElasticSearchIndexer(models=[Project, Campaign, PRSPurchaseOrder])
        indexer.reindex_database()
        db.session.close()