mirror of
https://github.com/hyperknot/openfreemap.git
synced 2026-05-21 22:12:15 +00:00
162 lines
4.0 KiB
Python
162 lines
4.0 KiB
Python
import os
|
|
import secrets
|
|
import string
|
|
from pathlib import Path
|
|
|
|
|
|
def put(
|
|
c, local_path, remote_path, permissions=None, owner='root', group=None, create_parent_dir=False
|
|
):
|
|
tmp_path = f'/tmp/fabtmp_{random_string(8)}'
|
|
c.put(local_path, tmp_path)
|
|
|
|
if create_parent_dir:
|
|
dirname = os.path.dirname(remote_path)
|
|
c.sudo(f'mkdir -p {dirname}')
|
|
set_permission(c, dirname, owner=owner, group=group)
|
|
|
|
if is_dir(c, remote_path):
|
|
if not remote_path.endswith('/'):
|
|
remote_path += '/'
|
|
|
|
filename = os.path.basename(local_path)
|
|
remote_path += filename
|
|
|
|
c.sudo(f"mv '{tmp_path}' '{remote_path}'")
|
|
c.sudo(f"rm -rf '{tmp_path}'")
|
|
|
|
set_permission(c, remote_path, permissions=permissions, owner=owner, group=group)
|
|
|
|
|
|
def put_dir(
|
|
c,
|
|
local_dir: Path,
|
|
remote_dir,
|
|
dir_permissions=None,
|
|
file_permissions=None,
|
|
owner='root',
|
|
group=None,
|
|
exclude_set=None,
|
|
):
|
|
"""
|
|
copies all files from local path to remote path
|
|
not recursive
|
|
"""
|
|
|
|
files = [file for file in local_dir.iterdir() if file.is_file()]
|
|
|
|
if exclude_set:
|
|
files = [f for f in files if f.name not in exclude_set]
|
|
|
|
c.sudo(f'mkdir -p "{remote_dir}"')
|
|
set_permission(c, remote_dir, permissions=dir_permissions, owner=owner, group=group)
|
|
|
|
for file in files:
|
|
print(f'uploading {remote_dir}/{file.name}')
|
|
put(c, file, f'{remote_dir}/{file.name}', file_permissions, owner, group)
|
|
|
|
|
|
def put_str(c, remote_path, str_):
|
|
tmp_file = 'tmp.txt'
|
|
with open(tmp_file, 'w') as outfile:
|
|
outfile.write(str_ + '\n')
|
|
put(c, tmp_file, remote_path)
|
|
os.remove(tmp_file)
|
|
|
|
|
|
def append_str(c, remote_path, str_):
|
|
tmp_path = f'/tmp/fabtmp_{random_string(8)}'
|
|
put_str(c, tmp_path, str_)
|
|
|
|
sudo_cmd(c, f"cat '{tmp_path}' >> '{remote_path}'")
|
|
c.sudo(f'rm -f {tmp_path}')
|
|
|
|
|
|
def sudo_cmd(c, cmd, user=None):
|
|
cmd = cmd.replace('"', '\\"')
|
|
c.sudo(f'bash -c "{cmd}"', user=user)
|
|
|
|
|
|
def set_permission(c, path, *, permissions=None, owner=None, group=None):
|
|
if owner:
|
|
if not group:
|
|
group = owner
|
|
c.sudo(f"chown {owner}:{group} '{path}'")
|
|
|
|
if permissions:
|
|
c.sudo(f"chmod {permissions} '{path}'")
|
|
|
|
|
|
def reboot(c):
|
|
print('Rebooting')
|
|
try:
|
|
c.sudo('reboot')
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def exists(c, path):
|
|
return c.sudo(f"test -e '{path}'", hide=True, warn=True).ok
|
|
|
|
|
|
def is_dir(c, path):
|
|
return c.sudo(f"test -d '{path}'", hide=True, warn=True).ok
|
|
|
|
|
|
def random_string(length):
|
|
return ''.join(secrets.choice(string.ascii_uppercase + string.digits) for _ in range(length))
|
|
|
|
|
|
def ubuntu_release(c):
|
|
return c.run('lsb_release -rs').stdout.strip()[:2]
|
|
|
|
|
|
def ubuntu_codename(c):
|
|
return c.run('lsb_release -cs').stdout.strip()
|
|
|
|
|
|
def apt_get_update(c):
|
|
c.sudo('apt-get update')
|
|
|
|
|
|
def apt_get_install(c, pkgs, warn=False):
|
|
c.sudo(
|
|
f'DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends {pkgs}',
|
|
warn=warn,
|
|
echo=True,
|
|
)
|
|
|
|
|
|
def apt_get_purge(c, pkgs):
|
|
c.sudo(f'DEBIAN_FRONTEND=noninteractive apt-get purge -y {pkgs}')
|
|
|
|
|
|
def apt_get_autoremove(c):
|
|
c.sudo('DEBIAN_FRONTEND=noninteractive apt-get autoremove -y')
|
|
|
|
|
|
def get_username(c):
|
|
return c.run('whoami').stdout.strip()
|
|
|
|
|
|
def add_user(c, username, passwd=None, uid=None):
|
|
uid_str = f'--uid={uid}' if uid else ''
|
|
|
|
# --disabled-password -> ssh-key login only
|
|
c.sudo(f'adduser --disabled-password --gecos "" {uid_str} {username}', warn=True)
|
|
if passwd:
|
|
sudo_cmd(c, f'echo "{username}:{passwd}" | chpasswd')
|
|
|
|
|
|
def remove_user(c, username):
|
|
c.sudo(f'userdel -r {username}', warn=True)
|
|
c.sudo(f'rm -rf /home/{username}')
|
|
|
|
|
|
def enable_sudo(c, username, nopasswd=False):
|
|
c.sudo(f'usermod -aG sudo {username}')
|
|
if nopasswd:
|
|
put_str(c, '/etc/sudoers.d/tmp.', f'{username} ALL=(ALL) NOPASSWD:ALL')
|
|
set_permission(c, '/etc/sudoers.d/tmp.', permissions='440', owner='root')
|
|
c.sudo(f'mv /etc/sudoers.d/tmp. /etc/sudoers.d/{username}')
|