Mini Shell

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

# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2018 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
#

import os

PROC_LVE = '/proc/lve'
LIMIT_LVP_ID = 4294967295


class ProcLveError(Exception):
    pass


class NoSuchLvp(ProcLveError):
    pass


class ProcLve:

    def __init__(self, proc_lve=PROC_LVE):
        self._lvp = 'lvp'
        self.list = 'list'
        self.proc_lve = proc_lve
        self._version = None

    def proc_lve_list(self):
        """
        Return path to /proc/lve/list
        """
        return os.path.join(self.proc_lve, self.list)

    def version(self):
        """
        Return /proc/lve filesystem version
        """
        if self._version is not None:
            return self._version
        with open(self.proc_lve_list(), encoding='utf-8') as f:
            line = f.read(3)  # we need only first three symbols for parse version
        self._version = int(line.rsplit(':', 1)[0])
        return self._version

    def proc_lve_map(self):
        return os.path.join(self.proc_lve, 'map')

    def proc_lve_resellers(self):
        """
        Return path to /proc/lve/resellers
        """
        return os.path.join(self.proc_lve, 'resellers')

    def resellers_supported(self):
        """
        Check present /proc/lve/resellers
        """
        return os.path.exists(self.proc_lve_resellers())

    def get_list_path(self, lvp_id=0):
        """
        Generate path to list file
        """
        if lvp_id == 0:
            return self.proc_lve_list()
        else:
            return os.path.join(self.proc_lve_resellers(), self._lvp + str(lvp_id), self.list)

    def _extract_id(self, line):
        return int(line.split()[0].split(',')[-1])

    def _lines(self, lvp_id=0):
        path = self.get_list_path(lvp_id=lvp_id)
        try:
            with open(path, encoding='utf-8') as list_:
                lines = list_.read().strip()
        except IOError as e:
            if e.errno == 2:  # 2 code error - No such file or directory
                raise NoSuchLvp(f'No such lvp id {lvp_id}') from e
            else:
                raise
        lines = lines.split('\n')[1:]
        for line in lines:
            yield self._extract_id(line), line.rstrip(os.linesep)

    def lines(self, lvp_id=0, without_limits=True):
        for lve_id, line in self._lines(lvp_id=lvp_id):
            if lve_id == LIMIT_LVP_ID and without_limits:
                continue
            yield line

    def lve_id_list(self, lvp_id=0):
        for lve_id, _ in self._lines(lvp_id=lvp_id):
            if lve_id in (LIMIT_LVP_ID, 0):
                continue
            yield lve_id

    def lvp_id_list(self):
        """
        Obtain from /proc/lve/resellers lvp id list
        """
        for lvp_dir in os.listdir(self.proc_lve_resellers()):
            if lvp_dir.startswith(self._lvp):
                yield int(lvp_dir[len(self._lvp):])

    def check_inside_list(self, id_, lvp_id=0):
        for lve_id in self.lve_id_list(lvp_id=lvp_id):
            if lve_id == id_:
                return True

    def exist_lvp(self, lvp_id):
        """
        Check present lve top container
        """
        return os.path.exists(self.get_list_path(lvp_id))

    def detect_inside_lvp(self, id_):
        """
        Find in which lve top container
        """
        for lvp_id in self.lvp_id_list():
            if self.exist_lvp(lvp_id) and self.check_inside_list(id_, lvp_id):
                return lvp_id

    def map(self):
        """
        Obtain map from /proc/lve/map as dict
        """
        map_dict = {}
        with open(self.proc_lve_map(), encoding='utf-8') as map_:
            lines = map_.read().strip()
        lines = lines.split('\n')[1:]
        for line in lines:
            lve_id_str, lvp_id_str = line.strip().split()
            map_dict[int(lve_id_str)] = int(lvp_id_str)
        return map_dict

    def map_lve_id_list(self, lvp_id):
        return [lve_id_ for lve_id_, lvp_id_ in list(self.map().items()) if lvp_id_ == lvp_id]