СМС-рассылка

class Person:
    
    def __init__(self, name, oth,surname, numbook):
        self.name = name
        self.oth = oth
        self.surname = surname
        self.numbook = numbook
        
    def get_phone(self, key = 'private'): 
        return self.numbook.get(key)
        
    def get_name(self):
        return f'{self.surname} {self.name} {self.oth}'
 
    def get_work_phone(self):
        return self.get_phone('work')
    
    def get_sms_text(self):
        return f'Уважаемый {self.name} {self.oth}! \
Примите участие в нашем беспроигрышном конкурсе для физических лиц'
 
    
class Company:
    
    def __init__(self, name, type_, numbook, *arg):
        self.name = name
        self.type_ = type_ 
        self.numbook = numbook
        self.person = arg
        
    def get_phone(self, key = 'contact'):
        if key in self.numbook:
            return self.numbook.get(key)
        for person in self.person:
            number = person.get_work_phone()
            if number:
                return number
        
    def get_name(self):
        return f'{self.name}'
 
    def get_work_phone(self):
        return self.get_phone()
    
    def get_sms_text(self):
        return f'Для компании {self.name} есть супер предложение! \
Примите участие в нашем беспроигрышном конкурсе для {self.type_}'
 
 
def send_sms(*obj):
    for ob in obj:
        number = ob.get_phone()
        if number:
            print(f'Отправлено СМС на номер {number} \
с текстом: {ob.get_sms_text()}')
        else:
            print(f'Не удалось отправить сообщение абоненту: {ob.get_name()}')
 
if __name__ == '__main__':
    print('TEST 1  ___________________')
    person1 = Person("Ivan", "Ivanovich", "Ivanov", {"private": 123, "work": 456})
    person2 = Person("Ivan", "Petrovich", "Petrov", {"private": 789})
    person3 = Person("Ivan", "Petrovich", "Sidorov", {"work": 789})
    person4 = Person("John", "Unknown", "Doe", {})
    company1 = Company("Bell", "ООО", {"contact": 111}, person3, person4)
    company2 = Company("Cell", "АО", {"non_contact": 222}, person2, person3)
    company3 = Company("Dell", "Ltd", {"non_contact": 333}, person2, person4)
    send_sms(person1, person2, person3, person4, company1, company2, company3)
    print('TEST 2  ___________________')
    person1 = Person("Степан", "Петрович", "Джобсов", {"private": 555})
    person2 = Person("Боря", "Иванович", "Гейтсов", {"private": 777, "work": 888})
    person3 = Person("Семен", "Робертович", "Возняцкий", {"work": 789})
    person4 = Person("Леонид", "Арсенович", "Торвальдсон", {})
    company1 = Company("Яблочный комбинат", "ООО", {"contact": 111}, person1, person3)
    company2 = Company("ПластОкно", "АО", {"non_contact": 222}, person2)
    company3 = Company("Пингвинья ферма", "Ltd", {"non_contact": 333}, person4)
    send_sms(person1, person2, person3, person4, company1, company2, company3)

class Person:
    
    def __init__(self, name, otch, famil, nomera):
        self.name = name
        self.otch = otch
        self.famil = famil
        self.nomera = nomera
        
    def get_phone(self):
        if 'private' in self.nomera:
            return self.nomera['private']
        else:
            return None
    
    def get_name(self):
        return f'{self.famil} {self.name} {self.otch}'
    
    def get_work_phone(self):
        if 'work' in self.nomera:
            return self.nomera['work']
        else:
            return None
        
    def get_sms_text(self):
        return f'Уважаемый {self.name} {self.otch}! Примите участие в нашем беспроигрышном конкурсе для физических лиц'
    
    
class Company:
    
    def __init__(self, nazvanie, tip, telefone, *args):
        self.persons = []
        for i in args:
            self.persons.append(i)
        self.nazvanie = nazvanie
        self.tip = tip
        self.telefone = telefone
        
    def get_phone(self):
        n = 0
        if 'contact' in self.telefone:
            return self.telefone['contact']
        else:
            for i in self.persons:
                if i.get_work_phone() is not None and n < 1:
                    return i.get_work_phone()
                    n += 1
            if n == 0:
                return None
    
    def get_name(self):
        return self.nazvanie
    
    def get_sms_text(self):
        return f'Для компании {self.nazvanie} есть супер предложение! Примите ' \
               f'участие в нашем беспроигрышном конкурсе для {self.tip}'
    
 
def send_sms(*args):
    for i in args:
        if i.get_phone() is not None:
            print(f'Отправлено СМС на номер {i.get_phone()} с текстом: {i.get_sms_text()}')
        else:
            print(f'Не удалось отправить сообщение абоненту: {i.get_name()}')