我编写了一个Python包,其中包含一个bsddb数据库,其中包含一个更耗时的计算的预计算值.为简单起见,我的安装脚本将数据库文件安装在与访问数据库的代码相同的目录中(在Unix上,类似于/usr/lib/python2.5/site-packages/mypackage/).
如何存储数据库文件的最终位置,以便我的代码可以访问它?现在,我正在使用基于__file__
访问数据库的模块中的变量的hack :
dbname = os.path.join(os.path.dirname(__file__), "database.dat")
它有效,但看起来......是hackish.有一个更好的方法吗?我想让安装脚本从distutils模块中获取最终安装位置,并将其填入"dbconfig.py"文件,该文件与访问数据库的代码一起安装.
尝试使用pkg_resources,它是setuptools的一部分(并且可以在我现在可以访问的所有pythons上使用):
>>> 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获取egg页面和pkg_resources页面上的资源的更多讨论.
另请注意,在可能的情况下,建议使用pkg_resources.resource_stream或pkg_resources.resource_string,因为如果包是egg的一部分,resource_filename会将文件复制到临时目录.
使用pkgutil.get_data
.它是表兄pkg_resources.resource_stream
,但在标准库中,应该使用平面文件系统安装以及压缩包和其他导入程序.