Django运维系统之---批量执行shell脚本，并记录操作用于审计
背景
为了进一步完善自己写的小运维系统，今天就继续来补充一个批量执行shell脚本，并记录操作用于审计！
一、思路
实现批量执行shell脚本刚开始的思路是，获取本地脚本的路径，用Paramiko上传脚本，给权限，执行，反馈结果。但是没过一会就醒悟，呀，这不符合一个系统的思路阿！这不是脚本呀！所以需要将本地的脚本上传到系统地址，再上传到服务器进行执行才对！
那么思路就明确啦
1.当建立websocket与ssh连接后，上传脚本到系统
2.由系统上传脚本导服务器
3.给脚本权限并执行脚本
4.反馈脚本信息
二、实现
1.当建立websocket与ssh连接后，上传脚本到系统
一开始是想既然建立了websocket连接了，直接用websocket上传文件就好啦，无奈测试了几遍都没达到效果，估计是我这才刚开始用websocket吧。下面是上传文件的实现的核心：
html：
<!--选择文件-->
<label class="form-inline">执行的shell脚本:
<input id="myfile" type="file" accept=".sh" style="display: none" multiple>
<!--显示上传的文件名字-->
<input type="text" id="showfilename" class="form-control" value="点击选择脚本" readonly="true" onclick="$('input[id=myfile]').click();">
<button class="btn btn-primary " type="button" id="shangchuan" >上传</button>
<button class="btn btn-success " type="button" id="sendbashscrip">执行</button><br>
（先上传脚本，再执行脚本；<br>上传后将脚本上传到/usr/目录，脚本命名为：名字+上传时间；<br>
自动给予脚本755权限；）
</label>
js：
$('input[id=myfile]').change(function () {
var myfiles = document.getElementById("myfile").files;
var filename = "";
filename = myfiles.name;
$('#showfilename').val(filename);
})
function Upload(){
var myfiles = document.getElementById("myfile").files;
var data = new FormData();
data.append('file', myfiles);
$.ajax({
type: "POST",
url: "{% url 'Upload' %}",
data: data,
processData: false,
contentType: false,
dataType:"json",
success: function(result){
}
});
}
上传逻辑函数：
@csrf_exempt
def Upload(req):
global file_name
file = req.FILES.get('file')
nowtime = time.strftime("%Y%m%d%H%M%S")
uploadshellpath = settings.BASE_DIR + '/static/batchshellscript/'
if not os.path.exists(uploadshellpath):
os.makedirs(uploadshellpath)
if file:
file_name = nowtime+file.name
shellwrite= open(uploadshellpath+file_name, 'wb+')
for chunk in file.chunks():
shellwrite.write(chunk)
shellwrite.close()
file_path = uploadshellpath + file_name
json_data = {}
response_data['filename'] = file_name
response_data['path'] = file_path
json_data = json.dumps(response_data)
return HttpResponse(json_data)
批量处理脚本逻辑函数：
在昨天batch函数的基础上修改这部分逻辑：
for shell in request.websocket:
if shell.decode('utf-8') == 'sendbatchscript':
addactionshell = models.batchaction()
addactionshell.hostaddress = hostip
addactionshell.username = request.session['username']
addactionshell.actionshell = 'acction:' + globals()['file_name']
addactionshell.starttime = time.strftime("%Y%m%d%H%M%S")
addactionshell.save()
for i in range(hostnum):
try :
e[i].put(settings.BASE_DIR + r'/static/batchshellscript/'+globals()['file_name'],r'/usr/'+globals()['file_name'])
cmd = 'cd /usr;chmod 755 '+ globals()['file_name']+';' + './' + globals()['file_name']
b[i], c[i], d[i] = a[i].exec_command(cmd) # num个对象执行e命令
output = '-' * 97 + f'{hostip[i]}执行结果' + '-' * 90 + c[i].read().decode('utf-8') + d[i].read().decode('utf-8')
request.websocket.send(output.encode('utf-8'))
except:
print('脚本执行失败')
shellscrpmess = '-' * 97 + f'{hostip[i]}执行结果' + '-' * 94 +'执行失败！'
request.websocket.send(shellscrpmess.encode('utf-8'))
else:
addactionshell = models.batchaction()
addactionshell.hostaddress = hostip
addactionshell.username = request.session['username']
addactionshell.actionshell = shell.decode('utf-8')
addactionshell.starttime = time.strftime("%Y%m%d%H%M%S")
addactionshell.save()
for i in range(hostnum):
b[i], c[i], d[i] = a[i].exec_command(shell) # num个对象执行e命令
# print(f'----------------------{hostip[i]}执行结果----------------------')
# print(c[i].read().decode('utf-8')) # 打印正确输出
# print(d[i].read().decode('utf-8')) # 打印错误输出
output = '-'*97 + f'{hostip[i]}执行结果' + '-'*90 + c[i].read().decode('utf-8') + d[i].read().decode('utf-8')
request.websocket.send(output.encode('utf-8'))
效果：
执行一个helloword脚本：
审计：
目前实现批量执行shell脚本的思路与实现就是这个样子啦~要实现成什么样子就看大家怎么去做了。
（个人感觉这一个页面按钮太多了。。。但是更高级的技术或者实现技巧目前还不太会，只能慢慢改善吧，有大佬指点指点就更好拉~）
三、结束
好了，今天又实现一个小功能，只能说只是实现了哈，人性化或者隐藏bug多得很！！