warp-lang 1.0.2__py3-none-manylinux2014_x86_64.whl → 1.1.0__py3-none-manylinux2014_x86_64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of warp-lang might be problematic. Click here for more details.

Files changed (346) hide show
  1. warp/__init__.py +108 -97
  2. warp/__init__.pyi +1 -1
  3. warp/bin/warp-clang.so +0 -0
  4. warp/bin/warp.so +0 -0
  5. warp/build.py +115 -113
  6. warp/build_dll.py +383 -375
  7. warp/builtins.py +3425 -3354
  8. warp/codegen.py +2878 -2792
  9. warp/config.py +40 -36
  10. warp/constants.py +45 -45
  11. warp/context.py +5194 -5102
  12. warp/dlpack.py +442 -442
  13. warp/examples/__init__.py +16 -16
  14. warp/examples/assets/bear.usd +0 -0
  15. warp/examples/assets/bunny.usd +0 -0
  16. warp/examples/assets/cartpole.urdf +110 -110
  17. warp/examples/assets/crazyflie.usd +0 -0
  18. warp/examples/assets/cube.usd +0 -0
  19. warp/examples/assets/nv_ant.xml +92 -92
  20. warp/examples/assets/nv_humanoid.xml +183 -183
  21. warp/examples/assets/quadruped.urdf +267 -267
  22. warp/examples/assets/rocks.nvdb +0 -0
  23. warp/examples/assets/rocks.usd +0 -0
  24. warp/examples/assets/sphere.usd +0 -0
  25. warp/examples/benchmarks/benchmark_api.py +383 -383
  26. warp/examples/benchmarks/benchmark_cloth.py +278 -277
  27. warp/examples/benchmarks/benchmark_cloth_cupy.py +88 -88
  28. warp/examples/benchmarks/benchmark_cloth_jax.py +97 -100
  29. warp/examples/benchmarks/benchmark_cloth_numba.py +146 -142
  30. warp/examples/benchmarks/benchmark_cloth_numpy.py +77 -77
  31. warp/examples/benchmarks/benchmark_cloth_pytorch.py +86 -86
  32. warp/examples/benchmarks/benchmark_cloth_taichi.py +112 -112
  33. warp/examples/benchmarks/benchmark_cloth_warp.py +146 -146
  34. warp/examples/benchmarks/benchmark_launches.py +295 -295
  35. warp/examples/browse.py +29 -29
  36. warp/examples/core/example_dem.py +234 -219
  37. warp/examples/core/example_fluid.py +293 -267
  38. warp/examples/core/example_graph_capture.py +144 -126
  39. warp/examples/core/example_marching_cubes.py +188 -174
  40. warp/examples/core/example_mesh.py +174 -155
  41. warp/examples/core/example_mesh_intersect.py +205 -193
  42. warp/examples/core/example_nvdb.py +176 -170
  43. warp/examples/core/example_raycast.py +105 -90
  44. warp/examples/core/example_raymarch.py +199 -178
  45. warp/examples/core/example_render_opengl.py +185 -141
  46. warp/examples/core/example_sph.py +405 -387
  47. warp/examples/core/example_torch.py +222 -181
  48. warp/examples/core/example_wave.py +263 -248
  49. warp/examples/fem/bsr_utils.py +378 -380
  50. warp/examples/fem/example_apic_fluid.py +407 -389
  51. warp/examples/fem/example_convection_diffusion.py +182 -168
  52. warp/examples/fem/example_convection_diffusion_dg.py +219 -209
  53. warp/examples/fem/example_convection_diffusion_dg0.py +204 -194
  54. warp/examples/fem/example_deformed_geometry.py +177 -159
  55. warp/examples/fem/example_diffusion.py +201 -173
  56. warp/examples/fem/example_diffusion_3d.py +177 -152
  57. warp/examples/fem/example_diffusion_mgpu.py +221 -214
  58. warp/examples/fem/example_mixed_elasticity.py +244 -222
  59. warp/examples/fem/example_navier_stokes.py +259 -243
  60. warp/examples/fem/example_stokes.py +220 -192
  61. warp/examples/fem/example_stokes_transfer.py +265 -249
  62. warp/examples/fem/mesh_utils.py +133 -109
  63. warp/examples/fem/plot_utils.py +292 -287
  64. warp/examples/optim/example_bounce.py +260 -246
  65. warp/examples/optim/example_cloth_throw.py +222 -209
  66. warp/examples/optim/example_diffray.py +566 -536
  67. warp/examples/optim/example_drone.py +864 -835
  68. warp/examples/optim/example_inverse_kinematics.py +176 -168
  69. warp/examples/optim/example_inverse_kinematics_torch.py +185 -169
  70. warp/examples/optim/example_spring_cage.py +239 -231
  71. warp/examples/optim/example_trajectory.py +223 -199
  72. warp/examples/optim/example_walker.py +306 -293
  73. warp/examples/sim/example_cartpole.py +139 -129
  74. warp/examples/sim/example_cloth.py +196 -186
  75. warp/examples/sim/example_granular.py +124 -111
  76. warp/examples/sim/example_granular_collision_sdf.py +197 -186
  77. warp/examples/sim/example_jacobian_ik.py +236 -214
  78. warp/examples/sim/example_particle_chain.py +118 -105
  79. warp/examples/sim/example_quadruped.py +193 -180
  80. warp/examples/sim/example_rigid_chain.py +197 -187
  81. warp/examples/sim/example_rigid_contact.py +189 -177
  82. warp/examples/sim/example_rigid_force.py +127 -125
  83. warp/examples/sim/example_rigid_gyroscopic.py +109 -95
  84. warp/examples/sim/example_rigid_soft_contact.py +134 -122
  85. warp/examples/sim/example_soft_body.py +190 -177
  86. warp/fabric.py +337 -335
  87. warp/fem/__init__.py +60 -27
  88. warp/fem/cache.py +401 -388
  89. warp/fem/dirichlet.py +178 -179
  90. warp/fem/domain.py +262 -263
  91. warp/fem/field/__init__.py +100 -101
  92. warp/fem/field/field.py +148 -149
  93. warp/fem/field/nodal_field.py +298 -299
  94. warp/fem/field/restriction.py +22 -21
  95. warp/fem/field/test.py +180 -181
  96. warp/fem/field/trial.py +183 -183
  97. warp/fem/geometry/__init__.py +15 -19
  98. warp/fem/geometry/closest_point.py +69 -70
  99. warp/fem/geometry/deformed_geometry.py +270 -271
  100. warp/fem/geometry/element.py +744 -744
  101. warp/fem/geometry/geometry.py +184 -186
  102. warp/fem/geometry/grid_2d.py +380 -373
  103. warp/fem/geometry/grid_3d.py +441 -435
  104. warp/fem/geometry/hexmesh.py +953 -953
  105. warp/fem/geometry/partition.py +374 -376
  106. warp/fem/geometry/quadmesh_2d.py +532 -532
  107. warp/fem/geometry/tetmesh.py +840 -840
  108. warp/fem/geometry/trimesh_2d.py +577 -577
  109. warp/fem/integrate.py +1630 -1615
  110. warp/fem/operator.py +190 -191
  111. warp/fem/polynomial.py +214 -213
  112. warp/fem/quadrature/__init__.py +2 -2
  113. warp/fem/quadrature/pic_quadrature.py +243 -245
  114. warp/fem/quadrature/quadrature.py +295 -294
  115. warp/fem/space/__init__.py +294 -292
  116. warp/fem/space/basis_space.py +488 -489
  117. warp/fem/space/collocated_function_space.py +100 -105
  118. warp/fem/space/dof_mapper.py +236 -236
  119. warp/fem/space/function_space.py +148 -145
  120. warp/fem/space/grid_2d_function_space.py +267 -267
  121. warp/fem/space/grid_3d_function_space.py +305 -306
  122. warp/fem/space/hexmesh_function_space.py +350 -352
  123. warp/fem/space/partition.py +350 -350
  124. warp/fem/space/quadmesh_2d_function_space.py +368 -369
  125. warp/fem/space/restriction.py +158 -160
  126. warp/fem/space/shape/__init__.py +13 -15
  127. warp/fem/space/shape/cube_shape_function.py +738 -738
  128. warp/fem/space/shape/shape_function.py +102 -103
  129. warp/fem/space/shape/square_shape_function.py +611 -611
  130. warp/fem/space/shape/tet_shape_function.py +565 -567
  131. warp/fem/space/shape/triangle_shape_function.py +429 -429
  132. warp/fem/space/tetmesh_function_space.py +294 -292
  133. warp/fem/space/topology.py +297 -295
  134. warp/fem/space/trimesh_2d_function_space.py +223 -221
  135. warp/fem/types.py +77 -77
  136. warp/fem/utils.py +495 -495
  137. warp/jax.py +166 -141
  138. warp/jax_experimental.py +341 -339
  139. warp/native/array.h +1072 -1025
  140. warp/native/builtin.h +1560 -1560
  141. warp/native/bvh.cpp +398 -398
  142. warp/native/bvh.cu +525 -525
  143. warp/native/bvh.h +429 -429
  144. warp/native/clang/clang.cpp +495 -464
  145. warp/native/crt.cpp +31 -31
  146. warp/native/crt.h +334 -334
  147. warp/native/cuda_crt.h +1049 -1049
  148. warp/native/cuda_util.cpp +549 -540
  149. warp/native/cuda_util.h +288 -203
  150. warp/native/cutlass_gemm.cpp +34 -34
  151. warp/native/cutlass_gemm.cu +372 -372
  152. warp/native/error.cpp +66 -66
  153. warp/native/error.h +27 -27
  154. warp/native/fabric.h +228 -228
  155. warp/native/hashgrid.cpp +301 -278
  156. warp/native/hashgrid.cu +78 -77
  157. warp/native/hashgrid.h +227 -227
  158. warp/native/initializer_array.h +32 -32
  159. warp/native/intersect.h +1204 -1204
  160. warp/native/intersect_adj.h +365 -365
  161. warp/native/intersect_tri.h +322 -322
  162. warp/native/marching.cpp +2 -2
  163. warp/native/marching.cu +497 -497
  164. warp/native/marching.h +2 -2
  165. warp/native/mat.h +1498 -1498
  166. warp/native/matnn.h +333 -333
  167. warp/native/mesh.cpp +203 -203
  168. warp/native/mesh.cu +293 -293
  169. warp/native/mesh.h +1887 -1887
  170. warp/native/nanovdb/NanoVDB.h +4782 -4782
  171. warp/native/nanovdb/PNanoVDB.h +2553 -2553
  172. warp/native/nanovdb/PNanoVDBWrite.h +294 -294
  173. warp/native/noise.h +850 -850
  174. warp/native/quat.h +1084 -1084
  175. warp/native/rand.h +299 -299
  176. warp/native/range.h +108 -108
  177. warp/native/reduce.cpp +156 -156
  178. warp/native/reduce.cu +348 -348
  179. warp/native/runlength_encode.cpp +61 -61
  180. warp/native/runlength_encode.cu +46 -46
  181. warp/native/scan.cpp +30 -30
  182. warp/native/scan.cu +36 -36
  183. warp/native/scan.h +7 -7
  184. warp/native/solid_angle.h +442 -442
  185. warp/native/sort.cpp +94 -94
  186. warp/native/sort.cu +97 -97
  187. warp/native/sort.h +14 -14
  188. warp/native/sparse.cpp +337 -337
  189. warp/native/sparse.cu +544 -544
  190. warp/native/spatial.h +630 -630
  191. warp/native/svd.h +562 -562
  192. warp/native/temp_buffer.h +30 -30
  193. warp/native/vec.h +1132 -1132
  194. warp/native/volume.cpp +297 -297
  195. warp/native/volume.cu +32 -32
  196. warp/native/volume.h +538 -538
  197. warp/native/volume_builder.cu +425 -425
  198. warp/native/volume_builder.h +19 -19
  199. warp/native/warp.cpp +1057 -1052
  200. warp/native/warp.cu +2943 -2828
  201. warp/native/warp.h +313 -305
  202. warp/optim/__init__.py +9 -9
  203. warp/optim/adam.py +120 -120
  204. warp/optim/linear.py +1104 -939
  205. warp/optim/sgd.py +104 -92
  206. warp/render/__init__.py +10 -10
  207. warp/render/render_opengl.py +3217 -3204
  208. warp/render/render_usd.py +768 -749
  209. warp/render/utils.py +152 -150
  210. warp/sim/__init__.py +52 -59
  211. warp/sim/articulation.py +685 -685
  212. warp/sim/collide.py +1594 -1590
  213. warp/sim/import_mjcf.py +489 -481
  214. warp/sim/import_snu.py +220 -221
  215. warp/sim/import_urdf.py +536 -516
  216. warp/sim/import_usd.py +887 -881
  217. warp/sim/inertia.py +316 -317
  218. warp/sim/integrator.py +234 -233
  219. warp/sim/integrator_euler.py +1956 -1956
  220. warp/sim/integrator_featherstone.py +1910 -1991
  221. warp/sim/integrator_xpbd.py +3294 -3312
  222. warp/sim/model.py +4473 -4314
  223. warp/sim/particles.py +113 -112
  224. warp/sim/render.py +417 -403
  225. warp/sim/utils.py +413 -410
  226. warp/sparse.py +1227 -1227
  227. warp/stubs.py +2109 -2469
  228. warp/tape.py +1162 -225
  229. warp/tests/__init__.py +1 -1
  230. warp/tests/__main__.py +4 -4
  231. warp/tests/assets/torus.usda +105 -105
  232. warp/tests/aux_test_class_kernel.py +26 -26
  233. warp/tests/aux_test_compile_consts_dummy.py +10 -10
  234. warp/tests/aux_test_conditional_unequal_types_kernels.py +21 -21
  235. warp/tests/aux_test_dependent.py +22 -22
  236. warp/tests/aux_test_grad_customs.py +23 -23
  237. warp/tests/aux_test_reference.py +11 -11
  238. warp/tests/aux_test_reference_reference.py +10 -10
  239. warp/tests/aux_test_square.py +17 -17
  240. warp/tests/aux_test_unresolved_func.py +14 -14
  241. warp/tests/aux_test_unresolved_symbol.py +14 -14
  242. warp/tests/disabled_kinematics.py +239 -239
  243. warp/tests/run_coverage_serial.py +31 -31
  244. warp/tests/test_adam.py +157 -157
  245. warp/tests/test_arithmetic.py +1124 -1124
  246. warp/tests/test_array.py +2417 -2326
  247. warp/tests/test_array_reduce.py +150 -150
  248. warp/tests/test_async.py +668 -656
  249. warp/tests/test_atomic.py +141 -141
  250. warp/tests/test_bool.py +204 -149
  251. warp/tests/test_builtins_resolution.py +1292 -1292
  252. warp/tests/test_bvh.py +164 -171
  253. warp/tests/test_closest_point_edge_edge.py +228 -228
  254. warp/tests/test_codegen.py +566 -553
  255. warp/tests/test_compile_consts.py +97 -101
  256. warp/tests/test_conditional.py +246 -246
  257. warp/tests/test_copy.py +232 -215
  258. warp/tests/test_ctypes.py +632 -632
  259. warp/tests/test_dense.py +67 -67
  260. warp/tests/test_devices.py +91 -98
  261. warp/tests/test_dlpack.py +530 -529
  262. warp/tests/test_examples.py +400 -378
  263. warp/tests/test_fabricarray.py +955 -955
  264. warp/tests/test_fast_math.py +62 -54
  265. warp/tests/test_fem.py +1277 -1278
  266. warp/tests/test_fp16.py +130 -130
  267. warp/tests/test_func.py +338 -337
  268. warp/tests/test_generics.py +571 -571
  269. warp/tests/test_grad.py +746 -640
  270. warp/tests/test_grad_customs.py +333 -336
  271. warp/tests/test_hash_grid.py +210 -164
  272. warp/tests/test_import.py +39 -39
  273. warp/tests/test_indexedarray.py +1134 -1134
  274. warp/tests/test_intersect.py +67 -67
  275. warp/tests/test_jax.py +307 -307
  276. warp/tests/test_large.py +167 -164
  277. warp/tests/test_launch.py +354 -354
  278. warp/tests/test_lerp.py +261 -261
  279. warp/tests/test_linear_solvers.py +191 -171
  280. warp/tests/test_lvalue.py +421 -493
  281. warp/tests/test_marching_cubes.py +65 -65
  282. warp/tests/test_mat.py +1801 -1827
  283. warp/tests/test_mat_lite.py +115 -115
  284. warp/tests/test_mat_scalar_ops.py +2907 -2889
  285. warp/tests/test_math.py +126 -193
  286. warp/tests/test_matmul.py +500 -499
  287. warp/tests/test_matmul_lite.py +410 -410
  288. warp/tests/test_mempool.py +188 -190
  289. warp/tests/test_mesh.py +284 -324
  290. warp/tests/test_mesh_query_aabb.py +228 -241
  291. warp/tests/test_mesh_query_point.py +692 -702
  292. warp/tests/test_mesh_query_ray.py +292 -303
  293. warp/tests/test_mlp.py +276 -276
  294. warp/tests/test_model.py +110 -110
  295. warp/tests/test_modules_lite.py +39 -39
  296. warp/tests/test_multigpu.py +163 -163
  297. warp/tests/test_noise.py +248 -248
  298. warp/tests/test_operators.py +250 -250
  299. warp/tests/test_options.py +123 -125
  300. warp/tests/test_peer.py +133 -137
  301. warp/tests/test_pinned.py +78 -78
  302. warp/tests/test_print.py +54 -54
  303. warp/tests/test_quat.py +2086 -2086
  304. warp/tests/test_rand.py +288 -288
  305. warp/tests/test_reload.py +217 -217
  306. warp/tests/test_rounding.py +179 -179
  307. warp/tests/test_runlength_encode.py +190 -190
  308. warp/tests/test_sim_grad.py +243 -0
  309. warp/tests/test_sim_kinematics.py +91 -97
  310. warp/tests/test_smoothstep.py +168 -168
  311. warp/tests/test_snippet.py +305 -266
  312. warp/tests/test_sparse.py +468 -460
  313. warp/tests/test_spatial.py +2148 -2148
  314. warp/tests/test_streams.py +486 -473
  315. warp/tests/test_struct.py +710 -675
  316. warp/tests/test_tape.py +173 -148
  317. warp/tests/test_torch.py +743 -743
  318. warp/tests/test_transient_module.py +87 -87
  319. warp/tests/test_types.py +556 -659
  320. warp/tests/test_utils.py +490 -499
  321. warp/tests/test_vec.py +1264 -1268
  322. warp/tests/test_vec_lite.py +73 -73
  323. warp/tests/test_vec_scalar_ops.py +2099 -2099
  324. warp/tests/test_verify_fp.py +94 -94
  325. warp/tests/test_volume.py +737 -736
  326. warp/tests/test_volume_write.py +255 -265
  327. warp/tests/unittest_serial.py +37 -37
  328. warp/tests/unittest_suites.py +363 -359
  329. warp/tests/unittest_utils.py +603 -578
  330. warp/tests/unused_test_misc.py +71 -71
  331. warp/tests/walkthrough_debug.py +85 -85
  332. warp/thirdparty/appdirs.py +598 -598
  333. warp/thirdparty/dlpack.py +143 -143
  334. warp/thirdparty/unittest_parallel.py +566 -561
  335. warp/torch.py +321 -295
  336. warp/types.py +4504 -4450
  337. warp/utils.py +1008 -821
  338. {warp_lang-1.0.2.dist-info → warp_lang-1.1.0.dist-info}/LICENSE.md +126 -126
  339. {warp_lang-1.0.2.dist-info → warp_lang-1.1.0.dist-info}/METADATA +338 -400
  340. warp_lang-1.1.0.dist-info/RECORD +352 -0
  341. warp/examples/assets/cube.usda +0 -42
  342. warp/examples/assets/sphere.usda +0 -56
  343. warp/examples/assets/torus.usda +0 -105
  344. warp_lang-1.0.2.dist-info/RECORD +0 -352
  345. {warp_lang-1.0.2.dist-info → warp_lang-1.1.0.dist-info}/WHEEL +0 -0
  346. {warp_lang-1.0.2.dist-info → warp_lang-1.1.0.dist-info}/top_level.txt +0 -0
