From 26263ee67e2a4cea48222225afacb6f165ad82ef Mon Sep 17 00:00:00 2001 From: Twentysix Date: Sat, 3 Sep 2016 06:29:57 +0200 Subject: [PATCH] [Economy] Add optional argument initial_balance in create_account --- cogs/economy.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/cogs/economy.py b/cogs/economy.py index 18ce64cb0..e7ed180cf 100644 --- a/cogs/economy.py +++ b/cogs/economy.py @@ -46,7 +46,7 @@ class Bank: self.accounts = dataIO.load_json(file_path) self.bot = bot - def create_account(self, user): + def create_account(self, user, *, initial_balance=0): server = user.server if not self.account_exists(user): if server.id not in self.accounts: @@ -54,10 +54,12 @@ class Bank: if user.id in self.accounts: # Legacy account balance = self.accounts[user.id]["balance"] else: - balance = 0 + balance = initial_balance timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") - account = {"name" : user.name, "balance" : balance, - "created_at" : timestamp} + account = {"name" : user.name, + "balance" : balance, + "created_at" : timestamp + } self.accounts[server.id][user.id] = account self._save_bank() return self.get_account(user)