from ..connection import DatabaseConnection

class ServiceAvailability:
    def __init__(self, member_id: int, service_type_id: int):
        self.member_id = member_id
        self.service_type_id = service_type_id

    def save(self, db: DatabaseConnection):
        query = "INSERT INTO ServiceAvailability (MemberId, ServiceTypeId) VALUES (?, ?)"
        db.execute_query(query, (self.member_id, self.service_type_id))

    @classmethod
    def get_all(cls, db: DatabaseConnection):
        query = "SELECT * FROM ServiceAvailability"
        return db.execute_query_with_return(query)