from flask_sqlalchemy import BaseQuery

from atlas_um import pgdb
from .mixins import DeletedMixin


class ApplicationQuery(BaseQuery, DeletedMixin):
    def by_dna_account(self, dna_account):
        return self.filter(pgdb.Application.dna_account == dna_account)

    def by_id(self, id):
        return self.filter(pgdb.Application.id == id)

    def by_identities(self, client_id, client_secret, token_audience):
        return self.join(pgdb.ResourceGroup).filter(
            pgdb.Application.client_id == client_id,
            pgdb.Application.client_secret == client_secret,
            pgdb.Application.token_audience == token_audience,
        )