mirror of https://github.com/openspug/spug
90 lines
3.8 KiB
Python
90 lines
3.8 KiB
Python
# Copyright: (c) OpenSpug Organization. https://github.com/openspug/spug
|
|
# Copyright: (c) <spug.dev@gmail.com>
|
|
# Released under the AGPL-3.0 License.
|
|
from django.views.generic import View
|
|
from django_redis import get_redis_connection
|
|
from apps.host.models import Host
|
|
from apps.file.utils import FileResponseAfter, parse_sftp_attr
|
|
from libs import json_response, JsonParser, Argument
|
|
from functools import partial
|
|
import os
|
|
|
|
|
|
class FileView(View):
|
|
def get(self, request):
|
|
form, error = JsonParser(
|
|
Argument('id', type=int, help='参数错误'),
|
|
Argument('path', help='参数错误')
|
|
).parse(request.GET)
|
|
if error is None:
|
|
if not request.user.has_host_perm(form.id):
|
|
return json_response(error='无权访问主机，请联系管理员')
|
|
host = Host.objects.get(pk=form.id)
|
|
if not host:
|
|
return json_response(error='未找到指定主机')
|
|
cli = host.get_ssh()
|
|
objects = cli.list_dir_attr(form.path)
|
|
return json_response([parse_sftp_attr(x) for x in objects])
|
|
return json_response(error=error)
|
|
|
|
|
|
class ObjectView(View):
|
|
def get(self, request):
|
|
form, error = JsonParser(
|
|
Argument('id', type=int, help='参数错误'),
|
|
Argument('file', help='请输入文件路径')
|
|
).parse(request.GET)
|
|
if error is None:
|
|
if not request.user.has_host_perm(form.id):
|
|
return json_response(error='无权访问主机，请联系管理员')
|
|
host = Host.objects.filter(pk=form.id).first()
|
|
if not host:
|
|
return json_response(error='未找到指定主机')
|
|
filename = os.path.basename(form.file)
|
|
cli = host.get_ssh().get_client()
|
|
sftp = cli.open_sftp()
|
|
f = sftp.open(form.file)
|
|
return FileResponseAfter(cli.close, f, as_attachment=True, filename=filename)
|
|
return json_response(error=error)
|
|
|
|
def post(self, request):
|
|
form, error = JsonParser(
|
|
Argument('id', type=int, help='参数错误'),
|
|
Argument('token', help='参数错误'),
|
|
Argument('path', help='参数错误'),
|
|
).parse(request.POST)
|
|
if error is None:
|
|
if not request.user.has_host_perm(form.id):
|
|
return json_response(error='无权访问主机，请联系管理员')
|
|
file = request.FILES.get('file')
|
|
if not file:
|
|
return json_response(error='请选择要上传的文件')
|
|
host = Host.objects.get(pk=form.id)
|
|
if not host:
|
|
return json_response(error='未找到指定主机')
|
|
cli = host.get_ssh()
|
|
rds_cli = get_redis_connection()
|
|
callback = partial(self._compute_progress, rds_cli, form.token, file.size)
|
|
cli.put_file_by_fl(file, f'{form.path}/{file.name}', callback=callback)
|
|
return json_response(error=error)
|
|
|
|
def delete(self, request):
|
|
form, error = JsonParser(
|
|
Argument('id', type=int, help='参数错误'),
|
|
Argument('file', help='请输入文件路径')
|
|
).parse(request.GET)
|
|
if error is None:
|
|
if not request.user.has_host_perm(form.id):
|
|
return json_response(error='无权访问主机，请联系管理员')
|
|
host = Host.objects.get(pk=form.id)
|
|
if not host:
|
|
return json_response(error='未找到指定主机')
|
|
cli = host.get_ssh()
|
|
cli.remove_file(form.file)
|
|
return json_response(error=error)
|
|
|
|
def _compute_progress(self, rds_cli, token, total, value, *args):
|
|
percent = '%.1f' % (value / total * 100)
|
|
rds_cli.lpush(token, percent)
|
|
rds_cli.expire(token, 300)
|