source: ogClient-Git/src/linux/ogOperations.py @ ca0a62f

#
# Copyright (C) 2020 Soleta Networks <info@soleta.eu>
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the
# Free Software Foundation, version 3.
#

import os
import subprocess

OG_PATH = '/opt/opengnsys/'

def parseGetConf(out):
        listConfigs = []
        disk = -1;

        configs = out.split('\n')
        configs = filter(None, configs)
        for item in configs:
                i = 0
                json = {}
                val = item.rstrip().split('\t')
                while i < len(val):
                        val[i] = val[i].split('=')[1]
                        i += 1

                json['partition'] = val[1]
                json['code'] = val[4]
                json['filesystem'] = val[2]
                json['size'] = val[5]
                json['format'] = val[6]
                disk = val[0]
                listConfigs.append(json)

        return [disk, listConfigs]

def poweroff():
        if os.path.exists('/scripts/oginit'):
                subprocess.call('source ' + OG_SCRIPT_PATH + 'etc/preinit/loadenviron.sh; ' + OG_SCRIPT_PATH + 'scripts/poweroff', shell=True)
        else:
                subprocess.call(['/sbin/poweroff'])

def reboot():
        if os.path.exists('/scripts/oginit'):
                subprocess.call('source ' + OG_SCRIPT_PATH + 'etc/preinit/loadenviron.sh; ' + OG_SCRIPT_PATH + 'scripts/reboot', shell=True)
        else:
                subprocess.call(['/sbin/reboot'])

def execCMD(request, ogRest):
        cmd = request.getrun()
        cmds = cmd.split(";|\n")
        try:
                ogRest.proc = subprocess.Popen(cmds, stdout=subprocess.PIPE, shell=True)
                (output, error) = ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

        return output.decode('utf-8')

def session(request, ogRest):
        disk = request.getDisk()
        partition = request.getPartition()

        try:
                ogRest.proc = subprocess.Popen([OG_PATH + 'interfaceAdm/IniciarSesion', disk, partition], stdout=subprocess.PIPE, shell=True)
                (output, error) = ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

        return output.decode('utf-8')

def software(request, path, ogRest):
        disk = request.getDisk()
        partition = request.getPartition()

        try:
                cmd = OG_PATH + 'interfaceAdm/InventarioSoftware '
                cmd += str(disk) + ' '
                cmd += str(partition) + ' '
                cmd += path
                ogRest.proc = subprocess.Popen([cmd],
                                               stdout=subprocess.PIPE,
                                               shell=True)
                (output, error) = ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

        return output.decode('utf-8')

def hardware(path, ogRest):
        try:
                cmd = [OG_PATH + 'interfaceAdm/InventarioHardware ' + path]
                ogRest.proc = subprocess.Popen(cmd,
                                               stdout=subprocess.PIPE,
                                               shell=True)
                (output, error) = ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

        return output.decode('utf-8')

def setup(request, ogRest):
        disk = request.getDisk()
        cache = request.getCache()
        cachesize = request.getCacheSize()
        partlist = request.getPartitionSetup()
        listConfigs = []

        for part in partlist:
                cfg = 'dis=' + disk + '*che=' + cache + '*tch=' + cachesize + '!par=' + part["partition"] + '*cpt='+part["code"] + '*sfi=' + part['filesystem'] + '*tam=' + part['size'] + '*ope=' + part['format'] + '%'
                if ogRest.terminated:
                        break

                try:
                        ogRest.proc = subprocess.Popen([OG_PATH + 'interfaceAdm/Configurar', disk, cfg], stdout=subprocess.PIPE, shell=True)
                        (output, error) = ogRest.proc.communicate()
                except:
                        continue

        result = subprocess.check_output([OG_PATH + 'interfaceAdm/getConfiguration'], shell=True)
        return parseGetConf(result.decode('utf-8'))[1]

def image_restore(request, ogRest):
        disk = request.getDisk()
        partition = request.getPartition()
        name = request.getName()
        repo = request.getRepo()
        ctype = request.getType()
        profile = request.getProfile()
        cid = request.getId()

        try:
                ogRest.proc = subprocess.Popen([OG_PATH + 'interfaceAdm/RestaurarImagen', disk, partition, name, repo, ctype], stdout=subprocess.PIPE, shell=True)
                (output, error) = ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

        return output.decode('utf-8')

def image_create(path, request, ogRest):
        disk = request.getDisk()
        partition = request.getPartition()
        name = request.getName()
        repo = request.getRepo()

        try:
                ogRest.proc = subprocess.Popen([OG_PATH + 'interfaceAdm/InventarioSoftware', disk, partition, path], stdout=subprocess.PIPE, shell=True)
                (output, error) = ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

        if ogRest.terminated:
                return

        try:
                ogRest.proc = subprocess.Popen([OG_PATH + 'interfaceAdm/CrearImagen', disk, partition, name, repo], stdout=subprocess.PIPE, shell=True)
                ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

        return output.decode('utf-8')

def refresh(ogRest):
        listConfigs = []
        disk = -1;

        try:
                ogRest.proc = subprocess.Popen([OG_PATH + 'interfaceAdm/getConfiguration'], stdout=subprocess.PIPE, shell=True)
                (output, error) = ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

        return parseGetConf(output.decode('utf-8'))