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

CPANEL_CONFIG_PATH = "/var/cpanel/cpanel.config"

def get_param(path, param):
    """
    Get param from config file
    :param path: Path to config file
    :type path: str
    :param param: Key for param in config file
    :type param: str
    """
    try:
        with open(path, encoding="utf-8") as f:
            data = f.readlines()
    except (OSError, IOError):
        return False
    for line in data:
        if line.startswith(f"{param}="):
            return (line.strip().split("="))[1]
    return None


def replace_param(path, param, value):
    """
    Replaces config file with path to config file with line param=value
    :param path: Path to config file
    :type path: str
    :param param: Key for param in config file
    :type key: str
    :param value: Value for param in config file
    :type value: str
    """
    new_line = f'{param}={value}\n'
    if not os.path.exists(path):
        # config file is absent
        with open(path, 'w', encoding='utf-8') as f:
            f.write(new_line)
        return
    # config file present
    with open(path, encoding='utf-8') as f:
        data = f.readlines()
    for i, line in enumerate(data):
        if line.startswith(f'{param}='):
            data[i] = new_line
            with open(path, 'w', encoding='utf-8') as f:
                f.write(''.join(data))
            return
    # add new line with given param if it wasn't in original config
    with open(path, 'a', encoding='utf-8') as f:
        f.write(new_line)