pystow

import pystow import os # Configure a custom base directory for PyStow (optional, for testing/isolation) # Defaults to ~/.data if not set os.environ['PYSTOW_HOME'] = os.path.join(os.getcwd(), 'my_pystow_data') # Get a directory for your application data app_dir = pystow.join('my_app') print(f"Application data directory: {app_dir}") app_dir.mkdir(parents=True, exist_ok=True) # Ensure a file is downloaded from a URL into a specific subdirectory url = 'https://raw.githubusercontent.com/pykeen/pykeen/master/src/pykeen/datasets/nations/test.txt' file_path = pystow.ensure( 'my_app', 'datasets', 'nations', url=url, name='nations_test.txt' ) print(f"Ensured file path: {file_path}") # You can also use modules for cleaner access my_app_module = pystow.module('my_app') module_file_path = my_app_module.ensure( 'another_data', url='https://example.com/some_file.txt', name='some_file.txt' # This URL is a placeholder and will likely fail. ) print(f"Ensured file via module: {module_file_path}") # Clean up the custom directory for demonstration # import shutil # if os.path.exists(os.environ['PYSTOW_HOME']): # shutil.rmtree(os.environ['PYSTOW_HOME']) # del os.environ['PYSTOW_HOME']