You've already forked haproxy-wi
mirror of
https://github.com/roxy-wi/roxy-wi.git
synced 2025-12-15 11:54:05 +08:00
3a9c18392b
Added support for managing Docker settings and improved the handling of service-specific configurations. Updated various backup methods and database columns to ensure consistency and clarity. Included new database migration scripts to handle renamed columns.
123 lines
2.9 KiB
Python
123 lines
2.9 KiB
Python
from app.modules.db.db_model import Backup, S3Backup, GitSetting
|
|
from app.modules.db.common import out_error
|
|
from app.modules.roxywi.exception import RoxywiResourceNotFound
|
|
|
|
models = {
|
|
'fs': Backup,
|
|
's3': S3Backup,
|
|
'git': GitSetting,
|
|
}
|
|
|
|
|
|
def insert_backup_job(server_id, rserver, rpath, backup_type, time, cred, description):
|
|
try:
|
|
return Backup.insert(
|
|
server_id=server_id, rserver=rserver, rpath=rpath, type=backup_type, time=time,
|
|
cred_id=cred, description=description
|
|
).execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
|
|
|
|
def insert_s3_backup_job(**kwargs):
|
|
try:
|
|
return S3Backup.insert(**kwargs).execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
|
|
|
|
def update_backup_job(backup_id: int, model: str, **kwargs):
|
|
model = models[model]
|
|
try:
|
|
model.update(**kwargs).where(model.id == backup_id).execute()
|
|
except model.DoesNotExist:
|
|
raise RoxywiResourceNotFound
|
|
except Exception as e:
|
|
out_error(e)
|
|
|
|
|
|
def delete_backup(backup_id: int, model: str) -> None:
|
|
model = models[model]
|
|
try:
|
|
model.delete().where(model.id == backup_id).execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
|
|
|
|
def insert_new_git(server_id, service_id, repo, branch, time, cred, description) -> int:
|
|
try:
|
|
return GitSetting.insert(
|
|
server_id=server_id, service_id=service_id, repo=repo, branch=branch, time=time,
|
|
cred_id=cred, description=description
|
|
).execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
|
|
|
|
def select_gits(**kwargs):
|
|
if kwargs.get("server_id") is not None and kwargs.get("service_id") is not None:
|
|
query = GitSetting.select().where(
|
|
(GitSetting.server_id == kwargs.get("server_id")) & (GitSetting.service_id == kwargs.get("service_id")))
|
|
else:
|
|
query = GitSetting.select().order_by(GitSetting.id)
|
|
|
|
try:
|
|
query_res = query.execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
else:
|
|
return query_res
|
|
|
|
|
|
def select_backups(**kwargs):
|
|
if kwargs.get("backup_id") is not None:
|
|
query = Backup.select().where(Backup.id == kwargs.get("backup_id"))
|
|
else:
|
|
query = Backup.select().order_by(Backup.id)
|
|
|
|
try:
|
|
return query.execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
|
|
|
|
def select_s3_backups(**kwargs):
|
|
if kwargs.get("server") is not None and kwargs.get("bucket") is not None:
|
|
query = S3Backup.select().where(
|
|
(S3Backup.server_id == kwargs.get("server")) &
|
|
(S3Backup.s3_server == kwargs.get("s3_server")) &
|
|
(S3Backup.bucket == kwargs.get("bucket"))
|
|
)
|
|
else:
|
|
query = S3Backup.select().order_by(S3Backup.id)
|
|
|
|
try:
|
|
query_res = query.execute()
|
|
except Exception as e:
|
|
out_error(e)
|
|
else:
|
|
return query_res
|
|
|
|
|
|
def check_exists_backup(server_id: int, model: str) -> bool:
|
|
model = models[model]
|
|
try:
|
|
backup = model.get(model.server_id == server_id)
|
|
except Exception:
|
|
pass
|
|
else:
|
|
if backup.id is not None:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def get_backup(backup_id: int, model: str) -> Backup:
|
|
model = models[model]
|
|
try:
|
|
return model.get(model.id == backup_id)
|
|
except model.DoesNotExist:
|
|
raise RoxywiResourceNotFound
|
|
except Exception as e:
|
|
out_error(e)
|