131 lines
3.1 KiB
Python
Executable File
131 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Копирование файлов из/на сервер
|
|
# scp_helper [host1:]/path [host2:]/path
|
|
#
|
|
|
|
import argparse
|
|
import os
|
|
|
|
CMD_TAR_DECOMP='tar -xz'
|
|
CMD_TAR_COMP='tar -cz'
|
|
SUDO_REMOTE=True
|
|
|
|
def parse_arg(s):
|
|
# Парсим путь. Проверяем тип пути
|
|
if s.find(':') > -1:
|
|
connection_name, path = s.split(':')
|
|
else:
|
|
connection_name = 'local'
|
|
path = s
|
|
|
|
if path[0] != '/':
|
|
path = './' + path
|
|
|
|
path_parts = path.split('/')
|
|
work_path = '/'.join(path_parts[0:-1])
|
|
obj = path_parts[-1]
|
|
|
|
return {
|
|
'connection_name': connection_name,
|
|
'work_path': work_path,
|
|
'obj': obj
|
|
}
|
|
|
|
def make_pack_cmd(work_path, obj):
|
|
cmd = f'tar -C {work_path} -c {obj}'
|
|
return cmd
|
|
|
|
def make_unpack_cmd(work_path, obj):
|
|
cmd = f'tar -C {work_path}/{obj} -x'
|
|
return cmd
|
|
|
|
def add_sudo(s):
|
|
res = 'sudo ' + s
|
|
return res
|
|
|
|
def make_ssh_cmd(host, port, user, cmd):
|
|
res = f"ssh {user}@{host} -p {port} '{cmd}'"
|
|
return res
|
|
|
|
def get_connection_config(con_name):
|
|
# Получаем конфигурацию подключения по SSH
|
|
config_path = os.environ.get('SSH_CONNECTOR_HOST_LIST')
|
|
res = None
|
|
with open(config_path, 'r') as f:
|
|
for line in f:
|
|
if line[0] not in ('#', ' ', '\n'):
|
|
connection_name, host, user, port, *_ = line.split()
|
|
if connection_name == con_name:
|
|
res = {
|
|
'connection_name': connection_name,
|
|
'host': host,
|
|
'user': user,
|
|
'port': port
|
|
}
|
|
break
|
|
if res is None:
|
|
raise Exception('Connection name not found')
|
|
return res
|
|
|
|
def get_path_config(path):
|
|
# Объединяем конфигурацию пути и конфигурацию подключения, если подключение удаленное
|
|
config = parse_arg(path)
|
|
|
|
if config['connection_name'] != 'local':
|
|
res_config = {
|
|
**config,
|
|
**get_connection_config(
|
|
config['connection_name']
|
|
)
|
|
}
|
|
else:
|
|
res_config = config
|
|
|
|
return res_config
|
|
|
|
def make_full_cmd(config, tar_func):
|
|
cmd = tar_func(config['work_path'], config['obj'])
|
|
|
|
if SUDO_REMOTE and config['connection_name'] != 'local':
|
|
cmd = add_sudo(cmd)
|
|
|
|
if config['connection_name'] != 'local':
|
|
cmd = make_ssh_cmd(
|
|
config['host'],
|
|
config['port'],
|
|
config['user'],
|
|
cmd
|
|
)
|
|
|
|
return cmd
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description='Copy files'
|
|
)
|
|
parser.add_argument('in_path', type=str)
|
|
parser.add_argument('out_path', type=str)
|
|
|
|
|
|
args = parser.parse_args()
|
|
|
|
in_config = get_path_config(args.in_path)
|
|
out_config = get_path_config(args.out_path)
|
|
|
|
in_cmd = make_full_cmd(
|
|
in_config,
|
|
make_pack_cmd
|
|
)
|
|
|
|
out_cmd = make_full_cmd(
|
|
out_config,
|
|
make_unpack_cmd
|
|
)
|
|
|
|
full_cmd = f'{in_cmd} | {out_cmd}'
|
|
print(full_cmd)
|
|
|
|
|
|
|
|
|