From 3ffd267a42528c1fea5c79084ed6c05f6b6a3dff Mon Sep 17 00:00:00 2001
From: Flugschwein <flugschwein@gmx.at>
Date: Thu, 11 Oct 2018 14:05:46 +0200
Subject: [PATCH] Add additional parameter for get_recharge_timedelta

---
 beem/account.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/beem/account.py b/beem/account.py
index 1f9973ab..c3d8d7aa 100644
--- a/beem/account.py
+++ b/beem/account.py
@@ -525,13 +525,18 @@ class Account(BlockchainObject):
         remainingTime = self.get_recharge_timedelta(voting_power_goal=voting_power_goal)
         return formatTimedelta(remainingTime)
 
-    def get_recharge_timedelta(self, voting_power_goal=100):
+    def get_recharge_timedelta(self, voting_power_goal=100, starting_voting_power=None):
         """ Returns the account voting power recharge time as timedelta object
 
             :param float voting_power_goal: voting power goal in percentage (default is 100)
 
         """
-        missing_vp = voting_power_goal - self.get_voting_power()
+        if starting_voting_power is None:
+            missing_vp = voting_power_goal - self.get_voting_power()
+        elif isinstance(starting_voting_power, int) or isinstance(starting_voting_power, float):
+            missing_vp = voting_power_goal - starting_voting_power
+        else:
+            raise ValueError('starting_voting_power must be a number.')
         if missing_vp < 0:
             return 0
         recharge_seconds = missing_vp * 100 * STEEM_VOTING_MANA_REGENERATION_SECONDS / STEEM_100_PERCENT
-- 
GitLab