我编写了一个Python程序包，其中包含一个bsddb数据库，该数据库包含预先计算的值，用于更耗时的计算之一。为简单起见，我的安装脚本将数据库文件安装在与访问数据库的代码相同的目录中(在Unix上，类似于/usr/lib/python2.5/site-packages/mypackage/)。

如何存储数据库文件的最终位置，以便我的代码可以访问它？现在，我正在基于访问数据库的模块中的__file__变量使用黑客：

dbname = os.path.join(os.path.dirname(__file__), "database.dat")

它可以工作，但似乎...有点黑。有一个更好的方法吗？我想让安装脚本从distutils模块中获取最终安装位置，并将其填充到一个" dbconfig.py"文件中，该文件与访问数据库的代码一起安装。

解决方案

回答

这可能就是这样做的方法，而无需诉诸更高级的功能，例如使用setuptools将文件安装在它们所属的位置。

请注意，这种方法存在问题，因为在具有真正安全框架(UNIX等)的操作系统上，运行脚本的用户可能无权访问安装了该数据库的系统目录中的数据库。

回答

尝试使用pkg_resources，它是setuptools的一部分(并且在我现在可以访问的所有python上都可用)：

>>> import pkg_resources
>>> pkg_resources.resource_ filename(__name__, "foo.config")
'foo.config'
>>> pkg_resources.resource_filename('tempfile', "foo.config")
'/usr/lib/python2.4/foo.config'

有关使用pkg_resources来获取鸡蛋页面和pkg_resources页面上的资源的更多讨论。

还要注意，在可能的情况下，建议使用pkg_resources.resource_stream或者pkg_resources.resource_string，因为如果包是鸡蛋的一部分，则resource_filename会将文件复制到一个临时目录中。