Forgive me Danny
This commit is contained in:
Twentysix 2016-07-04 01:40:23 +02:00
parent 6b123b9299
commit 57cd3afec9

View File

@ -62,7 +62,7 @@ class Bank:
self._save_bank() self._save_bank()
return self.get_account(user) return self.get_account(user)
else: else:
raise AccountAlreadyExists raise AccountAlreadyExists()
def account_exists(self, user): def account_exists(self, user):
try: try:
@ -75,7 +75,7 @@ class Bank:
server = user.server server = user.server
if amount < 0: if amount < 0:
raise NegativeValue raise NegativeValue()
account = self._get_account(user) account = self._get_account(user)
if account["balance"] >= amount: if account["balance"] >= amount:
@ -83,12 +83,12 @@ class Bank:
self.accounts[server.id][user.id] = account self.accounts[server.id][user.id] = account
self._save_bank() self._save_bank()
else: else:
raise InsufficientBalance raise InsufficientBalance()
def deposit_credits(self, user, amount): def deposit_credits(self, user, amount):
server = user.server server = user.server
if amount < 0: if amount < 0:
raise NegativeValue raise NegativeValue()
account = self._get_account(user) account = self._get_account(user)
account["balance"] += amount account["balance"] += amount
self.accounts[server.id][user.id] = account self.accounts[server.id][user.id] = account
@ -97,7 +97,7 @@ class Bank:
def set_credits(self, user, amount): def set_credits(self, user, amount):
server = user.server server = user.server
if amount < 0: if amount < 0:
raise NegativeValue raise NegativeValue()
account = self._get_account(user) account = self._get_account(user)
account["balance"] = amount account["balance"] = amount
self.accounts[server.id][user.id] = account self.accounts[server.id][user.id] = account
@ -106,17 +106,17 @@ class Bank:
def transfer_credits(self, sender, receiver, amount): def transfer_credits(self, sender, receiver, amount):
server = sender.server server = sender.server
if amount < 0: if amount < 0:
raise NegativeValue raise NegativeValue()
if sender is receiver: if sender is receiver:
raise SameSenderAndReceiver raise SameSenderAndReceiver()
if self.account_exists(sender) and self.account_exists(receiver): if self.account_exists(sender) and self.account_exists(receiver):
sender_acc = self._get_account(sender) sender_acc = self._get_account(sender)
if sender_acc["balance"] < amount: if sender_acc["balance"] < amount:
raise InsufficientBalance raise InsufficientBalance()
self.withdraw_credits(sender, amount) self.withdraw_credits(sender, amount)
self.deposit_credits(receiver, amount) self.deposit_credits(receiver, amount)
else: else:
raise NoAccount raise NoAccount()
def can_spend(self, user, amount): def can_spend(self, user, amount):
account = self._get_account(user) account = self._get_account(user)
@ -182,7 +182,7 @@ class Bank:
try: try:
return deepcopy(self.accounts[server.id][user.id]) return deepcopy(self.accounts[server.id][user.id])
except KeyError: except KeyError:
raise NoAccount raise NoAccount()
class Economy: class Economy:
"""Economy """Economy