From 6c25aee5cc3b8dcd36c2861c5012dca4eeab30e1 Mon Sep 17 00:00:00 2001 From: izyak Date: Tue, 15 Feb 2022 16:27:24 +0545 Subject: [PATCH 1/2] Set total staked instead of user staked --- score/ommToken/snapshot/snapshot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/score/ommToken/snapshot/snapshot.py b/score/ommToken/snapshot/snapshot.py index f9db83d9..a76f9ce1 100644 --- a/score/ommToken/snapshot/snapshot.py +++ b/score/ommToken/snapshot/snapshot.py @@ -51,7 +51,7 @@ def _createSnapshot(self, _owner: Address, _user_old_staked_balance: int, _user_ """ _now = self.now() self._snapshot.create_checkpoints(_owner, _now, _user_new_staked_balance) - self._snapshot.create_total_checkpoints(_now, _user_new_staked_balance) + self._snapshot.create_total_checkpoints(_now, _total_staked_balance) self.SnapshotCreated(_owner, _user_new_staked_balance, _user_old_staked_balance) From c3a971f6704481cac98a882595a704e5308aeffb Mon Sep 17 00:00:00 2001 From: izyak Date: Tue, 15 Feb 2022 16:28:20 +0545 Subject: [PATCH 2/2] Temporary method to update total staked balance at a time --- score/ommToken/tokens/IRC2.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/score/ommToken/tokens/IRC2.py b/score/ommToken/tokens/IRC2.py index 45dcacb6..f8268ec9 100644 --- a/score/ommToken/tokens/IRC2.py +++ b/score/ommToken/tokens/IRC2.py @@ -478,3 +478,8 @@ def _mint(self, _to: Address, _amount: int, _data: bytes = None) -> None: self._balances[_to] += _amount self.Transfer(ZERO_SCORE_ADDRESS, _to, _amount, _data) + + @external + @only_owner + def updateTotalStakedBalanceOfAt(self, _timestamp: int, _staked: int): + self._snapshot.create_total_checkpoints(_timestamp, _staked) \ No newline at end of file