diff options
Diffstat (limited to 'geldschieberbot.py')
| -rw-r--r-- | geldschieberbot.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/geldschieberbot.py b/geldschieberbot.py index 046d543..ff859f0 100644 --- a/geldschieberbot.py +++ b/geldschieberbot.py @@ -1009,8 +1009,6 @@ class Geldschieberbot: self.dry_run = dry_run # Run without changing the stored state self.record_changes = True # Should changes be recorded - self.load_state() - # Command dispatch table self.cmds = { 'reg': self.register, @@ -1046,8 +1044,13 @@ class Geldschieberbot: 'thanks': self.thanks, } - def __del__(self): - self.save_state() + def __enter__(self): + self.load_state() + return self + + def __exit__(self, exc_type, exc_value, exc_tb): + if not exc_type: + self.save_state() def enable_dry_run(self) -> bool: """Enable dry run""" |