CyberPanel/dockerManager/views.py

 # -*- coding: utf-8 -*-


from django.shortcuts import redirect, HttpResponse
from loginSystem.models import Administrator
from loginSystem.views import loadLoginPage
from plogical.httpProc import httpProc
from .container import ContainerManager
from .decorators import preDockerRun
from plogical.acl import ACLManager
import json

# Create your views here.

# This function checks if user has admin permissions
def dockerPermission(request, userID, context):
    currentACL = ACLManager.loadedACL(userID)
    
    if currentACL['admin'] != 1:
        if request.method == "POST":
            return ACLManager.loadErrorJson()
        else:
            return ACLManager.loadError()
    else:
        return 0
        
@preDockerRun
def loadDockerHome(request):
    userID = request.session['userID']
    admin = Administrator.objects.get(pk=userID)
    template = 'dockerManager/index.html'
    proc = httpProc(request, template, {"type": admin.type}, 'admin')
    return proc.render()

def installDocker(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm

        cm = ContainerManager(userID, 'submitInstallDocker')
        cm.start()

        data_ret = {'status': 1, 'error_message': 'None'}
        json_data = json.dumps(data_ret)
        return HttpResponse(json_data)

    except BaseException as msg:
        data_ret = {'status': 0, 'error_message': str(msg)}
        json_data = json.dumps(data_ret)
        return HttpResponse(json_data)

@preDockerRun    
def installImage(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm

        cm = ContainerManager()
        coreResult = cm.submitInstallImage(userID, json.loads(request.body))

        return coreResult

    except KeyError:
        return redirect(loadLoginPage)
    
@preDockerRun
def viewContainer(request, name):
    try:
        if not request.GET._mutable:
            request.GET._mutable = True
        request.GET['name'] = name

        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        cm = ContainerManager(name)
        coreResult = cm.loadContainerHome(request, userID)

        return coreResult

    except KeyError:
        return redirect(loadLoginPage)    

@preDockerRun
def getTags(request): 
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm

        cm = ContainerManager()
        coreResult = cm.getTags(userID, json.loads(request.body))

        return coreResult

    except KeyError:
        return redirect(loadLoginPage)
    
@preDockerRun
def delContainer(request): 
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm

        cm = ContainerManager()
        coreResult = cm.submitContainerDeletion(userID, json.loads(request.body))

        return coreResult

    except KeyError:
        return redirect(loadLoginPage)
    
@preDockerRun    
def recreateContainer(request): 
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm

        cm = ContainerManager()
        coreResult = cm.recreateContainer(userID, json.loads(request.body))

        return coreResult

    except KeyError:
        return redirect(loadLoginPage)    
    
@preDockerRun    
def runContainer(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        return cm.createContainer(request, userID)
    except KeyError:
        return redirect(loadLoginPage)    

@preDockerRun    
def listContainers(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        return cm.listContainers(request, userID)
    except KeyError:
        return redirect(loadLoginPage)

@preDockerRun    
def getContainerLogs(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm

        cm = ContainerManager()
        coreResult = cm.getContainerLogs(userID, json.loads(request.body))
        return coreResult

    except KeyError:
        return redirect(loadLoginPage)    

@preDockerRun    
def submitContainerCreation(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm

        cm = ContainerManager()
        coreResult = cm.submitContainerCreation(userID, json.loads(request.body))

        return coreResult

    except KeyError:
        return redirect(loadLoginPage)

@preDockerRun    
def getContainerList(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        return cm.getContainerList(userID, json.loads(request.body))
    except KeyError:
        return redirect(loadLoginPage)

@preDockerRun    
def doContainerAction(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.doContainerAction(userID, json.loads(request.body))
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)

@preDockerRun    
def getContainerStatus(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.getContainerStatus(userID, json.loads(request.body))
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)    

@preDockerRun    
def exportContainer(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.exportContainer(request, userID)
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)    
        
@preDockerRun        
def saveContainerSettings(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.saveContainerSettings(userID, json.loads(request.body))
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage) 
    
@preDockerRun    
def getContainerTop(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.getContainerTop(userID, json.loads(request.body))
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)

@preDockerRun    
def assignContainer(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.assignContainer(userID, json.loads(request.body))
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)
    
@preDockerRun    
def searchImage(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.searchImage(userID, json.loads(request.body))
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)
    
@preDockerRun    
def images(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'images')
        if perm: return perm
        
        
        cm = ContainerManager()
        coreResult = cm.images(request, userID)
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)
    
@preDockerRun    
def manageImages(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.manageImages(request, userID)
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)
    
@preDockerRun    
def getImageHistory(request):
    try:
        userID = request.session['userID']

        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.getImageHistory(userID, json.loads(request.body))
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)
    
@preDockerRun    
def removeImage(request):
    try:
        userID = request.session['userID']
        perm = dockerPermission(request, userID, 'loadDockerHome')
        if perm: return perm
        
        cm = ContainerManager()
        coreResult = cm.removeImage(userID, json.loads(request.body))
        
        return coreResult
    except KeyError:
        return redirect(loadLoginPage)