Datos de acceso Python en el subdirectorio de paquete

Question

Estoy escribiendo un paquete python con módulos que necesitan abrir archivos de datos en un subdirectorio ./data/. En este momento tengo las rutas a los archivos codificados en mis clases y funciones. Me gustaría escribir un código más robusto que pueda acceder al subdirectorio independientemente de dónde esté instalado en el sistema del usuario.

He intentado una variedad de métodos, pero hasta ahora no he tenido suerte. Parece que la mayoría de los comandos del "directorio actual" devuelven el directorio del intérprete python del sistema, y ​​no el directorio del módulo.

Parece que debería ser un problema común y trivial. Sin embargo, parece que no puedo resolverlo. Parte del problema es que mis archivos de datos no son archivos .py, por lo que no puedo usar funciones de importación y similares.

¿Alguna sugerencia?

Ahora mismo mi directorio del paquete se parece a:

/ 
__init__.py 
module1.py 
module2.py 
data/ 
    data.txt 

Estoy tratando de acceder data.txt de module*.py

Gracias!

Answer 1

Puede usar guión bajo-underscore- archivo-guión-guión bajo (__file__) para obtener la ruta al paquete, así:

import os 
this_dir, this_filename = os.path.split(__file__) 
DATA_PATH = os.path.join(this_dir, "data", "data.txt") 
print open(DATA_PATH).read() 

Answer 2

Creo que he buscado una respuesta.

mía es data_path.py módulo, que importo en mis otros módulos que contienen:

data_path = os.path.join(os.path.dirname(__file__),'data') 

Y entonces abro todos mis archivos con

open(os.path.join(data_path,'filename'), <param>) 

Answer 3

La forma estándar de hacerlo es con los paquetes setuptools y pkg_resources.

Usted puede diseñar su paquete de acuerdo con la siguiente jerarquía, y configurar el archivo de instalación del paquete de señalar que los recursos de datos, de acuerdo con este enlace:

http://docs.python.org/distutils/setupscript.html#installing-package-data

A continuación, puede volver a encontrar y utilizar esos archivos utilizando pkg_resources, según este enlace:

http://peak.telecommunity.com/DevCenter/PkgResources#basic-resource-access

import pkg_resources 

DATA_PATH = pkg_resources.resource_filename('<package name>', 'data/') 
DB_FILE = pkg_resources.resource_filename('<package name>', 'data/sqlite.db') 

Answer 4

Para proporcionar una solución de trabajo hoy. Definitivamente use esta API para no reinventar todas esas ruedas.

Se necesita un verdadero nombre de archivo del sistema de archivos. Los huevos comprimidos se extraerán en un directorio de caché:

from pkg_resources import resource_filename, Requirement 

path_to_vik_logo = resource_filename(Requirement.parse("enb.portals"), "enb/portals/reports/VIK_logo.png") 

Devuelve un objeto similar a un archivo legible para el recurso especificado; puede ser un archivo real, un StringIO u otro objeto similar. La secuencia está en "modo binario", en el sentido de que los bytes que se encuentran en el recurso se leerán tal cual.árbol de directorios

from pkg_resources import resource_stream, Requirement 

vik_logo_as_stream = resource_stream(Requirement.parse("enb.portals"), "enb/portals/reports/VIK_logo.png") 

Paquete Descubrimiento y acceso al recurso utilizando pkg_resources

• https://setuptools.readthedocs.io/en/latest/pkg_resources.html#resource-extraction
• https://setuptools.readthedocs.io/en/latest/pkg_resources.html#basic-resource-access

Answer 5

Es necesario un nombre para todo el módulo, que está dado no lo hace la lista ese detalle, para mí esto funcionó:

import pkg_resources 
print( 
    pkg_resources.resource_filename(__name__, 'data/data.txt') 
) 

Parece que las herramientas de configuración no resuelven los archivos basados ​​en una coincidencia de nombre con los archivos de datos empaquetados, por lo que tiene que incluir el prefijo data/ prácticamente sin importar nada. Puede usar os.path.join('data', 'data.txt) si necesita separadores de directorio alternativos. Sin embargo, en general no encuentro problemas de compatibilidad con los separadores de directorios de estilo Unix codificados.