from app.main.data_model.common.category import Category


class CategoryRepository:

    def __init__(self, session):
        self.session = session
    
    def get_categories(self):
        categories = self.session.query(Category).all()
        return categories