import logging

from app import app
from data.model.repository import (find_repository_with_garbage, garbage_collect_repo,
                                   get_random_gc_policy)

from workers.worker import Worker

logger = logging.getLogger(__name__)

class GarbageCollectionWorker(Worker):
  def __init__(self):
    super(GarbageCollectionWorker, self).__init__()
    self.add_operation(self._garbage_collection_repos,
                       app.config.get('GARBAGE_COLLECTION_FREQUENCY', 30))

  def _garbage_collection_repos(self):
    """ Performs garbage collection on repositories. """
    repository = find_repository_with_garbage(get_random_gc_policy())
    if repository is None:
      logger.debug('No repository with garbage found')
      return

    logger.debug('Starting GC of repository #%s (%s)', repository.id, repository.name)
    garbage_collect_repo(repository)
    logger.debug('Finished GC of repository #%s (%s)', repository.id, repository.name)

if __name__ == "__main__":
  worker = GarbageCollectionWorker()
  worker.start()