Tùy biến chức năng qua kế thừa

Lập trình hướng đối tượng trong Python

Alex Yarosh

Content Quality Analyst @ DataCamp

một hệ phân cấp các lớp tài khoản

Những gì đã có

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

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

# Lớp trống kế thừa từ BankAccount
class SavingsAccount(BankAccount):
    pass

Tùy biến constructor

class SavingsAccount(BankAccount):    

    # Hàm khởi tạo riêng cho SavingsAccount với tham số bổ sung
    def __init__(self, balance, interest_rate):  

       # Gọi hàm khởi tạo của lớp cha bằng ClassName.__init__()
       BankAccount.__init__(self, balance)  # <--- self là SavingsAccount nhưng cũng là BankAccount

       # Bổ sung chức năng
       self.interest_rate = interest_rate

Có thể chạy constructor của lớp cha trước bằng Parent.__init__(self, args...)
Thêm chức năng
Không bắt buộc phải gọi constructor của lớp cha

Tạo đối tượng với constructor tùy biến

# Tạo đối tượng bằng constructor mới
acct = SavingsAccount(1000, 0.03)
acct.interest_rate

0.03

Bổ sung chức năng

Thêm phương thức như thường lệ
Có thể dùng dữ liệu từ cả lớp cha và lớp con

class SavingsAccount(BankAccount):    

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

  # Chức năng mới
  def compute_interest(self, n_periods = 1):
     return self.balance * ( (1 + self.interest_rate) ** n_periods - 1)

một hệ phân cấp các lớp tài khoản

Tùy biến chức năng

class CheckingAccount(BankAccount):

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

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

  def withdraw(self, amount, fee=0):

    if fee <= self.limit:
      BankAccount.withdraw(self, amount + fee)  
    else:
      BankAccount.withdraw(self, 
                           amount + self.limit)

Có thể đổi chữ ký (thêm tham số)
Dùng Parent.method(self, args...) để gọi phương thức từ lớp cha

check_acct = CheckingAccount(1000, 25)

# Sẽ gọi withdraw của CheckingAccount
check_acct.withdraw(200)

# Sẽ gọi withdraw của CheckingAccount
check_acct.withdraw(200, fee=15)

bank_acct = BankAccount(1000)

# Sẽ gọi withdraw của BankAccount
bank_acct.withdraw(200)

# Sẽ báo lỗi
bank_acct.withdraw(200, fee=15)

TypeError: withdraw() nhận đối số
 keyword không mong đợi 'fee'

Ayo berlatih!

Lập trình hướng đối tượng trong Python