mirror of
https://github.com/ZettaIO/restic-compose-backup.git
synced 2025-12-14 01:44:59 +00:00
127 lines
3.1 KiB
Python
127 lines
3.1 KiB
Python
import os
|
|
import logging
|
|
from typing import List, TYPE_CHECKING
|
|
from contextlib import contextmanager
|
|
import docker
|
|
|
|
if TYPE_CHECKING:
|
|
from restic_compose_backup.containers import Container
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
TRUE_VALUES = ['1', 'true', 'True', True, 1]
|
|
|
|
|
|
def docker_client():
|
|
"""
|
|
Create a docker client from the following environment variables::
|
|
|
|
DOCKER_HOST=unix://tmp/docker.sock
|
|
DOCKER_TLS_VERIFY=1
|
|
DOCKER_CERT_PATH=''
|
|
"""
|
|
# NOTE: Remove this fallback in 1.0
|
|
if not os.environ.get('DOCKER_HOST'):
|
|
os.environ['DOCKER_HOST'] = 'unix://tmp/docker.sock'
|
|
|
|
return docker.from_env()
|
|
|
|
|
|
def list_containers() -> List[dict]:
|
|
"""
|
|
List all containers.
|
|
|
|
Returns:
|
|
List of raw container json data from the api
|
|
"""
|
|
client = docker_client()
|
|
all_containers = client.containers.list(all=True)
|
|
client.close()
|
|
return [c.attrs for c in all_containers]
|
|
|
|
|
|
def get_swarm_nodes():
|
|
client = docker_client()
|
|
# NOTE: If not a swarm node docker.errors.APIError is raised
|
|
# 503 Server Error: Service Unavailable
|
|
# ("This node is not a swarm manager. Use "docker swarm init" or
|
|
# "docker swarm join" to connect this node to swarm and try again.")
|
|
try:
|
|
return client.nodes.list()
|
|
except docker.errors.APIError:
|
|
return []
|
|
|
|
|
|
def remove_containers(containers: List['Container']):
|
|
client = docker_client()
|
|
logger.info('Attempting to delete stale backup process containers')
|
|
for container in containers:
|
|
logger.info(' -> deleting %s', container.name)
|
|
try:
|
|
c = client.containers.get(container.name)
|
|
c.remove()
|
|
except Exception as ex:
|
|
logger.exception(ex)
|
|
|
|
def join_file_content(file):
|
|
if not os.path.isfile(file):
|
|
return ""
|
|
with open(file) as c:
|
|
output = ""
|
|
lines = c.readlines()
|
|
for i in range(len(lines)):
|
|
output += lines[i] + ","
|
|
return output
|
|
|
|
|
|
def is_true(value):
|
|
"""
|
|
Evaluates the truthfullness of a bool value in container labels
|
|
"""
|
|
return value in TRUE_VALUES
|
|
|
|
|
|
def strip_root(path):
|
|
"""
|
|
Removes the root slash in a path.
|
|
Example: /srv/data becomes srv/data
|
|
"""
|
|
path = path.strip()
|
|
if path.startswith('/'):
|
|
return path[1:]
|
|
|
|
return path
|
|
|
|
def format_tags(tags: str, arg = "--tag") -> List[str]:
|
|
"""
|
|
Takes a comma separated list of tags.
|
|
Splits them and appends --tag to each tag.
|
|
Use the output as the command line argument for the restic cli.
|
|
Example: foo,bar,test becomes --tag foo --tag bar --tag test
|
|
"""
|
|
if not tags:
|
|
return []
|
|
|
|
tags = tags.strip()
|
|
splitTags = tags.split(",")
|
|
output = []
|
|
for tag in splitTags:
|
|
tag = tag.strip()
|
|
if tag:
|
|
output.extend([arg, tag])
|
|
|
|
return output
|
|
|
|
@contextmanager
|
|
def environment(name, value):
|
|
"""Tempset env var"""
|
|
old_val = os.environ.get(name)
|
|
os.environ[name] = value
|
|
try:
|
|
yield
|
|
finally:
|
|
if old_val is None:
|
|
del os.environ[name]
|
|
else:
|
|
os.environ[name] = old_val
|