如何从另一个 python 文件写入 python 驱动的配置文件

Question

我有 2 个 python 文件。 一个是我的主文件 mainFile.py 另一个文件基本上是配置文件 config.py

config.py 的结构

class Configurations(object):
    CONFIGURE_VM = True
    CPU_CORES = 2
    RAM = 96
    THREADS = 2

这就是我在 mainFile.py 中读取 config.py 值的方式：-

from config import Configurations

def writeValues():
    vm_configure = Configurations.CONFIGURE_VM
    vcpu = Configurations.CPU_CORES
    memory = Configurations.RAM

在我的 mainFile.py 中，我需要编写代码来更改所有上述参数的值，以便它们可以变为：-

class Configurations(object):
    CONFIGURE_VM = False
    CPU_CORES = 1
    RAM = 32

用例是每次 mainFile.py 运行时从 config.py 读取新值并在其中写入新值。

有什么帮助吗？

Answer 1

我可以通过使用 python FileInput 库来解决我自己的问题。

def writeValues(filename, input, output):
cwd = os.getcwd()
filename = os.path.join(cwd[:cwd.find('home')], 'home', 'config', filename)
for line in fileinput.input([filename], inplace=True):
    line = line.strip("\n").split("=")
    if input in line[0]:
        line[1] = " " + str(output)
    line = '='.join(line)
    print(line)