@@ -1,598 +1,598 @@
1
- # -*- coding: utf-8 -*-
2
- # Copyright (c) 2005-2010 ActiveState Software Inc.
3
- # Copyright (c) 2013 Eddy Petrișor
4
-
5
- """Utilities for determining application-specific dirs.
6
-
7
- See <https://github.com/ActiveState/appdirs> for details and usage.
8
- """
9
- # Dev Notes:
10
- # - MSDN on where to store app data files:
11
- # http://support.microsoft.com/default.aspx?scid=kb;en-us;310294#XSLTH3194121123120121120120
12
- # - Mac OS X: http://developer.apple.com/documentation/MacOSX/Conceptual/BPFileSystem/index.html
13
- # - XDG spec for Un*x: https://standards.freedesktop.org/basedir-spec/basedir-spec-latest.html
14
-
15
- __version__ = "1.4.4"
16
- __version_info__ = tuple(int(segment) for segment in __version__.split("."))
17
-
18
-
19
- import sys
20
- import os
21
-
22
- PY3 = sys.version_info[0] == 3
23
-
24
- if PY3:
25
- unicode = str
26
-
27
- if sys.platform.startswith("java"):
28
- import platform
29
-
30
- os_name = platform.java_ver()[3][0]
31
- if os_name.startswith("Windows"): # "Windows XP", "Windows 7", etc.
32
- system = "win32"
33
- elif os_name.startswith("Mac"): # "Mac OS X", etc.
34
- system = "darwin"
35
- else: # "Linux", "SunOS", "FreeBSD", etc.
36
- # Setting this to "linux2" is not ideal, but only Windows or Mac
37
- # are actually checked for and the rest of the module expects
38
- # *sys.platform* style strings.
39
- system = "linux2"
40
- else:
41
- system = sys.platform
42
-
43
-
44
- def user_data_dir(appname=None, appauthor=None, version=None, roaming=False):
45
- r"""Return full path to the user-specific data dir for this application.
46
-
47
- "appname" is the name of application.
48
- If None, just the system directory is returned.
49
- "appauthor" (only used on Windows) is the name of the
50
- appauthor or distributing body for this application. Typically
51
- it is the owning company name. This falls back to appname. You may
52
- pass False to disable it.
53
- "version" is an optional version path element to append to the
54
- path. You might want to use this if you want multiple versions
55
- of your app to be able to run independently. If used, this
56
- would typically be "<major>.<minor>".
57
- Only applied when appname is present.
58
- "roaming" (boolean, default False) can be set True to use the Windows
59
- roaming appdata directory. That means that for users on a Windows
60
- network setup for roaming profiles, this user data will be
61
- sync'd on login. See
62
- <http://technet.microsoft.com/en-us/library/cc766489(WS.10).aspx>
63
- for a discussion of issues.
64
-
65
- Typical user data directories are:
66
- Mac OS X: ~/Library/Application Support/<AppName>
67
- Unix: ~/.local/share/<AppName> # or in $XDG_DATA_HOME, if defined
68
- Win XP (not roaming): C:\Documents and Settings\<username>\Application Data\<AppAuthor>\<AppName>
69
- Win XP (roaming): C:\Documents and Settings\<username>\Local Settings\Application Data\<AppAuthor>\<AppName>
70
- Win 7 (not roaming): C:\Users\<username>\AppData\Local\<AppAuthor>\<AppName>
71
- Win 7 (roaming): C:\Users\<username>\AppData\Roaming\<AppAuthor>\<AppName>
72
-
73
- For Unix, we follow the XDG spec and support $XDG_DATA_HOME.
74
- That means, by default "~/.local/share/<AppName>".
75
- """
76
- if system == "win32":
77
- if appauthor is None:
78
- appauthor = appname
79
- const = "CSIDL_APPDATA" if roaming else "CSIDL_LOCAL_APPDATA"
80
- path = os.path.normpath(_get_win_folder(const))
81
- if appname:
82
- if appauthor is not False:
83
- path = os.path.join(path, appauthor, appname)
84
- else:
85
- path = os.path.join(path, appname)
86
- elif system == "darwin":
87
- path = os.path.expanduser("~/Library/Application Support/")
88
- if appname:
89
- path = os.path.join(path, appname)
90
- else:
91
- path = os.getenv("XDG_DATA_HOME", os.path.expanduser("~/.local/share"))
92
- if appname:
93
- path = os.path.join(path, appname)
94
- if appname and version:
95
- path = os.path.join(path, version)
96
- return path
97
-
98
-
99
- def site_data_dir(appname=None, appauthor=None, version=None, multipath=False):
100
- r"""Return full path to the user-shared data dir for this application.
101
-
102
- "appname" is the name of application.
103
- If None, just the system directory is returned.
104
- "appauthor" (only used on Windows) is the name of the
105
- appauthor or distributing body for this application. Typically
106
- it is the owning company name. This falls back to appname. You may
107
- pass False to disable it.
108
- "version" is an optional version path element to append to the
109
- path. You might want to use this if you want multiple versions
110
- of your app to be able to run independently. If used, this
111
- would typically be "<major>.<minor>".
112
- Only applied when appname is present.
113
- "multipath" is an optional parameter only applicable to *nix
114
- which indicates that the entire list of data dirs should be
115
- returned. By default, the first item from XDG_DATA_DIRS is
116
- returned, or '/usr/local/share/<AppName>',
117
- if XDG_DATA_DIRS is not set
118
-
119
- Typical site data directories are:
120
- Mac OS X: /Library/Application Support/<AppName>
121
- Unix: /usr/local/share/<AppName> or /usr/share/<AppName>
122
- Win XP: C:\Documents and Settings\All Users\Application Data\<AppAuthor>\<AppName>
123
- Vista: (Fail! "C:\ProgramData" is a hidden *system* directory on Vista.)
124
- Win 7: C:\ProgramData\<AppAuthor>\<AppName> # Hidden, but writeable on Win 7.
125
-
126
- For Unix, this is using the $XDG_DATA_DIRS[0] default.
127
-
128
- WARNING: Do not use this on Windows. See the Vista-Fail note above for why.
129
- """
130
- if system == "win32":
131
- if appauthor is None:
132
- appauthor = appname
133
- path = os.path.normpath(_get_win_folder("CSIDL_COMMON_APPDATA"))
134
- if appname:
135
- if appauthor is not False:
136
- path = os.path.join(path, appauthor, appname)
137
- else:
138
- path = os.path.join(path, appname)
139
- elif system == "darwin":
140
- path = os.path.expanduser("/Library/Application Support")
141
- if appname:
142
- path = os.path.join(path, appname)
143
- else:
144
- # XDG default for $XDG_DATA_DIRS
145
- # only first, if multipath is False
146
- path = os.getenv("XDG_DATA_DIRS", os.pathsep.join(["/usr/local/share", "/usr/share"]))
147
- pathlist = [os.path.expanduser(x.rstrip(os.sep)) for x in path.split(os.pathsep)]
148
- if appname:
149
- if version:
150
- appname = os.path.join(appname, version)
151
- pathlist = [os.sep.join([x, appname]) for x in pathlist]
152
-
153
- if multipath:
154
- path = os.pathsep.join(pathlist)
155
- else:
156
- path = pathlist[0]
157
- return path
158
-
159
- if appname and version:
160
- path = os.path.join(path, version)
161
- return path
162
-
163
-
164
- def user_config_dir(appname=None, appauthor=None, version=None, roaming=False):
165
- r"""Return full path to the user-specific config dir for this application.
166
-
167
- "appname" is the name of application.
168
- If None, just the system directory is returned.
169
- "appauthor" (only used on Windows) is the name of the
170
- appauthor or distributing body for this application. Typically
171
- it is the owning company name. This falls back to appname. You may
172
- pass False to disable it.
173
- "version" is an optional version path element to append to the
174
- path. You might want to use this if you want multiple versions
175
- of your app to be able to run independently. If used, this
176
- would typically be "<major>.<minor>".
177
- Only applied when appname is present.
178
- "roaming" (boolean, default False) can be set True to use the Windows
179
- roaming appdata directory. That means that for users on a Windows
180
- network setup for roaming profiles, this user data will be
181
- sync'd on login. See
182
- <http://technet.microsoft.com/en-us/library/cc766489(WS.10).aspx>
183
- for a discussion of issues.
184
-
185
- Typical user config directories are:
186
- Mac OS X: ~/Library/Preferences/<AppName>
187
- Unix: ~/.config/<AppName> # or in $XDG_CONFIG_HOME, if defined
188
- Win *: same as user_data_dir
189
-
190
- For Unix, we follow the XDG spec and support $XDG_CONFIG_HOME.
191
- That means, by default "~/.config/<AppName>".
192
- """
193
- if system == "win32":
194
- path = user_data_dir(appname, appauthor, None, roaming)
195
- elif system == "darwin":
196
- path = os.path.expanduser("~/Library/Preferences/")
197
- if appname:
198
- path = os.path.join(path, appname)
199
- else:
200
- path = os.getenv("XDG_CONFIG_HOME", os.path.expanduser("~/.config"))
201
- if appname:
202
- path = os.path.join(path, appname)
203
- if appname and version:
204
- path = os.path.join(path, version)
205
- return path
206
-
207
-
208
- def site_config_dir(appname=None, appauthor=None, version=None, multipath=False):
209
- r"""Return full path to the user-shared data dir for this application.
210
-
211
- "appname" is the name of application.
212
- If None, just the system directory is returned.
213
- "appauthor" (only used on Windows) is the name of the
214
- appauthor or distributing body for this application. Typically
215
- it is the owning company name. This falls back to appname. You may
216
- pass False to disable it.
217
- "version" is an optional version path element to append to the
218
- path. You might want to use this if you want multiple versions
219
- of your app to be able to run independently. If used, this
220
- would typically be "<major>.<minor>".
221
- Only applied when appname is present.
222
- "multipath" is an optional parameter only applicable to *nix
223
- which indicates that the entire list of config dirs should be
224
- returned. By default, the first item from XDG_CONFIG_DIRS is
225
- returned, or '/etc/xdg/<AppName>', if XDG_CONFIG_DIRS is not set
226
-
227
- Typical site config directories are:
228
- Mac OS X: same as site_data_dir
229
- Unix: /etc/xdg/<AppName> or $XDG_CONFIG_DIRS[i]/<AppName> for each value in
230
- $XDG_CONFIG_DIRS
231
- Win *: same as site_data_dir
232
- Vista: (Fail! "C:\ProgramData" is a hidden *system* directory on Vista.)
233
-
234
- For Unix, this is using the $XDG_CONFIG_DIRS[0] default, if multipath=False
235
-
236
- WARNING: Do not use this on Windows. See the Vista-Fail note above for why.
237
- """
238
- if system == "win32":
239
- path = site_data_dir(appname, appauthor)
240
- if appname and version:
241
- path = os.path.join(path, version)
242
- elif system == "darwin":
243
- path = os.path.expanduser("/Library/Preferences")
244
- if appname:
245
- path = os.path.join(path, appname)
246
- else:
247
- # XDG default for $XDG_CONFIG_DIRS
248
- # only first, if multipath is False
249
- path = os.getenv("XDG_CONFIG_DIRS", "/etc/xdg")
250
- pathlist = [os.path.expanduser(x.rstrip(os.sep)) for x in path.split(os.pathsep)]
251
- if appname:
252
- if version:
253
- appname = os.path.join(appname, version)
254
- pathlist = [os.sep.join([x, appname]) for x in pathlist]
255
-
256
- if multipath:
257
- path = os.pathsep.join(pathlist)
258
- else:
259
- path = pathlist[0]
260
- return path
261
-
262
-
263
- def user_cache_dir(appname=None, appauthor=None, version=None, opinion=True):
264
- r"""Return full path to the user-specific cache dir for this application.
265
-
266
- "appname" is the name of application.
267
- If None, just the system directory is returned.
268
- "appauthor" (only used on Windows) is the name of the
269
- appauthor or distributing body for this application. Typically
270
- it is the owning company name. This falls back to appname. You may
271
- pass False to disable it.
272
- "version" is an optional version path element to append to the
273
- path. You might want to use this if you want multiple versions
274
- of your app to be able to run independently. If used, this
275
- would typically be "<major>.<minor>".
276
- Only applied when appname is present.
277
- "opinion" (boolean) can be False to disable the appending of
278
- "Cache" to the base app data dir for Windows. See
279
- discussion below.
280
-
281
- Typical user cache directories are:
282
- Mac OS X: ~/Library/Caches/<AppName>
283
- Unix: ~/.cache/<AppName> (XDG default)
284
- Win XP: C:\Documents and Settings\<username>\Local Settings\Application Data\<AppAuthor>\<AppName>\Cache
285
- Vista: C:\Users\<username>\AppData\Local\<AppAuthor>\<AppName>\Cache
286
-
287
- On Windows the only suggestion in the MSDN docs is that local settings go in
288
- the `CSIDL_LOCAL_APPDATA` directory. This is identical to the non-roaming
289
- app data dir (the default returned by `user_data_dir` above). Apps typically
290
- put cache data somewhere *under* the given dir here. Some examples:
291
- ...\Mozilla\Firefox\Profiles\<ProfileName>\Cache
292
- ...\Acme\SuperApp\Cache\1.0
293
- OPINION: This function appends "Cache" to the `CSIDL_LOCAL_APPDATA` value.
294
- This can be disabled with the `opinion=False` option.
295
- """
296
- if system == "win32":
297
- if appauthor is None:
298
- appauthor = appname
299
- path = os.path.normpath(_get_win_folder("CSIDL_LOCAL_APPDATA"))
300
- if appname:
301
- if appauthor is not False:
302
- path = os.path.join(path, appauthor, appname)
303
- else:
304
- path = os.path.join(path, appname)
305
- if opinion:
306
- path = os.path.join(path, "Cache")
307
- elif system == "darwin":
308
- path = os.path.expanduser("~/Library/Caches")
309
- if appname:
310
- path = os.path.join(path, appname)
311
- else:
312
- path = os.getenv("XDG_CACHE_HOME", os.path.expanduser("~/.cache"))
313
- if appname:
314
- path = os.path.join(path, appname)
315
- if appname and version:
316
- path = os.path.join(path, version)
317
- return path
318
-
319
-
320
- def user_state_dir(appname=None, appauthor=None, version=None, roaming=False):
321
- r"""Return full path to the user-specific state dir for this application.
322
-
323
- "appname" is the name of application.
324
- If None, just the system directory is returned.
325
- "appauthor" (only used on Windows) is the name of the
326
- appauthor or distributing body for this application. Typically
327
- it is the owning company name. This falls back to appname. You may
328
- pass False to disable it.
329
- "version" is an optional version path element to append to the
330
- path. You might want to use this if you want multiple versions
331
- of your app to be able to run independently. If used, this
332
- would typically be "<major>.<minor>".
333
- Only applied when appname is present.
334
- "roaming" (boolean, default False) can be set True to use the Windows
335
- roaming appdata directory. That means that for users on a Windows
336
- network setup for roaming profiles, this user data will be
337
- sync'd on login. See
338
- <http://technet.microsoft.com/en-us/library/cc766489(WS.10).aspx>
339
- for a discussion of issues.
340
-
341
- Typical user state directories are:
342
- Mac OS X: same as user_data_dir
343
- Unix: ~/.local/state/<AppName> # or in $XDG_STATE_HOME, if defined
344
- Win *: same as user_data_dir
345
-
346
- For Unix, we follow this Debian proposal <https://wiki.debian.org/XDGBaseDirectorySpecification#state>
347
- to extend the XDG spec and support $XDG_STATE_HOME.
348
-
349
- That means, by default "~/.local/state/<AppName>".
350
- """
351
- if system in ["win32", "darwin"]:
352
- path = user_data_dir(appname, appauthor, None, roaming)
353
- else:
354
- path = os.getenv("XDG_STATE_HOME", os.path.expanduser("~/.local/state"))
355
- if appname:
356
- path = os.path.join(path, appname)
357
- if appname and version:
358
- path = os.path.join(path, version)
359
- return path
360
-
361
-
362
- def user_log_dir(appname=None, appauthor=None, version=None, opinion=True):
363
- r"""Return full path to the user-specific log dir for this application.
364
-
365
- "appname" is the name of application.
366
- If None, just the system directory is returned.
367
- "appauthor" (only used on Windows) is the name of the
368
- appauthor or distributing body for this application. Typically
369
- it is the owning company name. This falls back to appname. You may
370
- pass False to disable it.
371
- "version" is an optional version path element to append to the
372
- path. You might want to use this if you want multiple versions
373
- of your app to be able to run independently. If used, this
374
- would typically be "<major>.<minor>".
375
- Only applied when appname is present.
376
- "opinion" (boolean) can be False to disable the appending of
377
- "Logs" to the base app data dir for Windows, and "log" to the
378
- base cache dir for Unix. See discussion below.
379
-
380
- Typical user log directories are:
381
- Mac OS X: ~/Library/Logs/<AppName>
382
- Unix: ~/.cache/<AppName>/log # or under $XDG_CACHE_HOME if defined
383
- Win XP: C:\Documents and Settings\<username>\Local Settings\Application Data\<AppAuthor>\<AppName>\Logs
384
- Vista: C:\Users\<username>\AppData\Local\<AppAuthor>\<AppName>\Logs
385
-
386
- On Windows the only suggestion in the MSDN docs is that local settings
387
- go in the `CSIDL_LOCAL_APPDATA` directory. (Note: I'm interested in
388
- examples of what some windows apps use for a logs dir.)
389
-
390
- OPINION: This function appends "Logs" to the `CSIDL_LOCAL_APPDATA`
391
- value for Windows and appends "log" to the user cache dir for Unix.
392
- This can be disabled with the `opinion=False` option.
393
- """
394
- if system == "darwin":
395
- path = os.path.join(os.path.expanduser("~/Library/Logs"), appname)
396
- elif system == "win32":
397
- path = user_data_dir(appname, appauthor, version)
398
- version = False
399
- if opinion:
400
- path = os.path.join(path, "Logs")
401
- else:
402
- path = user_cache_dir(appname, appauthor, version)
403
- version = False
404
- if opinion:
405
- path = os.path.join(path, "log")
406
- if appname and version:
407
- path = os.path.join(path, version)
408
- return path
409
-
410
-
411
- class AppDirs(object):
412
- """Convenience wrapper for getting application dirs."""
413
-
414
- def __init__(self, appname=None, appauthor=None, version=None, roaming=False, multipath=False):
415
- self.appname = appname
416
- self.appauthor = appauthor
417
- self.version = version
418
- self.roaming = roaming
419
- self.multipath = multipath
420
-
421
- @property
422
- def user_data_dir(self):
423
- return user_data_dir(self.appname, self.appauthor, version=self.version, roaming=self.roaming)
424
-
425
- @property
426
- def site_data_dir(self):
427
- return site_data_dir(self.appname, self.appauthor, version=self.version, multipath=self.multipath)
428
-
429
- @property
430
- def user_config_dir(self):
431
- return user_config_dir(self.appname, self.appauthor, version=self.version, roaming=self.roaming)
432
-
433
- @property
434
- def site_config_dir(self):
435
- return site_config_dir(self.appname, self.appauthor, version=self.version, multipath=self.multipath)
436
-
437
- @property
438
- def user_cache_dir(self):
439
- return user_cache_dir(self.appname, self.appauthor, version=self.version)
440
-
441
- @property
442
- def user_state_dir(self):
443
- return user_state_dir(self.appname, self.appauthor, version=self.version)
444
-
445
- @property
446
- def user_log_dir(self):
447
- return user_log_dir(self.appname, self.appauthor, version=self.version)
448
-
449
-
450
- # ---- internal support stuff
451
-
452
-
453
- def _get_win_folder_from_registry(csidl_name):
454
- """This is a fallback technique at best. I'm not sure if using the
455
- registry for this guarantees us the correct answer for all CSIDL_*
456
- names.
457
- """
458
- if PY3:
459
- import winreg as _winreg
460
- else:
461
- import _winreg
462
-
463
- shell_folder_name = {
464
- "CSIDL_APPDATA": "AppData",
465
- "CSIDL_COMMON_APPDATA": "Common AppData",
466
- "CSIDL_LOCAL_APPDATA": "Local AppData",
467
- }[csidl_name]
468
-
469
- key = _winreg.OpenKey(
470
- _winreg.HKEY_CURRENT_USER, r"Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders"
471
- )
472
- dir, type = _winreg.QueryValueEx(key, shell_folder_name)
473
- return dir
474
-
475
-
476
- def _get_win_folder_with_ctypes(csidl_name):
477
- import ctypes
478
-
479
- csidl_const = {
480
- "CSIDL_APPDATA": 26,
481
- "CSIDL_COMMON_APPDATA": 35,
482
- "CSIDL_LOCAL_APPDATA": 28,
483
- }[csidl_name]
484
-
485
- buf = ctypes.create_unicode_buffer(1024)
486
- ctypes.windll.shell32.SHGetFolderPathW(None, csidl_const, None, 0, buf)
487
-
488
- # Downgrade to short path name if have highbit chars. See
489
- # <http://bugs.activestate.com/show_bug.cgi?id=85099>.
490
- has_high_char = False
491
- for c in buf:
492
- if ord(c) > 255:
493
- has_high_char = True
494
- break
495
- if has_high_char:
496
- buf2 = ctypes.create_unicode_buffer(1024)
497
- if ctypes.windll.kernel32.GetShortPathNameW(buf.value, buf2, 1024):
498
- buf = buf2
499
-
500
- return buf.value
501
-
502
-
503
- def _get_win_folder_with_jna(csidl_name):
504
- import array
505
- from com.sun import jna
506
- from com.sun.jna.platform import win32
507
-
508
- buf_size = win32.WinDef.MAX_PATH * 2
509
- buf = array.zeros("c", buf_size)
510
- shell = win32.Shell32.INSTANCE
511
- shell.SHGetFolderPath(None, getattr(win32.ShlObj, csidl_name), None, win32.ShlObj.SHGFP_TYPE_CURRENT, buf)
512
- dir = jna.Native.toString(buf.tostring()).rstrip("\0")
513
-
514
- # Downgrade to short path name if have highbit chars. See
515
- # <http://bugs.activestate.com/show_bug.cgi?id=85099>.
516
- has_high_char = False
517
- for c in dir:
518
- if ord(c) > 255:
519
- has_high_char = True
520
- break
521
- if has_high_char:
522
- buf = array.zeros("c", buf_size)
523
- kernel = win32.Kernel32.INSTANCE
524
- if kernel.GetShortPathName(dir, buf, buf_size):
525
- dir = jna.Native.toString(buf.tostring()).rstrip("\0")
526
-
527
- return dir
528
-
529
-
530
- def _get_win_folder_from_environ(csidl_name):
531
- env_var_name = {
532
- "CSIDL_APPDATA": "APPDATA",
533
- "CSIDL_COMMON_APPDATA": "ALLUSERSPROFILE",
534
- "CSIDL_LOCAL_APPDATA": "LOCALAPPDATA",
535
- }[csidl_name]
536
-
537
- return os.environ[env_var_name]
538
-
539
-
540
- if system == "win32":
541
- try:
542
- from ctypes import windll
543
- except ImportError:
544
- try:
545
- import com.sun.jna
546
- except ImportError:
547
- try:
548
- if PY3:
549
- import winreg as _winreg
550
- else:
551
- import _winreg
552
- except ImportError:
553
- _get_win_folder = _get_win_folder_from_environ
554
- else:
555
- _get_win_folder = _get_win_folder_from_registry
556
- else:
557
- _get_win_folder = _get_win_folder_with_jna
558
- else:
559
- _get_win_folder = _get_win_folder_with_ctypes
560
-
561
-
562
- # ---- self test code
563
-
564
- if __name__ == "__main__":
565
- appname = "MyApp"
566
- appauthor = "MyCompany"
567
-
568
- props = (
569
- "user_data_dir",
570
- "user_config_dir",
571
- "user_cache_dir",
572
- "user_state_dir",
573
- "user_log_dir",
574
- "site_data_dir",
575
- "site_config_dir",
576
- )
577
-
578
- print("-- app dirs %s --" % __version__)
579
-
580
- print("-- app dirs (with optional 'version')")
581
- dirs = AppDirs(appname, appauthor, version="1.0")
582
- for prop in props:
583
- print("%s: %s" % (prop, getattr(dirs, prop)))
584
-
585
- print("\n-- app dirs (without optional 'version')")
586
- dirs = AppDirs(appname, appauthor)
587
- for prop in props:
588
- print("%s: %s" % (prop, getattr(dirs, prop)))
589
-
590
- print("\n-- app dirs (without optional 'appauthor')")
591
- dirs = AppDirs(appname)
592
- for prop in props:
593
- print("%s: %s" % (prop, getattr(dirs, prop)))
594
-
595
- print("\n-- app dirs (with disabled 'appauthor')")
596
- dirs = AppDirs(appname, appauthor=False)
597
- for prop in props:
598
- print("%s: %s" % (prop, getattr(dirs, prop)))
1
+ # -*- coding: utf-8 -*-
2
+ # Copyright (c) 2005-2010 ActiveState Software Inc.
3
+ # Copyright (c) 2013 Eddy Petrișor
4
+
5
+ """Utilities for determining application-specific dirs.
6
+
7
+ See <https://github.com/ActiveState/appdirs> for details and usage.
8
+ """
9
+ # Dev Notes:
10
+ # - MSDN on where to store app data files:
11
+ # http://support.microsoft.com/default.aspx?scid=kb;en-us;310294#XSLTH3194121123120121120120
12
+ # - Mac OS X: http://developer.apple.com/documentation/MacOSX/Conceptual/BPFileSystem/index.html
13
+ # - XDG spec for Un*x: https://standards.freedesktop.org/basedir-spec/basedir-spec-latest.html
14
+
15
+ __version__ = "1.4.4"
16
+ __version_info__ = tuple(int(segment) for segment in __version__.split("."))
17
+
18
+
19
+ import sys
20
+ import os
21
+
22
+ PY3 = sys.version_info[0] == 3
23
+
24
+ if PY3:
25
+ unicode = str
26
+
27
+ if sys.platform.startswith("java"):
28
+ import platform
29
+
30
+ os_name = platform.java_ver()[3][0]
31
+ if os_name.startswith("Windows"): # "Windows XP", "Windows 7", etc.
32
+ system = "win32"
33
+ elif os_name.startswith("Mac"): # "Mac OS X", etc.
34
+ system = "darwin"
35
+ else: # "Linux", "SunOS", "FreeBSD", etc.
36
+ # Setting this to "linux2" is not ideal, but only Windows or Mac
37
+ # are actually checked for and the rest of the module expects
38
+ # *sys.platform* style strings.
39
+ system = "linux2"
40
+ else:
41
+ system = sys.platform
42
+
43
+
44
+ def user_data_dir(appname=None, appauthor=None, version=None, roaming=False):
45
+ r"""Return full path to the user-specific data dir for this application.
46
+
47
+ "appname" is the name of application.
48
+ If None, just the system directory is returned.
49
+ "appauthor" (only used on Windows) is the name of the
50
+ appauthor or distributing body for this application. Typically
51
+ it is the owning company name. This falls back to appname. You may
52
+ pass False to disable it.
53
+ "version" is an optional version path element to append to the
54
+ path. You might want to use this if you want multiple versions
55
+ of your app to be able to run independently. If used, this
56
+ would typically be "<major>.<minor>".
57
+ Only applied when appname is present.
58
+ "roaming" (boolean, default False) can be set True to use the Windows
59
+ roaming appdata directory. That means that for users on a Windows
60
+ network setup for roaming profiles, this user data will be
61
+ sync'd on login. See
62
+ <http://technet.microsoft.com/en-us/library/cc766489(WS.10).aspx>
63
+ for a discussion of issues.
64
+
65
+ Typical user data directories are:
66
+ Mac OS X: ~/Library/Application Support/<AppName>
67
+ Unix: ~/.local/share/<AppName> # or in $XDG_DATA_HOME, if defined
68
+ Win XP (not roaming): C:\Documents and Settings\<username>\Application Data\<AppAuthor>\<AppName>
69
+ Win XP (roaming): C:\Documents and Settings\<username>\Local Settings\Application Data\<AppAuthor>\<AppName>
70
+ Win 7 (not roaming): C:\Users\<username>\AppData\Local\<AppAuthor>\<AppName>
71
+ Win 7 (roaming): C:\Users\<username>\AppData\Roaming\<AppAuthor>\<AppName>
72
+
73
+ For Unix, we follow the XDG spec and support $XDG_DATA_HOME.
74
+ That means, by default "~/.local/share/<AppName>".
75
+ """
76
+ if system == "win32":
77
+ if appauthor is None:
78
+ appauthor = appname
79
+ const = "CSIDL_APPDATA" if roaming else "CSIDL_LOCAL_APPDATA"
80
+ path = os.path.normpath(_get_win_folder(const))
81
+ if appname:
82
+ if appauthor is not False:
83
+ path = os.path.join(path, appauthor, appname)
84
+ else:
85
+ path = os.path.join(path, appname)
86
+ elif system == "darwin":
87
+ path = os.path.expanduser("~/Library/Application Support/")
88
+ if appname:
89
+ path = os.path.join(path, appname)
90
+ else:
91
+ path = os.getenv("XDG_DATA_HOME", os.path.expanduser("~/.local/share"))
92
+ if appname:
93
+ path = os.path.join(path, appname)
94
+ if appname and version:
95
+ path = os.path.join(path, version)
96
+ return path
97
+
98
+
99
+ def site_data_dir(appname=None, appauthor=None, version=None, multipath=False):
100
+ r"""Return full path to the user-shared data dir for this application.
101
+
102
+ "appname" is the name of application.
103
+ If None, just the system directory is returned.
104
+ "appauthor" (only used on Windows) is the name of the
105
+ appauthor or distributing body for this application. Typically
106
+ it is the owning company name. This falls back to appname. You may
107
+ pass False to disable it.
108
+ "version" is an optional version path element to append to the
109
+ path. You might want to use this if you want multiple versions
110
+ of your app to be able to run independently. If used, this
111
+ would typically be "<major>.<minor>".
112
+ Only applied when appname is present.
113
+ "multipath" is an optional parameter only applicable to *nix
114
+ which indicates that the entire list of data dirs should be
115
+ returned. By default, the first item from XDG_DATA_DIRS is
116
+ returned, or '/usr/local/share/<AppName>',
117
+ if XDG_DATA_DIRS is not set
118
+
119
+ Typical site data directories are:
120
+ Mac OS X: /Library/Application Support/<AppName>
121
+ Unix: /usr/local/share/<AppName> or /usr/share/<AppName>
122
+ Win XP: C:\Documents and Settings\All Users\Application Data\<AppAuthor>\<AppName>
123
+ Vista: (Fail! "C:\ProgramData" is a hidden *system* directory on Vista.)
124
+ Win 7: C:\ProgramData\<AppAuthor>\<AppName> # Hidden, but writeable on Win 7.
125
+
126
+ For Unix, this is using the $XDG_DATA_DIRS[0] default.
127
+
128
+ WARNING: Do not use this on Windows. See the Vista-Fail note above for why.
129
+ """
130
+ if system == "win32":
131
+ if appauthor is None:
132
+ appauthor = appname
133
+ path = os.path.normpath(_get_win_folder("CSIDL_COMMON_APPDATA"))
134
+ if appname:
135
+ if appauthor is not False:
136
+ path = os.path.join(path, appauthor, appname)
137
+ else:
138
+ path = os.path.join(path, appname)
139
+ elif system == "darwin":
140
+ path = os.path.expanduser("/Library/Application Support")
141
+ if appname:
142
+ path = os.path.join(path, appname)
143
+ else:
144
+ # XDG default for $XDG_DATA_DIRS
145
+ # only first, if multipath is False
146
+ path = os.getenv("XDG_DATA_DIRS", os.pathsep.join(["/usr/local/share", "/usr/share"]))
147
+ pathlist = [os.path.expanduser(x.rstrip(os.sep)) for x in path.split(os.pathsep)]
148
+ if appname:
149
+ if version:
150
+ appname = os.path.join(appname, version)
151
+ pathlist = [os.sep.join([x, appname]) for x in pathlist]
152
+
153
+ if multipath:
154
+ path = os.pathsep.join(pathlist)
155
+ else:
156
+ path = pathlist[0]
157
+ return path
158
+
159
+ if appname and version:
160
+ path = os.path.join(path, version)
161
+ return path
162
+
163
+
164
+ def user_config_dir(appname=None, appauthor=None, version=None, roaming=False):
165
+ r"""Return full path to the user-specific config dir for this application.
166
+
167
+ "appname" is the name of application.
168
+ If None, just the system directory is returned.
169
+ "appauthor" (only used on Windows) is the name of the
170
+ appauthor or distributing body for this application. Typically
171
+ it is the owning company name. This falls back to appname. You may
172
+ pass False to disable it.
173
+ "version" is an optional version path element to append to the
174
+ path. You might want to use this if you want multiple versions
175
+ of your app to be able to run independently. If used, this
176
+ would typically be "<major>.<minor>".
177
+ Only applied when appname is present.
178
+ "roaming" (boolean, default False) can be set True to use the Windows
179
+ roaming appdata directory. That means that for users on a Windows
180
+ network setup for roaming profiles, this user data will be
181
+ sync'd on login. See
182
+ <http://technet.microsoft.com/en-us/library/cc766489(WS.10).aspx>
183
+ for a discussion of issues.
184
+
185
+ Typical user config directories are:
186
+ Mac OS X: ~/Library/Preferences/<AppName>
187
+ Unix: ~/.config/<AppName> # or in $XDG_CONFIG_HOME, if defined
188
+ Win *: same as user_data_dir
189
+
190
+ For Unix, we follow the XDG spec and support $XDG_CONFIG_HOME.
191
+ That means, by default "~/.config/<AppName>".
192
+ """
193
+ if system == "win32":
194
+ path = user_data_dir(appname, appauthor, None, roaming)
195
+ elif system == "darwin":
196
+ path = os.path.expanduser("~/Library/Preferences/")
197
+ if appname:
198
+ path = os.path.join(path, appname)
199
+ else:
200
+ path = os.getenv("XDG_CONFIG_HOME", os.path.expanduser("~/.config"))
201
+ if appname:
202
+ path = os.path.join(path, appname)
203
+ if appname and version:
204
+ path = os.path.join(path, version)
205
+ return path
206
+
207
+
208
+ def site_config_dir(appname=None, appauthor=None, version=None, multipath=False):
209
+ r"""Return full path to the user-shared data dir for this application.
210
+
211
+ "appname" is the name of application.
212
+ If None, just the system directory is returned.
213
+ "appauthor" (only used on Windows) is the name of the
214
+ appauthor or distributing body for this application. Typically
215
+ it is the owning company name. This falls back to appname. You may
216
+ pass False to disable it.
217
+ "version" is an optional version path element to append to the
218
+ path. You might want to use this if you want multiple versions
219
+ of your app to be able to run independently. If used, this
220
+ would typically be "<major>.<minor>".
221
+ Only applied when appname is present.
222
+ "multipath" is an optional parameter only applicable to *nix
223
+ which indicates that the entire list of config dirs should be
224
+ returned. By default, the first item from XDG_CONFIG_DIRS is
225
+ returned, or '/etc/xdg/<AppName>', if XDG_CONFIG_DIRS is not set
226
+
227
+ Typical site config directories are:
228
+ Mac OS X: same as site_data_dir
229
+ Unix: /etc/xdg/<AppName> or $XDG_CONFIG_DIRS[i]/<AppName> for each value in
230
+ $XDG_CONFIG_DIRS
231
+ Win *: same as site_data_dir
232
+ Vista: (Fail! "C:\ProgramData" is a hidden *system* directory on Vista.)
233
+
234
+ For Unix, this is using the $XDG_CONFIG_DIRS[0] default, if multipath=False
235
+
236
+ WARNING: Do not use this on Windows. See the Vista-Fail note above for why.
237
+ """
238
+ if system == "win32":
239
+ path = site_data_dir(appname, appauthor)
240
+ if appname and version:
241
+ path = os.path.join(path, version)
242
+ elif system == "darwin":
243
+ path = os.path.expanduser("/Library/Preferences")
244
+ if appname:
245
+ path = os.path.join(path, appname)
246
+ else:
247
+ # XDG default for $XDG_CONFIG_DIRS
248
+ # only first, if multipath is False
249
+ path = os.getenv("XDG_CONFIG_DIRS", "/etc/xdg")
250
+ pathlist = [os.path.expanduser(x.rstrip(os.sep)) for x in path.split(os.pathsep)]
251
+ if appname:
252
+ if version:
253
+ appname = os.path.join(appname, version)
254
+ pathlist = [os.sep.join([x, appname]) for x in pathlist]
255
+
256
+ if multipath:
257
+ path = os.pathsep.join(pathlist)
258
+ else:
259
+ path = pathlist[0]
260
+ return path
261
+
262
+
263
+ def user_cache_dir(appname=None, appauthor=None, version=None, opinion=True):
264
+ r"""Return full path to the user-specific cache dir for this application.
265
+
266
+ "appname" is the name of application.
267
+ If None, just the system directory is returned.
268
+ "appauthor" (only used on Windows) is the name of the
269
+ appauthor or distributing body for this application. Typically
270
+ it is the owning company name. This falls back to appname. You may
271
+ pass False to disable it.
272
+ "version" is an optional version path element to append to the
273
+ path. You might want to use this if you want multiple versions
274
+ of your app to be able to run independently. If used, this
275
+ would typically be "<major>.<minor>".
276
+ Only applied when appname is present.
277
+ "opinion" (boolean) can be False to disable the appending of
278
+ "Cache" to the base app data dir for Windows. See
279
+ discussion below.
280
+
281
+ Typical user cache directories are:
282
+ Mac OS X: ~/Library/Caches/<AppName>
283
+ Unix: ~/.cache/<AppName> (XDG default)
284
+ Win XP: C:\Documents and Settings\<username>\Local Settings\Application Data\<AppAuthor>\<AppName>\Cache
285
+ Vista: C:\Users\<username>\AppData\Local\<AppAuthor>\<AppName>\Cache
286
+
287
+ On Windows the only suggestion in the MSDN docs is that local settings go in
288
+ the `CSIDL_LOCAL_APPDATA` directory. This is identical to the non-roaming
289
+ app data dir (the default returned by `user_data_dir` above). Apps typically
290
+ put cache data somewhere *under* the given dir here. Some examples:
291
+ ...\Mozilla\Firefox\Profiles\<ProfileName>\Cache
292
+ ...\Acme\SuperApp\Cache\1.0
293
+ OPINION: This function appends "Cache" to the `CSIDL_LOCAL_APPDATA` value.
294
+ This can be disabled with the `opinion=False` option.
295
+ """
296
+ if system == "win32":
297
+ if appauthor is None:
298
+ appauthor = appname
299
+ path = os.path.normpath(_get_win_folder("CSIDL_LOCAL_APPDATA"))
300
+ if appname:
301
+ if appauthor is not False:
302
+ path = os.path.join(path, appauthor, appname)
303
+ else:
304
+ path = os.path.join(path, appname)
305
+ if opinion:
306
+ path = os.path.join(path, "Cache")
307
+ elif system == "darwin":
308
+ path = os.path.expanduser("~/Library/Caches")
309
+ if appname:
310
+ path = os.path.join(path, appname)
311
+ else:
312
+ path = os.getenv("XDG_CACHE_HOME", os.path.expanduser("~/.cache"))
313
+ if appname:
314
+ path = os.path.join(path, appname)
315
+ if appname and version:
316
+ path = os.path.join(path, version)
317
+ return path
318
+
319
+
320
+ def user_state_dir(appname=None, appauthor=None, version=None, roaming=False):
321
+ r"""Return full path to the user-specific state dir for this application.
322
+
323
+ "appname" is the name of application.
324
+ If None, just the system directory is returned.
325
+ "appauthor" (only used on Windows) is the name of the
326
+ appauthor or distributing body for this application. Typically
327
+ it is the owning company name. This falls back to appname. You may
328
+ pass False to disable it.
329
+ "version" is an optional version path element to append to the
330
+ path. You might want to use this if you want multiple versions
331
+ of your app to be able to run independently. If used, this
332
+ would typically be "<major>.<minor>".
333
+ Only applied when appname is present.
334
+ "roaming" (boolean, default False) can be set True to use the Windows
335
+ roaming appdata directory. That means that for users on a Windows
336
+ network setup for roaming profiles, this user data will be
337
+ sync'd on login. See
338
+ <http://technet.microsoft.com/en-us/library/cc766489(WS.10).aspx>
339
+ for a discussion of issues.
340
+
341
+ Typical user state directories are:
342
+ Mac OS X: same as user_data_dir
343
+ Unix: ~/.local/state/<AppName> # or in $XDG_STATE_HOME, if defined
344
+ Win *: same as user_data_dir
345
+
346
+ For Unix, we follow this Debian proposal <https://wiki.debian.org/XDGBaseDirectorySpecification#state>
347
+ to extend the XDG spec and support $XDG_STATE_HOME.
348
+
349
+ That means, by default "~/.local/state/<AppName>".
350
+ """
351
+ if system in ["win32", "darwin"]:
352
+ path = user_data_dir(appname, appauthor, None, roaming)
353
+ else:
354
+ path = os.getenv("XDG_STATE_HOME", os.path.expanduser("~/.local/state"))
355
+ if appname:
356
+ path = os.path.join(path, appname)
357
+ if appname and version:
358
+ path = os.path.join(path, version)
359
+ return path
360
+
361
+
362
+ def user_log_dir(appname=None, appauthor=None, version=None, opinion=True):
363
+ r"""Return full path to the user-specific log dir for this application.
364
+
365
+ "appname" is the name of application.
366
+ If None, just the system directory is returned.
367
+ "appauthor" (only used on Windows) is the name of the
368
+ appauthor or distributing body for this application. Typically
369
+ it is the owning company name. This falls back to appname. You may
370
+ pass False to disable it.
371
+ "version" is an optional version path element to append to the
372
+ path. You might want to use this if you want multiple versions
373
+ of your app to be able to run independently. If used, this
374
+ would typically be "<major>.<minor>".
375
+ Only applied when appname is present.
376
+ "opinion" (boolean) can be False to disable the appending of
377
+ "Logs" to the base app data dir for Windows, and "log" to the
378
+ base cache dir for Unix. See discussion below.
379
+
380
+ Typical user log directories are:
381
+ Mac OS X: ~/Library/Logs/<AppName>
382
+ Unix: ~/.cache/<AppName>/log # or under $XDG_CACHE_HOME if defined
383
+ Win XP: C:\Documents and Settings\<username>\Local Settings\Application Data\<AppAuthor>\<AppName>\Logs
384
+ Vista: C:\Users\<username>\AppData\Local\<AppAuthor>\<AppName>\Logs
385
+
386
+ On Windows the only suggestion in the MSDN docs is that local settings
387
+ go in the `CSIDL_LOCAL_APPDATA` directory. (Note: I'm interested in
388
+ examples of what some windows apps use for a logs dir.)
389
+
390
+ OPINION: This function appends "Logs" to the `CSIDL_LOCAL_APPDATA`
391
+ value for Windows and appends "log" to the user cache dir for Unix.
392
+ This can be disabled with the `opinion=False` option.
393
+ """
394
+ if system == "darwin":
395
+ path = os.path.join(os.path.expanduser("~/Library/Logs"), appname)
396
+ elif system == "win32":
397
+ path = user_data_dir(appname, appauthor, version)
398
+ version = False
399
+ if opinion:
400
+ path = os.path.join(path, "Logs")
401
+ else:
402
+ path = user_cache_dir(appname, appauthor, version)
403
+ version = False
404
+ if opinion:
405
+ path = os.path.join(path, "log")
406
+ if appname and version:
407
+ path = os.path.join(path, version)
408
+ return path
409
+
410
+
411
+ class AppDirs(object):
412
+ """Convenience wrapper for getting application dirs."""
413
+
414
+ def __init__(self, appname=None, appauthor=None, version=None, roaming=False, multipath=False):
415
+ self.appname = appname
416
+ self.appauthor = appauthor
417
+ self.version = version
418
+ self.roaming = roaming
419
+ self.multipath = multipath
420
+
421
+ @property
422
+ def user_data_dir(self):
423
+ return user_data_dir(self.appname, self.appauthor, version=self.version, roaming=self.roaming)
424
+
425
+ @property
426
+ def site_data_dir(self):
427
+ return site_data_dir(self.appname, self.appauthor, version=self.version, multipath=self.multipath)
428
+
429
+ @property
430
+ def user_config_dir(self):
431
+ return user_config_dir(self.appname, self.appauthor, version=self.version, roaming=self.roaming)
432
+
433
+ @property
434
+ def site_config_dir(self):
435
+ return site_config_dir(self.appname, self.appauthor, version=self.version, multipath=self.multipath)
436
+
437
+ @property
438
+ def user_cache_dir(self):
439
+ return user_cache_dir(self.appname, self.appauthor, version=self.version)
440
+
441
+ @property
442
+ def user_state_dir(self):
443
+ return user_state_dir(self.appname, self.appauthor, version=self.version)
444
+
445
+ @property
446
+ def user_log_dir(self):
447
+ return user_log_dir(self.appname, self.appauthor, version=self.version)
448
+
449
+
450
+ # ---- internal support stuff
451
+
452
+
453
+ def _get_win_folder_from_registry(csidl_name):
454
+ """This is a fallback technique at best. I'm not sure if using the
455
+ registry for this guarantees us the correct answer for all CSIDL_*
456
+ names.
457
+ """
458
+ if PY3:
459
+ import winreg as _winreg
460
+ else:
461
+ import _winreg
462
+
463
+ shell_folder_name = {
464
+ "CSIDL_APPDATA": "AppData",
465
+ "CSIDL_COMMON_APPDATA": "Common AppData",
466
+ "CSIDL_LOCAL_APPDATA": "Local AppData",
467
+ }[csidl_name]
468
+
469
+ key = _winreg.OpenKey(
470
+ _winreg.HKEY_CURRENT_USER, r"Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders"
471
+ )
472
+ dir, type = _winreg.QueryValueEx(key, shell_folder_name)
473
+ return dir
474
+
475
+
476
+ def _get_win_folder_with_ctypes(csidl_name):
477
+ import ctypes
478
+
479
+ csidl_const = {
480
+ "CSIDL_APPDATA": 26,
481
+ "CSIDL_COMMON_APPDATA": 35,
482
+ "CSIDL_LOCAL_APPDATA": 28,
483
+ }[csidl_name]
484
+
485
+ buf = ctypes.create_unicode_buffer(1024)
486
+ ctypes.windll.shell32.SHGetFolderPathW(None, csidl_const, None, 0, buf)
487
+
488
+ # Downgrade to short path name if have highbit chars. See
489
+ # <http://bugs.activestate.com/show_bug.cgi?id=85099>.
490
+ has_high_char = False
491
+ for c in buf:
492
+ if ord(c) > 255:
493
+ has_high_char = True
494
+ break
495
+ if has_high_char:
496
+ buf2 = ctypes.create_unicode_buffer(1024)
497
+ if ctypes.windll.kernel32.GetShortPathNameW(buf.value, buf2, 1024):
498
+ buf = buf2
499
+
500
+ return buf.value
501
+
502
+
503
+ def _get_win_folder_with_jna(csidl_name):
504
+ import array
505
+ from com.sun import jna
506
+ from com.sun.jna.platform import win32
507
+
508
+ buf_size = win32.WinDef.MAX_PATH * 2
509
+ buf = array.zeros("c", buf_size)
510
+ shell = win32.Shell32.INSTANCE
511
+ shell.SHGetFolderPath(None, getattr(win32.ShlObj, csidl_name), None, win32.ShlObj.SHGFP_TYPE_CURRENT, buf)
512
+ dir = jna.Native.toString(buf.tostring()).rstrip("\0")
513
+
514
+ # Downgrade to short path name if have highbit chars. See
515
+ # <http://bugs.activestate.com/show_bug.cgi?id=85099>.
516
+ has_high_char = False
517
+ for c in dir:
518
+ if ord(c) > 255:
519
+ has_high_char = True
520
+ break
521
+ if has_high_char:
522
+ buf = array.zeros("c", buf_size)
523
+ kernel = win32.Kernel32.INSTANCE
524
+ if kernel.GetShortPathName(dir, buf, buf_size):
525
+ dir = jna.Native.toString(buf.tostring()).rstrip("\0")
526
+
527
+ return dir
528
+
529
+
530
+ def _get_win_folder_from_environ(csidl_name):
531
+ env_var_name = {
532
+ "CSIDL_APPDATA": "APPDATA",
533
+ "CSIDL_COMMON_APPDATA": "ALLUSERSPROFILE",
534
+ "CSIDL_LOCAL_APPDATA": "LOCALAPPDATA",
535
+ }[csidl_name]
536
+
537
+ return os.environ[env_var_name]
538
+
539
+
540
+ if system == "win32":
541
+ try:
542
+ from ctypes import windll
543
+ except ImportError:
544
+ try:
545
+ import com.sun.jna
546
+ except ImportError:
547
+ try:
548
+ if PY3:
549
+ import winreg as _winreg
550
+ else:
551
+ import _winreg
552
+ except ImportError:
553
+ _get_win_folder = _get_win_folder_from_environ
554
+ else:
555
+ _get_win_folder = _get_win_folder_from_registry
556
+ else:
557
+ _get_win_folder = _get_win_folder_with_jna
558
+ else:
559
+ _get_win_folder = _get_win_folder_with_ctypes
560
+
561
+
562
+ # ---- self test code
563
+
564
+ if __name__ == "__main__":
565
+ appname = "MyApp"
566
+ appauthor = "MyCompany"
567
+
568
+ props = (
569
+ "user_data_dir",
570
+ "user_config_dir",
571
+ "user_cache_dir",
572
+ "user_state_dir",
573
+ "user_log_dir",
574
+ "site_data_dir",
575
+ "site_config_dir",
576
+ )
577
+
578
+ print("-- app dirs %s --" % __version__)
579
+
580
+ print("-- app dirs (with optional 'version')")
581
+ dirs = AppDirs(appname, appauthor, version="1.0")
582
+ for prop in props:
583
+ print("%s: %s" % (prop, getattr(dirs, prop)))
584
+
585
+ print("\n-- app dirs (without optional 'version')")
586
+ dirs = AppDirs(appname, appauthor)
587
+ for prop in props:
588
+ print("%s: %s" % (prop, getattr(dirs, prop)))
589
+
590
+ print("\n-- app dirs (without optional 'appauthor')")
591
+ dirs = AppDirs(appname)
592
+ for prop in props:
593
+ print("%s: %s" % (prop, getattr(dirs, prop)))
594
+
595
+ print("\n-- app dirs (with disabled 'appauthor')")
596
+ dirs = AppDirs(appname, appauthor=False)
597
+ for prop in props:
598
+ print("%s: %s" % (prop, getattr(dirs, prop)))