from app.main.data_model.games.finedge.fe_asset_class import FEAssetClass


class FEAssetClassRepository:
    def __init__(self, session):
        self.session = session

    def get_asset_class_by_id(self, asset_class_id):
        asset_class = self.session.query(FEAssetClass).filter(FEAssetClass.id == asset_class_id).all()
        if len(asset_class) == 0:
            return None
        return asset_class[0]

    def get_asset_classes(self):
        asset_classes = self.session.query(FEAssetClass).all()
        if len(asset_classes) == 0:
            return None
        return asset_classes