from Schemas import BuildLivingSpace, People

class Similarity:

    def __init__(self, similarity: float, garbage: str, cleaned: str):
        self.similarity = similarity
        self.garbage = garbage
        self.cleaned = cleaned
        self.living_space = None
        self.living_space_id = None
        self.build_part_id = None
        self.company = None
        self.company_id = None
        self.found_from = None
        self.send_person_id = None
        self.customer_id = None
    
    def set_customer_id(self, customer_id: int):
        self.customer_id = customer_id
    
    def set_living_space(self, living_space: BuildLivingSpace):
        self.living_space = living_space
    
    def set_company(self, company: People):
        self.company = company
    
    def set_found_from(self, found_from: str):
        self.found_from = found_from
    
    def set_send_person_id(self, send_person_id: int):
        self.send_person_id = send_person_id
    
    def set_similarity(self, similarity: float):
        self.similarity = similarity
    
    def set_garbage(self, garbage: str):
        self.garbage = garbage
    
    def set_cleaned(self, cleaned: str):
        self.cleaned = cleaned
        
    def set_living_space_id(self, living_space_id: int):
        self.living_space_id = living_space_id
        
    def set_build_part_id(self, build_part_id: int):
        self.build_part_id = build_part_id
        
    def set_company_id(self, company_id: int):
        self.company_id = company_id