# -*- 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)
|