battery.py [download]

class Battery:

    def __init__(self, capacity):
        self.mCapacity = capacity
        self.mCharge = capacity
        return

    def getCapacity(self):
        return self.mCapacity

    def getCharge(self):
        return self.mCharge

    def recharge(self, amount):
        if amount > 0 and self.mCharge < self.mCapacity:
            self.mCharge += amount
            if self.mCharge > self.mCapacity:
                self.mCharge = self.mCapacity
            return True
        return False

    def drain(self, amount):
        if amount > 0 and self.mCharge > 0:
            self.mCharge -= amount
            if self.mCharge < 0:
                self.mCharge = 0
            return True
        return False

Last Updated 08/24/2023