add auto backups

Author: yungwine

modules/__init__.py

@@ -58,7 +58,9 @@ class Setting:
     'debug': Setting(None, False, 'Debug mtc console mode. Prints Traceback on errors'),
     'subscribe_tg_channel': Setting('validator', False, 'Disables warning about subscribing to the `TON STATUS` channel'),
     'BotToken': Setting('alert-bot', None, 'Alerting Telegram bot token'),
-    'ChatId': Setting('alert-bot', None, 'Alerting Telegram chat id')
+    'ChatId': Setting('alert-bot', None, 'Alerting Telegram chat id'),
+    'auto_backup': Setting('validator', True, 'Make backup of validator every election'),
+    'auto_backup_path': Setting('validator', '/tmp/mytoncore/backups/', 'Path to store backups'),
 }
 
 

mytoncore/mytoncore.py

@@ -1455,6 +1455,7 @@ def ElectionEntry(self, args=None):
 		self.local.add_log("ElectionEntry completed. Start work time: " + str(startWorkTime))
 
 		self.clear_tmp()
+		self.make_backup(startWorkTime)
 
 	#end define
 
@@ -1471,6 +1472,18 @@ def clear_tmp(self):
 
 		self.local.add_log(f"Removed {count} old files from tmp dir for {int(time.time() - start)} seconds", "info")
 
+	def make_backup(self, election_id: str):
+		if not self.local.db.get("auto_backup"):
+			return
+		from mytonctrl.mytonctrl import create_backup
+		args = []
+		name = f"/mytonctrl_backup_elid{election_id}.zip"
+		if self.local.db.get("auto_backup_path"):
+			args.append(self.local.db.get("auto_backup_path") + name)
+		else:
+			args.append(self.tempDir + "/backups" + name)
+		create_backup(self.local, self, args + ['-y'])
+
 	def GetValidatorKeyByTime(self, startWorkTime, endWorkTime):
 		self.local.add_log("start GetValidatorKeyByTime function", "debug")
 		# Check temp key