diff --git a/aula152.py b/aula152.py new file mode 100644 index 0000000..f59292d --- /dev/null +++ b/aula152.py @@ -0,0 +1,54 @@ +# Funções decoradoras e decoradores com métodos + +def meu_repr(self): + class_name = self.__class__.__name__ + class_dict = self.__dict__ + class_repr = f'{class_name}({class_dict})' + return class_repr + + +def adiciona_repr(cls): + cls.__repr__ = meu_repr + return cls + + +def meu_planeta(metodo): + def interno(self, *args, **kwargs): + resultado = metodo(self, *args, **kwargs) + + if 'Terra' in resultado: + return 'Você está em casa' + return resultado + return interno + + +@adiciona_repr +class Time: + def __init__(self, nome): + self.nome = nome + + +@adiciona_repr +class Planeta: + def __init__(self, nome): + self.nome = nome + + @meu_planeta + def falar_nome(self): + return f'O planeta é {self.nome}' + + +brasil = Time('Brasil') +portugal = Time('Portugal') + +terra = Planeta('Terra') +marte = Planeta('Marte') + +print(brasil) +print(portugal) + +print(terra) +print(marte) + +print(terra.falar_nome()) +print(marte.falar_nome())