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

#
# 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):
        parsed = {'serial_number': '',
                  'disk_setup': '',
                  'partition_setup': list()}
        configs = out.split('\n')
        for line in configs[:-1]:
                if 'ser=' in line:
                        parsed['serial_number'] = line.partition('ser=')[2]
                else:
                        part_setup = {}
                        params = dict(param.split('=') for param in line.split('\t'))
                        # Parse partition configuration.
                        part_setup['disk'] = params['disk']
                        part_setup['partition'] = params['par']
                        part_setup['code'] = params['cpt']
                        part_setup['filesystem'] = params['fsi']
                        part_setup['os'] = params['soi']
                        part_setup['size'] = params['tam']
                        part_setup['used_size'] = params['uso']
                        if part_setup['partition'] == '0':
                                parsed['disk_setup'] = part_setup
                        else:
                                parsed['partition_setup'].append(part_setup)
        return parsed

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

def reboot():
        if os.path.exists('/scripts/oginit'):
                subprocess.call('source ' + OG_PATH + 'etc/preinit/loadenviron.sh; ' + OG_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 = []
        cfg = 'dis=' + disk + '*che=' + cache + '*tch=' + cachesize + '!'

        for part in partlist:
                cfg += 'par=' + part["partition"] + '*cpt='+part["code"] + \
                       '*sfi=' + part['filesystem'] + '*tam=' + \
                       part['size'] + '*ope=' + part['format'] + '%'
                if ogRest.terminated:
                        break

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

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

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):
        try:
                cmd = OG_PATH + 'interfaceAdm/getConfiguration'
                ogRest.proc = subprocess.Popen([cmd],
                                               stdout=subprocess.PIPE,
                                               shell=True)
                (output, error) = ogRest.proc.communicate()
        except:
                raise ValueError('Error: Incorrect command value')

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