134 lines
3.4 KiB
Python
134 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
"""Backups management module"""
|
|
import json
|
|
import subprocess
|
|
from flask import request
|
|
from flask_restful import Resource
|
|
|
|
from selfprivacy_api.resources.services import api
|
|
|
|
|
|
class ListAllBackups(Resource):
|
|
"""List all restic backups"""
|
|
|
|
def get(self):
|
|
"""
|
|
Get all restic backups
|
|
---
|
|
tags:
|
|
- Backups
|
|
security:
|
|
- bearerAuth: []
|
|
responses:
|
|
200:
|
|
description: A list of snapshots
|
|
400:
|
|
description: Bad request
|
|
401:
|
|
description: Unauthorized
|
|
"""
|
|
repository_name = request.headers.get("X-Repository-Name")
|
|
|
|
backup_listing_command = [
|
|
"restic",
|
|
"-r",
|
|
f"b2:{repository_name}:/sfbackup",
|
|
"snapshots",
|
|
"--json",
|
|
]
|
|
|
|
with subprocess.Popen(
|
|
backup_listing_command,
|
|
shell=False,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
) as backup_listing_process_descriptor:
|
|
snapshots_list = backup_listing_process_descriptor.communicate()[0]
|
|
|
|
return snapshots_list.decode("utf-8")
|
|
|
|
|
|
class AsyncCreateBackup(Resource):
|
|
"""Create a new restic backup"""
|
|
|
|
def put(self):
|
|
"""
|
|
Initiate a new restic backup
|
|
---
|
|
tags:
|
|
- Backups
|
|
security:
|
|
- bearerAuth: []
|
|
responses:
|
|
200:
|
|
description: Backup creation has started
|
|
400:
|
|
description: Bad request
|
|
401:
|
|
description: Unauthorized
|
|
"""
|
|
repository_name = request.headers.get("X-Repository-Name")
|
|
|
|
backup_command = [
|
|
"restic",
|
|
"-r",
|
|
f"b2:{repository_name}:/sfbackup",
|
|
"--verbose",
|
|
"--json",
|
|
"backup",
|
|
"/var",
|
|
]
|
|
|
|
with open("/tmp/backup.log", "w", encoding="utf-8") as log_file:
|
|
subprocess.Popen(
|
|
backup_command, shell=False, stdout=log_file, stderr=subprocess.STDOUT
|
|
)
|
|
|
|
return {
|
|
"status": 0,
|
|
"message": "Backup creation has started",
|
|
}
|
|
|
|
|
|
class CheckBackupStatus(Resource):
|
|
"""Check current backup status"""
|
|
|
|
def get(self):
|
|
"""
|
|
Get backup status
|
|
---
|
|
tags:
|
|
- Backups
|
|
security:
|
|
- bearerAuth: []
|
|
responses:
|
|
200:
|
|
description: Backup status
|
|
400:
|
|
description: Bad request
|
|
401:
|
|
description: Unauthorized
|
|
"""
|
|
backup_status_check_command = ["tail", "-1", "/tmp/backup.log"]
|
|
|
|
with subprocess.Popen(
|
|
backup_status_check_command,
|
|
shell=False,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
) as backup_status_check_process_descriptor:
|
|
backup_process_status = (
|
|
backup_status_check_process_descriptor.communicate()[0].decode("utf-8")
|
|
)
|
|
|
|
try:
|
|
json.loads(backup_process_status)
|
|
except ValueError:
|
|
return {"message": backup_process_status}
|
|
return backup_process_status
|
|
|
|
|
|
api.add_resource(ListAllBackups, "/restic/backup/list")
|
|
api.add_resource(AsyncCreateBackup, "/restic/backup/create")
|
|
api.add_resource(CheckBackupStatus, "/restic/backup/status")
|