allow optional user arg for some commands

Author: yungwine

modules/backups.py

@@ -7,7 +7,7 @@
 
 from modules.module import MtcModule
 from mypylib.mypylib import color_print, ip2int, run_as_root, parse, MyPyClass
-from mytonctrl.utils import get_current_user
+from mytonctrl.utils import get_current_user, pop_user_from_args
 from mytoninstaller.config import get_own_ip
 
 
@@ -29,20 +29,22 @@ def create_tmp_ton_dir(self):
         return dir_name
 
     @staticmethod
-    def run_create_backup(args):
-        user = get_current_user()
+    def run_create_backup(args, user: str = None):
+        if user is None:
+            user = get_current_user()
         backup_script_path = pkg_resources.resource_filename('mytonctrl', 'scripts/create_backup.sh')
         return subprocess.run(["bash", backup_script_path, "-u", user] + args, timeout=5)
 
     def create_backup(self, args):
-        if len(args) > 1:
-            color_print("{red}Bad args. Usage:{endc} create_backup [filename]")
+        if len(args) > 3:
+            color_print("{red}Bad args. Usage:{endc} create_backup [filename] [-u <user>]")
             return
         tmp_dir = self.create_tmp_ton_dir()
         command_args = ["-m", self.ton.local.buffer.my_work_dir, "-t", tmp_dir]
+        user = pop_user_from_args(args)
         if len(args) == 1:
             command_args += ["-d", args[0]]
-        process = self.run_create_backup(command_args)
+        process = self.run_create_backup(command_args, user=user)
 
         if process.returncode == 0:
             color_print("create_backup - {green}OK{endc}")
@@ -53,15 +55,17 @@ def create_backup(self, args):
     # end define
 
     @staticmethod
-    def run_restore_backup(args):
-        user = get_current_user()
+    def run_restore_backup(args, user: str = None):
+        if user is None:
+            user = get_current_user()
         restore_script_path = pkg_resources.resource_filename('mytonctrl', 'scripts/restore_backup.sh')
         return run_as_root(["bash", restore_script_path, "-u", user] + args)
 
     def restore_backup(self, args):
-        if len(args) == 0 or len(args) > 3:
-            color_print("{red}Bad args. Usage:{endc} restore_backup <filename> [-y] [--skip-create-backup]")
+        if len(args) == 0 or len(args) > 5:
+            color_print("{red}Bad args. Usage:{endc} restore_backup <filename> [-y] [--skip-create-backup] [-u <user>]")
             return
+        user = pop_user_from_args(args)
         if '-y' not in args:
             res = input(
                 f'This action will overwrite existing configuration with contents of backup archive, please make sure that donor node is not in operation prior to this action. Proceed [y/n]')
@@ -82,7 +86,7 @@ def restore_backup(self, args):
         ip = str(ip2int(get_own_ip()))
         command_args = ["-m", self.ton.local.buffer.my_work_dir, "-n", args[0], "-i", ip]
 
-        if self.run_restore_backup(command_args) == 0:
+        if self.run_restore_backup(command_args, user=user) == 0:
             self.ton.local.load_db()
             if self.ton.using_validator():
                 from modules.btc_teleport import BtcTeleportModule

modules/btc_teleport.py

@@ -57,29 +57,31 @@ def create_env_file(self, reinit=False):
         with open(env_path, 'w') as f:
             f.write(text)
 
-    def add_daemon(self):
+    def add_daemon(self, user: str = None):
         start = f'{self.bin_dir}/oracle'
         script_path = pkg_resources.resource_filename('mytoninstaller', 'scripts/add2systemd.sh')
-        user = get_current_user()
+        if user is None:
+            user = get_current_user()
         run_as_root(['bash', script_path, '-n', 'btc_teleport', '-u', user, '-g', user, '-s', start, '-w', self.bin_dir])
 
-    def install(self, branch):
+    def install(self, branch: str, user: str = None):
         script_path = pkg_resources.resource_filename('mytonctrl', 'scripts/btc_teleport1.sh')
-        user = get_current_user()
+        if user is None:
+            user = get_current_user()
         exit_code = run_as_root(["bash", script_path, "-s", '/usr/src', "-r", self.repo_name, "-b", branch, "-u", user])
         if exit_code != 0:
             raise Exception('Failed to install btc_teleport')
         script_path = pkg_resources.resource_filename('mytonctrl', 'scripts/btc_teleport2.sh')
         subprocess.run(["bash", script_path, "-s", self.src_dir])
 
-    def init(self, reinstall=False, branch: str = 'master'):
+    def init(self, reinstall=False, branch: str = 'master', user: str = None):
         if os.path.exists(self.src_dir) and not reinstall:
             return
         self.local.add_log('Installing btc_teleport', 'info')
         os.makedirs(self.keystore_path, mode=0o700, exist_ok=True)
-        self.install(branch)
+        self.install(branch, user=user)
         self.create_env_file()
-        self.add_daemon()
+        self.add_daemon(user=user)
         self.local.add_log('Installed btc_teleport', 'info')
 
     @staticmethod

