147 lines
4.5 KiB
Python
147 lines
4.5 KiB
Python
from restic_compose_backup.containers import Container
|
|
from restic_compose_backup.config import Config
|
|
from restic_compose_backup import (
|
|
commands,
|
|
restic,
|
|
)
|
|
from restic_compose_backup import utils
|
|
|
|
|
|
class MariadbContainer(Container):
|
|
container_type = 'mariadb'
|
|
|
|
def get_credentials(self) -> dict:
|
|
"""dict: get credentials for the service"""
|
|
return {
|
|
'host': self.hostname,
|
|
'username': self.get_config_env('MYSQL_USER'),
|
|
'password': self.get_config_env('MYSQL_PASSWORD'),
|
|
'port': "3306",
|
|
}
|
|
|
|
def ping(self) -> bool:
|
|
"""Check the availability of the service"""
|
|
creds = self.get_credentials()
|
|
|
|
with utils.environment('MYSQL_PWD', creds['password']):
|
|
return commands.ping_mariadb(
|
|
creds['host'],
|
|
creds['port'],
|
|
creds['username'],
|
|
)
|
|
|
|
def dump_command(self) -> list:
|
|
"""list: create a dump command restic and use to send data through stdin"""
|
|
creds = self.get_credentials()
|
|
return [
|
|
"mysqldump",
|
|
f"--host={creds['host']}",
|
|
f"--port={creds['port']}",
|
|
f"--user={creds['username']}",
|
|
"--all-databases",
|
|
]
|
|
|
|
def backup(self):
|
|
config = Config()
|
|
creds = self.get_credentials()
|
|
|
|
with utils.environment('MYSQL_PWD', creds['password']):
|
|
return restic.backup_from_stdin(
|
|
config.repository,
|
|
f'/databases/{self.service_name}/all_databases.sql',
|
|
self.dump_command(),
|
|
)
|
|
|
|
|
|
class MysqlContainer(Container):
|
|
container_type = 'mysql'
|
|
|
|
def get_credentials(self) -> dict:
|
|
"""dict: get credentials for the service"""
|
|
return {
|
|
'host': self.hostname,
|
|
'username': self.get_config_env('MYSQL_USER'),
|
|
'password': self.get_config_env('MYSQL_PASSWORD'),
|
|
'port': "3306",
|
|
}
|
|
|
|
def ping(self) -> bool:
|
|
"""Check the availability of the service"""
|
|
creds = self.get_credentials()
|
|
|
|
with utils.environment('MYSQL_PWD', creds['password']):
|
|
return commands.ping_mysql(
|
|
creds['host'],
|
|
creds['port'],
|
|
creds['username'],
|
|
)
|
|
|
|
def dump_command(self) -> list:
|
|
"""list: create a dump command restic and use to send data through stdin"""
|
|
creds = self.get_credentials()
|
|
return [
|
|
"mysqldump",
|
|
f"--host={creds['host']}",
|
|
f"--port={creds['port']}",
|
|
f"--user={creds['username']}",
|
|
"--all-databases",
|
|
]
|
|
|
|
def backup(self):
|
|
config = Config()
|
|
creds = self.get_credentials()
|
|
|
|
with utils.environment('MYSQL_PWD', creds['password']):
|
|
return restic.backup_from_stdin(
|
|
config.repository,
|
|
f'/databases/{self.service_name}/all_databases.sql',
|
|
self.dump_command(),
|
|
)
|
|
|
|
|
|
class PostgresContainer(Container):
|
|
container_type = 'postgres'
|
|
|
|
def get_credentials(self) -> dict:
|
|
"""dict: get credentials for the service"""
|
|
return {
|
|
'host': self.hostname,
|
|
'username': self.get_config_env('POSTGRES_USER'),
|
|
'password': self.get_config_env('POSTGRES_PASSWORD'),
|
|
'port': "5432",
|
|
'database': self.get_config_env('POSTGRES_DB'),
|
|
}
|
|
|
|
def ping(self) -> bool:
|
|
"""Check the availability of the service"""
|
|
creds = self.get_credentials()
|
|
return commands.ping_postgres(
|
|
creds['host'],
|
|
creds['port'],
|
|
creds['username'],
|
|
creds['password'],
|
|
)
|
|
|
|
def dump_command(self) -> list:
|
|
"""list: create a dump command restic and use to send data through stdin"""
|
|
# NOTE: Backs up a single database from POSTGRES_DB env var
|
|
creds = self.get_credentials()
|
|
return [
|
|
"pg_dump",
|
|
f"--host={creds['host']}",
|
|
f"--port={creds['port']}",
|
|
f"--username={creds['username']}",
|
|
creds['database'],
|
|
]
|
|
|
|
def backup(self):
|
|
config = Config()
|
|
creds = self.get_credentials()
|
|
|
|
with utils.environment('PGPASSWORD', creds['password']):
|
|
return restic.backup_from_stdin(
|
|
config.repository,
|
|
f"/databases/{self.service_name}/{creds['database']}.sql",
|
|
self.dump_command(),
|
|
)
|