在brightway2中指定自定义数据目录

Question

尽管在brightway2中指定自定义数据目录不是标准，但有些人可能仍然需要这样做。一种方法是将python path变量设置为这个自定义目录(请参阅此处：https://docs.brightwaylca.org/faq.html#setting-brightway2-dir-in-a-virtual-environment)，但这需要在执行BRIGHTWAY_DIR脚本之前完成。

因此，不要像下面这样预先设置目录路径(例如Windows)：设置BRIGHTWAY2_DIR=C:/.../MyDataDirectory

是否也可以在python中设置/更改brightway目录？

Answer 1

试试这个：

import os
from brightway2 import projects
from bw2data.project import ProjectDataset, create_database


def switch_directory(dirpath):
    assert os.path.isdir(dirpath)
    projects._base_data_dir = dirpath
    projects._base_logs_dir = os.path.join(dirpath, "logs")
    os.mkdir(projects._base_logs_dir)
    projects.db.close()
    projects.db = create_database(
        os.path.join(projects._base_data_dir, "projects.db"), 
        [ProjectDataset]
    )
    projects.set_current("default")

它使用与test runner基本相同的框架。