mytonctrl/mytonctrl.py

@@ -45,7 +45,8 @@
 )
 from mytoncore.telemetry import is_host_virtual
 from mytonctrl.migrate import run_migrations
-from mytonctrl.utils import GetItemFromList, timestamp2utcdatetime, fix_git_config, is_hex, GetColorInt
+from mytonctrl.utils import GetItemFromList, timestamp2utcdatetime, fix_git_config, is_hex, GetColorInt, \
+	pop_user_from_args
 
 import sys, getopt, os
 
@@ -331,9 +332,12 @@ def Update(local, args):
 #end define
 
 def Upgrade(local, ton, args: list):
-	if '--btc-teleport' in args:  # upgrade --btc-teleport [branch]
-		branch = args[args.index('--btc-teleport') + 1] if len(args) > args.index('--btc-teleport') + 1 else 'master'
-		upgrade_btc_teleport(local, ton, reinstall=True, branch=branch)
+	if '--btc-teleport' in args:  # upgrade --btc-teleport [branch] [-u <user>]
+		branch = 'master'
+		user = pop_user_from_args(args)
+		if len(args) > args.index('--btc-teleport') + 1:
+			branch = args[args.index('--btc-teleport') + 1]
+		upgrade_btc_teleport(local, ton, reinstall=True, branch=branch, user=user)
 		return
 	repo = "ton"
 	author, repo, branch = check_git(args, repo, "upgrade")
@@ -378,10 +382,10 @@ def Upgrade(local, ton, args: list):
 #end define
 
 
-def upgrade_btc_teleport(local, ton, reinstall=False, branch: str = 'master'):
+def upgrade_btc_teleport(local, ton, reinstall=False, branch: str = 'master', user = None):
 	from modules.btc_teleport import BtcTeleportModule
 	module = BtcTeleportModule(ton, local)
-	local.try_function(module.init, args=[reinstall, branch])
+	local.try_function(module.init, args=[reinstall, branch, user])
 
 
 def get_clang_major_version():

mytonctrl/utils.py

@@ -62,3 +62,13 @@ def GetColorInt(data, border, logic, ending=None):
 
 def get_current_user():
     return pwd.getpwuid(os.getuid()).pw_name
+
+def pop_user_from_args(args: list):
+    if '-u' in args:
+        user_index = args.index('-u') + 1
+        if user_index >= len(args):
+            raise Exception(f'User value not found after "-u" in args: {args}')
+        user = args.pop(user_index)
+        args.pop(args.index('-u'))
+        return user
+    return None

mytoninstaller/mytoninstaller.py

@@ -11,7 +11,7 @@
 
 from mypylib.mypylib import MyPyClass, run_as_root, color_print
 from mypyconsole.mypyconsole import MyPyConsole
-from mytonctrl.utils import get_current_user
+from mytonctrl.utils import get_current_user, pop_user_from_args
 
 from mytoninstaller.config import GetLiteServerConfig, get_ls_proxy_config
 from mytoninstaller.node_args import get_node_args
@@ -207,7 +207,9 @@ def CreateLocalConfigFile(local, args):
 		init_block["rootHash"] = b642hex(config_init_block['root_hash'])
 		init_block["fileHash"] = b642hex(config_init_block['file_hash'])
 	init_block_b64 = dict2b64(init_block)
-	user = local.buffer.user or get_current_user()
+	user = pop_user_from_args(args)
+	if user is None:
+		user = local.buffer.user or get_current_user()
 	args = ["python3", "-m", "mytoninstaller", "-u", user, "-e", "clc", "-i", init_block_b64]
 	run_as_root(args)
 #end define

mytoninstaller/settings.py

@@ -1134,10 +1134,10 @@ def ConfigureFromBackup(local):
 	os.makedirs(local.buffer.ton_work_dir, exist_ok=True)
 	if not local.buffer.only_mtc:
 		ip = str(ip2int(get_own_ip()))
-		BackupModule.run_restore_backup(["-m", mconfig_dir, "-n", backup_file, "-i", ip])
+		BackupModule.run_restore_backup(["-m", mconfig_dir, "-n", backup_file, "-i", ip], user=local.buffer.user)
 
 	if local.buffer.only_mtc:
-		BackupModule.run_restore_backup(["-m", mconfig_dir, "-n", backup_file])
+		BackupModule.run_restore_backup(["-m", mconfig_dir, "-n", backup_file], user=local.buffer.user)
 		local.add_log("Installing only mtc", "info")
 		vconfig_path = local.buffer.vconfig_path
 		vconfig = GetConfig(path=vconfig_path)
@@ -1161,7 +1161,7 @@ def ConfigureOnlyNode(local):
 	mconfig_dir = get_dir_from_path(mconfig_path)
 	local.add_log("start ConfigureOnlyNode function", "info")
 
-	process = BackupModule.run_create_backup(["-m", mconfig_dir, ])
+	process = BackupModule.run_create_backup(["-m", mconfig_dir], user=local.buffer.user)
 	if process.returncode != 0:
 		local.add_log("Backup creation failed", "error")
 		return