Personalizando funcionalidades via herança

Introdução à programação orientada a objetos em Python

George Boorman

Curriculum Manager, DataCamp

Uma hierarquia de classes de contas bancárias

O que temos até agora

class BankAccount:
    def __init__(self, balance):
       self.balance = balance

    def withdraw(self, amount):
       self.balance -=amount

# Classe vazia herdada de BankAccount
class SavingsAccount(BankAccount):
    pass

Personalizando construtores

class SavingsAccount(BankAccount):    
    # Construtor de SavingsAccount com argumento extra
    def __init__(self, balance, interest_rate):  

        # Chama o construtor do pai usando ClassName.__init__()
        # self é SavingsAccount e também BankAccount
        BankAccount.__init__(self, balance)

        # Adiciona mais funcionalidade
        self.interest_rate = interest_rate

Pode chamar primeiro o construtor do pai com Parent.__init__(self, args...)
Adicionar mais funcionalidades
Não é obrigatório chamar o construtor do pai

Criar objetos com construtor personalizado

# Constrói o objeto usando o novo construtor
acct = SavingsAccount(1000, 0.03)
acct.interest_rate

0.03

Adicionando funcionalidades

Adicione métodos como de costume
Pode usar dados tanto da classe pai quanto da filha

class SavingsAccount(BankAccount):
    def __init__(self, balance, interest_rate):
        BankAccount.__init__(self, balance)
        self.interest_rate = interest_rate

    # Nova funcionalidade
    def compute_interest(self, n_periods=1):
        return self.balance * ( (1 + self.interest_rate) ** n_periods - 1)

Uma hierarquia de classes de contas bancárias

Adicionando uma segunda classe filha

class CheckingAccount(BankAccount):

    def __init__(self, balance, limit):
        BankAccount.__init__(self, balance) # Chama o construtor da classe pai
        self.limit = limit

    def deposit(self, amount):
        self.balance += amount

    def withdraw(self, amount, fee=0): # Novo argumento fee

        if amount <= self.limit:
            BankAccount.withdraw(self, amount + fee)  
        else:
            pass # Não executa se a condição não for atendida

check_acct = CheckingAccount(1000, 25)

# Vai chamar withdraw de CheckingAccount
check_acct.withdraw(200)


# Vai chamar withdraw de CheckingAccount
check_acct.withdraw(200, fee=15)

bank_acct = BankAccount(1000)

# Vai chamar withdraw de BankAccount
bank_acct.withdraw(200)


# Vai gerar um erro
bank_acct.withdraw(200, fee=15)

TypeError: withdraw() got an unexpected
keyword argument 'fee'

Viola o polimorfismo
- Classes pai/filha têm métodos diferentes

Vamos praticar!

Introdução à programação orientada a objetos em Python