makewheel.py 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902
  1. """
  2. Generates a wheel (.whl) file from the output of makepanda.
  3. """
  4. import json
  5. import sys
  6. import os
  7. from os.path import join
  8. import zipfile
  9. import hashlib
  10. import tempfile
  11. import subprocess
  12. import time
  13. import struct
  14. from sysconfig import get_platform, get_config_var
  15. from optparse import OptionParser
  16. from base64 import urlsafe_b64encode
  17. from makepandacore import LocateBinary, GetExtensionSuffix, SetVerbose, GetVerbose, GetMetadataValue, CrossCompiling, GetThirdpartyDir, SDK, GetStrip
  18. def get_abi_tag():
  19. soabi = get_config_var('SOABI')
  20. if soabi and soabi.startswith('cpython-'):
  21. return 'cp' + soabi.split('-')[1]
  22. elif soabi:
  23. return soabi.replace('.', '_').replace('-', '_')
  24. soabi = 'cp%d%d' % (sys.version_info[:2])
  25. if sys.version_info >= (3, 8):
  26. return soabi
  27. debug_flag = get_config_var('Py_DEBUG')
  28. if (debug_flag is None and hasattr(sys, 'gettotalrefcount')) or debug_flag:
  29. soabi += 'd'
  30. malloc_flag = get_config_var('WITH_PYMALLOC')
  31. if malloc_flag is None or malloc_flag:
  32. soabi += 'm'
  33. return soabi
  34. def is_exe_file(path):
  35. return os.path.isfile(path) and path.lower().endswith('.exe')
  36. def is_elf_file(path):
  37. base = os.path.basename(path)
  38. return os.path.isfile(path) and '.' not in base and \
  39. open(path, 'rb').read(4) == b'\x7FELF'
  40. def is_macho_or_fat_file(path):
  41. base = os.path.basename(path)
  42. return os.path.isfile(path) and '.' not in base and \
  43. open(path, 'rb').read(4) in (b'\xFE\xED\xFA\xCE', b'\xCE\xFA\xED\xFE',
  44. b'\xFE\xED\xFA\xCF', b'\xCF\xFA\xED\xFE',
  45. b'\xCA\xFE\xBA\xBE', b'\xBE\xBA\xFE\xCA',
  46. b'\xCA\xFE\xBA\xBF', b'\xBF\xBA\xFE\xCA')
  47. def is_fat_file(path):
  48. return os.path.isfile(path) and \
  49. open(path, 'rb').read(4) in (b'\xCA\xFE\xBA\xBE', b'\xBE\xBA\xFE\xCA',
  50. b'\xCA\xFE\xBA\xBF', b'\xBF\xBA\xFE\xCA')
  51. def get_python_ext_module_dir():
  52. if CrossCompiling():
  53. return os.path.join(GetThirdpartyDir(), "python", "lib", SDK["PYTHONVERSION"], "lib-dynload")
  54. else:
  55. import _ctypes
  56. return os.path.dirname(_ctypes.__file__)
  57. if sys.platform in ('win32', 'cygwin'):
  58. is_executable = is_exe_file
  59. elif sys.platform == 'darwin':
  60. is_executable = is_macho_or_fat_file
  61. else:
  62. is_executable = is_elf_file
  63. # Other global parameters
  64. PY_VERSION = "cp{0}{1}".format(*sys.version_info)
  65. ABI_TAG = get_abi_tag()
  66. EXCLUDE_EXT = [".pyc", ".pyo", ".N", ".prebuilt", ".xcf", ".plist", ".vcproj", ".sln"]
  67. # Plug-ins to install.
  68. PLUGIN_LIBS = ["pandagl", "pandagles", "pandagles2", "pandadx9", "p3tinydisplay", "p3ptloader", "p3assimp", "p3ffmpeg", "p3openal_audio", "p3fmod_audio", "p3headlessgl"]
  69. # Libraries included in manylinux ABI that should be ignored. See PEP 513/571/599.
  70. MANYLINUX_LIBS = [
  71. "libgcc_s.so.1", "libstdc++.so.6", "libm.so.6", "libdl.so.2", "librt.so.1",
  72. "libcrypt.so.1", "libc.so.6", "libnsl.so.1", "libutil.so.1",
  73. "libpthread.so.0", "libresolv.so.2", "libX11.so.6", "libXext.so.6",
  74. "libXrender.so.1", "libICE.so.6", "libSM.so.6", "libGL.so.1",
  75. "libgobject-2.0.so.0", "libgthread-2.0.so.0", "libglib-2.0.so.0",
  76. # These are not mentioned in manylinux1 spec but should nonetheless always
  77. # be excluded.
  78. "linux-vdso.so.1", "linux-gate.so.1", "ld-linux.so.2", "libdrm.so.2",
  79. "libEGL.so.1", "libOpenGL.so.0", "libGLX.so.0", "libGLdispatch.so.0",
  80. ]
  81. # Binaries to never scan for dependencies on non-Windows systems.
  82. IGNORE_UNIX_DEPS_OF = [
  83. "panda3d_tools/pstats",
  84. ]
  85. WHEEL_DATA = """Wheel-Version: 1.0
  86. Generator: makepanda
  87. Root-Is-Purelib: false
  88. Tag: {0}-{1}-{2}
  89. """
  90. PROJECT_URLS = dict([line.split('=', 1) for line in GetMetadataValue('project_urls').strip().splitlines()])
  91. METADATA = {
  92. "license": GetMetadataValue('license'),
  93. "name": GetMetadataValue('name'),
  94. "metadata_version": "2.0",
  95. "generator": "makepanda",
  96. "summary": GetMetadataValue('description'),
  97. "extensions": {
  98. "python.details": {
  99. "project_urls": dict(PROJECT_URLS, Home=GetMetadataValue('url')),
  100. "document_names": {
  101. "license": "LICENSE.txt"
  102. },
  103. "contacts": [
  104. {
  105. "role": "author",
  106. "name": GetMetadataValue('author'),
  107. "email": GetMetadataValue('author_email'),
  108. }
  109. ]
  110. }
  111. },
  112. "classifiers": GetMetadataValue('classifiers'),
  113. }
  114. DESCRIPTION = """
  115. The Panda3D free 3D game engine
  116. ===============================
  117. Panda3D is a powerful 3D engine written in C++, with a complete set of Python
  118. bindings. Unlike other engines, these bindings are automatically generated,
  119. meaning that they are always up-to-date and complete: all functions of the
  120. engine can be controlled from Python. All major Panda3D applications have been
  121. written in Python, this is the intended way of using the engine.
  122. Panda3D now supports automatic shader generation, which now means you can use
  123. normal maps, gloss maps, glow maps, HDR, cartoon shading, and the like without
  124. having to write any shaders.
  125. Panda3D is a modern engine supporting advanced features such as shaders,
  126. stencil, and render-to-texture. Panda3D is unusual in that it emphasizes a
  127. short learning curve, rapid development, and extreme stability and robustness.
  128. Panda3D is free software that runs under Windows, Linux, or macOS.
  129. The Panda3D team is very concerned with making the engine accessible to new
  130. users. We provide a detailed manual, a complete API reference, and a large
  131. collection of sample programs to help you get started. We have active forums,
  132. with many helpful users, and the developers are regularly online to answer
  133. questions.
  134. """
  135. PANDA3D_TOOLS_INIT = """import os, sys
  136. import panda3d
  137. dir = os.path.dirname(panda3d.__file__)
  138. del panda3d
  139. if sys.platform in ('win32', 'cygwin'):
  140. path_var = 'PATH'
  141. if hasattr(os, 'add_dll_directory'):
  142. os.add_dll_directory(dir)
  143. elif sys.platform == 'darwin':
  144. path_var = 'DYLD_LIBRARY_PATH'
  145. else:
  146. path_var = 'LD_LIBRARY_PATH'
  147. if not os.environ.get(path_var):
  148. os.environ[path_var] = dir
  149. else:
  150. os.environ[path_var] = dir + os.pathsep + os.environ[path_var]
  151. del os, sys, path_var, dir
  152. def _exec_tool(tool):
  153. import os, sys
  154. from subprocess import Popen
  155. tools_dir = os.path.dirname(__file__)
  156. handle = Popen(sys.argv, executable=os.path.join(tools_dir, tool))
  157. try:
  158. try:
  159. return handle.wait()
  160. except KeyboardInterrupt:
  161. # Give the program a chance to handle the signal gracefully.
  162. return handle.wait()
  163. except:
  164. handle.kill()
  165. handle.wait()
  166. raise
  167. # Register all the executables in this directory as global functions.
  168. {0}
  169. """
  170. def parse_dependencies_windows(data):
  171. """ Parses the given output from dumpbin /dependents to determine the list
  172. of dll's this executable file depends on. """
  173. lines = data.splitlines()
  174. li = 0
  175. while li < len(lines):
  176. line = lines[li]
  177. li += 1
  178. if line.find(' has the following dependencies') != -1:
  179. break
  180. if li < len(lines):
  181. line = lines[li]
  182. if line.strip() == '':
  183. # Skip a blank line.
  184. li += 1
  185. # Now we're finding filenames, until the next blank line.
  186. filenames = []
  187. while li < len(lines):
  188. line = lines[li]
  189. li += 1
  190. line = line.strip()
  191. if line == '':
  192. # We're done.
  193. return filenames
  194. filenames.append(line)
  195. # At least we got some data.
  196. return filenames
  197. def parse_dependencies_unix(data):
  198. """ Parses the given output from otool -XL or ldd to determine the list of
  199. libraries this executable file depends on. """
  200. lines = data.splitlines()
  201. filenames = []
  202. for l in lines:
  203. l = l.strip()
  204. if l != "statically linked":
  205. filenames.append(l.split(' ', 1)[0])
  206. return filenames
  207. def _scan_dependencies_elf(elf):
  208. deps = []
  209. ident = elf.read(12)
  210. # Make sure we read in the correct endianness and integer size
  211. byte_order = "<>"[ord(ident[1:2]) - 1]
  212. elf_class = ord(ident[0:1]) - 1 # 0 = 32-bits, 1 = 64-bits
  213. header_struct = byte_order + ("HHIIIIIHHHHHH", "HHIQQQIHHHHHH")[elf_class]
  214. section_struct = byte_order + ("4xI8xIII8xI", "4xI16xQQI12xQ")[elf_class]
  215. dynamic_struct = byte_order + ("iI", "qQ")[elf_class]
  216. type, machine, version, entry, phoff, shoff, flags, ehsize, phentsize, phnum, shentsize, shnum, shstrndx \
  217. = struct.unpack(header_struct, elf.read(struct.calcsize(header_struct)))
  218. dynamic_sections = []
  219. string_tables = {}
  220. # Seek to the section header table and find the .dynamic section.
  221. elf.seek(shoff)
  222. for i in range(shnum):
  223. type, offset, size, link, entsize = struct.unpack_from(section_struct, elf.read(shentsize))
  224. if type == 6 and link != 0: # DYNAMIC type, links to string table
  225. dynamic_sections.append((offset, size, link, entsize))
  226. string_tables[link] = None
  227. # Read the relevant string tables.
  228. for idx in string_tables.keys():
  229. elf.seek(shoff + idx * shentsize)
  230. type, offset, size, link, entsize = struct.unpack_from(section_struct, elf.read(shentsize))
  231. if type != 3: continue
  232. elf.seek(offset)
  233. string_tables[idx] = elf.read(size)
  234. # Loop through the dynamic sections to get the NEEDED entries.
  235. needed = []
  236. for offset, size, link, entsize in dynamic_sections:
  237. elf.seek(offset)
  238. data = elf.read(entsize)
  239. tag, val = struct.unpack_from(dynamic_struct, data)
  240. # Read tags until we find a NULL tag.
  241. while tag != 0:
  242. if tag == 1: # A NEEDED entry. Read it from the string table.
  243. string = string_tables[link][val : string_tables[link].find(b'\0', val)]
  244. needed.append(string.decode('utf-8'))
  245. data = elf.read(entsize)
  246. tag, val = struct.unpack_from(dynamic_struct, data)
  247. elf.close()
  248. return needed
  249. def scan_dependencies(pathname):
  250. """ Checks the named file for DLL dependencies, and adds any appropriate
  251. dependencies found into pluginDependencies and dependentFiles. """
  252. with open(pathname, 'rb') as fh:
  253. if fh.read(4) == b'\x7FELF':
  254. return _scan_dependencies_elf(fh)
  255. if sys.platform == "darwin":
  256. command = ['otool', '-XL', pathname]
  257. elif sys.platform in ("win32", "cygwin"):
  258. command = ['dumpbin', '/dependents', pathname]
  259. else:
  260. sys.exit("Don't know how to determine dependencies from %s" % (pathname))
  261. process = subprocess.Popen(command, stdout=subprocess.PIPE, universal_newlines=True)
  262. output, unused_err = process.communicate()
  263. retcode = process.poll()
  264. if retcode:
  265. raise subprocess.CalledProcessError(retcode, command[0], output=output)
  266. filenames = None
  267. if sys.platform in ("win32", "cygwin"):
  268. filenames = parse_dependencies_windows(output)
  269. else:
  270. filenames = parse_dependencies_unix(output)
  271. if filenames is None:
  272. sys.exit("Unable to determine dependencies from %s" % (pathname))
  273. if sys.platform == "darwin" and len(filenames) > 0:
  274. # Filter out the library ID.
  275. if os.path.basename(filenames[0]).split('.', 1)[0] == os.path.basename(pathname).split('.', 1)[0]:
  276. del filenames[0]
  277. return filenames
  278. class WheelFile(object):
  279. def __init__(self, name, version, platform):
  280. self.name = name
  281. self.version = version
  282. self.platform = platform
  283. wheel_name = "{0}-{1}-{2}-{3}-{4}.whl".format(
  284. name, version, PY_VERSION, ABI_TAG, platform)
  285. print("Writing %s" % (wheel_name))
  286. self.zip_file = zipfile.ZipFile(wheel_name, 'w', zipfile.ZIP_DEFLATED)
  287. self.records = []
  288. # Used to locate dependency libraries.
  289. self.lib_path = []
  290. self.dep_paths = {}
  291. self.ignore_deps = set()
  292. # This can be set if a reproducible (deterministic) build is desired, in
  293. # which case we have to clamp all dates to the given SOURCE_DATE_EPOCH.
  294. epoch = os.environ.get('SOURCE_DATE_EPOCH')
  295. self.max_date_time = time.localtime(int(epoch) if epoch else time.time())[:6]
  296. if self.max_date_time < (1980, 1, 1, 0, 0, 0):
  297. # Earliest representable time in zip archives.
  298. self.max_date_time = (1980, 1, 1, 0, 0, 0)
  299. def consider_add_dependency(self, target_path, dep, search_path=None):
  300. """Considers adding a dependency library.
  301. Returns the target_path if it was added, which may be different from
  302. target_path if it was already added earlier, or None if it wasn't."""
  303. if dep in self.dep_paths:
  304. # Already considered this.
  305. return self.dep_paths[dep]
  306. self.dep_paths[dep] = None
  307. if dep in self.ignore_deps:
  308. if GetVerbose():
  309. print("Ignoring {0} (explicitly ignored)".format(dep))
  310. return
  311. if not self.platform.startswith("android"):
  312. if dep.lower().startswith("python") or os.path.basename(dep).startswith("libpython"):
  313. if GetVerbose():
  314. print("Ignoring {0} (explicitly ignored)".format(dep))
  315. return
  316. if self.platform.startswith("macosx"):
  317. if dep.endswith(".so"):
  318. # Temporary hack for 1.9, which had link deps on modules.
  319. return
  320. if dep.startswith("/System/"):
  321. return
  322. if dep.startswith('/'):
  323. source_path = dep
  324. else:
  325. source_path = None
  326. if search_path is None:
  327. search_path = self.lib_path
  328. for lib_dir in search_path:
  329. # Ignore static stuff.
  330. path = os.path.join(lib_dir, dep)
  331. if os.path.isfile(path):
  332. source_path = os.path.normpath(path)
  333. break
  334. if not source_path:
  335. # Couldn't find library in the panda3d lib dir.
  336. if GetVerbose():
  337. print("Ignoring {0} (not in search path)".format(dep))
  338. return
  339. self.dep_paths[dep] = target_path
  340. self.write_file(target_path, source_path)
  341. return target_path
  342. def write_file(self, target_path, source_path):
  343. """Adds the given file to the .whl file."""
  344. orig_source_path = source_path
  345. # If this is a .so file, we should set the rpath appropriately.
  346. temp = None
  347. basename, ext = os.path.splitext(source_path)
  348. if ext in ('.so', '.dylib') or '.so.' in os.path.basename(source_path) or \
  349. (not ext and is_executable(source_path)):
  350. # Scan Unix dependencies.
  351. if target_path not in IGNORE_UNIX_DEPS_OF:
  352. deps = scan_dependencies(source_path)
  353. else:
  354. deps = []
  355. suffix = ''
  356. if '.so' in os.path.basename(source_path):
  357. suffix = '.so'
  358. elif ext == '.dylib':
  359. suffix = '.dylib'
  360. temp = tempfile.NamedTemporaryFile(suffix=suffix, prefix='whl', delete=False)
  361. # On macOS, if no fat wheel was requested, extract the right architecture.
  362. if self.platform.startswith("macosx") and is_fat_file(source_path) \
  363. and not self.platform.endswith("_intel") \
  364. and "_fat" not in self.platform \
  365. and "_universal" not in self.platform:
  366. if self.platform.endswith("_x86_64"):
  367. arch = 'x86_64'
  368. else:
  369. arch = self.platform.split('_')[-1]
  370. subprocess.call(['lipo', source_path, '-extract', arch, '-output', temp.name])
  371. else:
  372. # Otherwise, just copy it over.
  373. temp.write(open(source_path, 'rb').read())
  374. temp.close()
  375. os.chmod(temp.name, os.stat(temp.name).st_mode | 0o711)
  376. # Now add dependencies. On macOS, fix @loader_path references.
  377. if self.platform.startswith("macosx"):
  378. if source_path.endswith('deploy-stubw'):
  379. deps_path = '@executable_path/../Frameworks'
  380. else:
  381. deps_path = '@loader_path'
  382. loader_path = [os.path.dirname(source_path)]
  383. for dep in deps:
  384. if dep.endswith('/Python'):
  385. # If this references the Python framework, change it
  386. # to reference libpython instead.
  387. new_dep = deps_path + '/libpython{0}.{1}.dylib'.format(*sys.version_info)
  388. elif '@loader_path' in dep:
  389. dep_path = dep.replace('@loader_path', '.')
  390. target_dep = os.path.dirname(target_path) + '/' + os.path.basename(dep)
  391. target_dep = self.consider_add_dependency(target_dep, dep_path, loader_path)
  392. if not target_dep:
  393. # It won't be included, so no use adjusting the path.
  394. continue
  395. new_dep = os.path.join(deps_path, os.path.relpath(target_dep, os.path.dirname(target_path)))
  396. elif '@rpath' in dep:
  397. # Unlike makepanda, CMake uses @rpath instead of
  398. # @loader_path. This means we can just search for the
  399. # dependencies like normal.
  400. dep_path = dep.replace('@rpath', '.')
  401. target_dep = os.path.dirname(target_path) + '/' + os.path.basename(dep)
  402. self.consider_add_dependency(target_dep, dep_path)
  403. continue
  404. elif dep.startswith('/Library/Frameworks/Python.framework/'):
  405. # Add this dependency if it's in the Python directory.
  406. target_dep = os.path.dirname(target_path) + '/' + os.path.basename(dep)
  407. target_dep = self.consider_add_dependency(target_dep, dep, loader_path)
  408. if not target_dep:
  409. # It won't be included, so no use adjusting the path.
  410. continue
  411. new_dep = os.path.join(deps_path, os.path.relpath(target_dep, os.path.dirname(target_path)))
  412. else:
  413. if '/' in dep:
  414. if GetVerbose():
  415. print("Ignoring dependency %s" % (dep))
  416. continue
  417. subprocess.call(["install_name_tool", "-change", dep, new_dep, temp.name])
  418. # Make sure it has an ad-hoc code signature.
  419. subprocess.call(["codesign", "-f", "-s", "-", temp.name])
  420. else:
  421. # On other unixes, we just add dependencies normally.
  422. for dep in deps:
  423. # Only include dependencies with relative path, for now.
  424. if '/' in dep:
  425. continue
  426. if self.platform.startswith('android') and '.so.' in dep:
  427. # Change .so.1.2 suffix to .so, to allow loading in .apk
  428. new_dep = dep.rpartition('.so.')[0] + '.so'
  429. subprocess.call(["patchelf", "--replace-needed", dep, new_dep, temp.name])
  430. target_dep = os.path.dirname(target_path) + '/' + new_dep
  431. else:
  432. target_dep = os.path.dirname(target_path) + '/' + dep
  433. self.consider_add_dependency(target_dep, dep)
  434. subprocess.call([GetStrip(), "-s", temp.name])
  435. if self.platform.startswith('android'):
  436. # We must link explicitly with Python, because the usual
  437. # -rdynamic trick doesn't work from a shared library loaded
  438. # through ANativeActivity.
  439. if suffix == '.so' and not os.path.basename(source_path).startswith('lib'):
  440. pylib_name = "libpython" + get_config_var('LDVERSION') + ".so"
  441. subprocess.call(["patchelf", "--add-needed", pylib_name, temp.name])
  442. else:
  443. # On other systems, we use the rpath to force it to locate
  444. # dependencies in the same directory.
  445. subprocess.call(["patchelf", "--force-rpath", "--set-rpath", "$ORIGIN", temp.name])
  446. source_path = temp.name
  447. ext = ext.lower()
  448. if ext in ('.dll', '.pyd', '.exe'):
  449. # Scan and add Win32 dependencies.
  450. for dep in scan_dependencies(source_path):
  451. target_dep = os.path.dirname(target_path) + '/' + dep
  452. self.consider_add_dependency(target_dep, dep)
  453. if GetVerbose():
  454. print("Adding {0} from {1}".format(target_path, orig_source_path))
  455. zinfo = zipfile.ZipInfo.from_file(source_path, target_path)
  456. zinfo.compress_type = self.zip_file.compression
  457. if zinfo.date_time > self.max_date_time:
  458. zinfo.date_time = self.max_date_time
  459. # Copy the data to the zip file, while also calculating the SHA-256.
  460. size = 0
  461. sha = hashlib.sha256()
  462. with open(source_path, 'rb') as source_fp, self.zip_file.open(zinfo, 'w') as target_fp:
  463. data = source_fp.read(1024 * 1024)
  464. while data:
  465. size += len(data)
  466. target_fp.write(data)
  467. sha.update(data)
  468. data = source_fp.read(1024 * 1024)
  469. # Save it in PEP-0376 format for writing out later.
  470. digest = urlsafe_b64encode(sha.digest()).decode('ascii')
  471. digest = digest.rstrip('=')
  472. self.records.append("{0},sha256={1},{2}\n".format(target_path, digest, size))
  473. #if temp:
  474. # os.unlink(temp.name)
  475. def write_file_data(self, target_path, source_data):
  476. """Adds the given file from a string."""
  477. sha = hashlib.sha256()
  478. sha.update(source_data.encode())
  479. digest = urlsafe_b64encode(sha.digest()).decode('ascii')
  480. digest = digest.rstrip('=')
  481. self.records.append("{0},sha256={1},{2}\n".format(target_path, digest, len(source_data)))
  482. if GetVerbose():
  483. print("Adding %s from data" % target_path)
  484. zinfo = zipfile.ZipInfo(filename=target_path,
  485. date_time=self.max_date_time)
  486. zinfo.compress_type = self.zip_file.compression
  487. zinfo.external_attr = 0o600 << 16
  488. self.zip_file.writestr(zinfo, source_data)
  489. def write_directory(self, target_dir, source_dir):
  490. """Adds the given directory recursively to the .whl file."""
  491. for root, dirs, files in os.walk(source_dir):
  492. dirs.sort()
  493. for file in sorted(files):
  494. if os.path.splitext(file)[1] in EXCLUDE_EXT:
  495. continue
  496. source_path = os.path.join(root, file)
  497. target_path = os.path.join(target_dir, os.path.relpath(source_path, source_dir))
  498. target_path = target_path.replace('\\', '/')
  499. self.write_file(target_path, source_path)
  500. def close(self):
  501. # Write the RECORD file.
  502. record_file = "{0}-{1}.dist-info/RECORD".format(self.name, self.version)
  503. self.records.append(record_file + ",,\n")
  504. zinfo = zipfile.ZipInfo(filename=record_file,
  505. date_time=self.max_date_time)
  506. zinfo.compress_type = self.zip_file.compression
  507. zinfo.external_attr = 0o600 << 16
  508. self.zip_file.writestr(zinfo, "".join(self.records))
  509. self.zip_file.close()
  510. def makewheel(version, output_dir, platform=None):
  511. if sys.platform not in ("win32", "darwin") and not sys.platform.startswith("cygwin"):
  512. if not LocateBinary("patchelf"):
  513. raise Exception("patchelf is required when building a Linux wheel.")
  514. if sys.version_info < (3, 6):
  515. raise Exception("Python 3.6 or higher is required to produce a wheel.")
  516. if platform is None:
  517. # Determine the platform from the build.
  518. platform_dat = os.path.join(output_dir, 'tmp', 'platform.dat')
  519. if os.path.isfile(platform_dat):
  520. platform = open(platform_dat, 'r').read().strip()
  521. else:
  522. print("Could not find platform.dat in build directory")
  523. platform = get_platform()
  524. if platform.startswith("linux-") and os.path.isdir("/opt/python"):
  525. # Is this manylinux?
  526. if os.path.isfile("/lib/libc-2.5.so") or os.path.isfile("/lib64/libc-2.5.so"):
  527. platform = platform.replace("linux", "manylinux1")
  528. elif os.path.isfile("/lib/libc-2.12.so") or os.path.isfile("/lib64/libc-2.12.so"):
  529. platform = platform.replace("linux", "manylinux2010")
  530. elif os.path.isfile("/lib/libc-2.17.so") or os.path.isfile("/lib64/libc-2.17.so"):
  531. platform = platform.replace("linux", "manylinux2014")
  532. elif os.path.isfile("/lib/i386-linux-gnu/libc-2.24.so") or os.path.isfile("/lib/x86_64-linux-gnu/libc-2.24.so"):
  533. platform = platform.replace("linux", "manylinux_2_24")
  534. platform = platform.replace('-', '_').replace('.', '_')
  535. is_windows = platform == 'win32' \
  536. or platform.startswith('win_') \
  537. or platform.startswith('cygwin_')
  538. is_macosx = platform.startswith('macosx_')
  539. # Global filepaths
  540. panda3d_dir = join(output_dir, "panda3d")
  541. pandac_dir = join(output_dir, "pandac")
  542. direct_dir = join(output_dir, "direct")
  543. models_dir = join(output_dir, "models")
  544. etc_dir = join(output_dir, "etc")
  545. bin_dir = join(output_dir, "bin")
  546. if is_windows:
  547. libs_dir = join(output_dir, "bin")
  548. else:
  549. libs_dir = join(output_dir, "lib")
  550. ext_mod_dir = get_python_ext_module_dir()
  551. license_src = "LICENSE"
  552. readme_src = "README.md"
  553. # Update relevant METADATA entries
  554. METADATA['version'] = version
  555. # Build out the metadata
  556. details = METADATA["extensions"]["python.details"]
  557. homepage = details["project_urls"]["Home"]
  558. author = details["contacts"][0]["name"]
  559. email = details["contacts"][0]["email"]
  560. metadata = ''.join([
  561. "Metadata-Version: {metadata_version}\n" \
  562. "Name: {name}\n" \
  563. "Version: {version}\n" \
  564. "Summary: {summary}\n" \
  565. "License: {license}\n".format(**METADATA),
  566. "Home-page: {0}\n".format(homepage),
  567. ] + ["Project-URL: {0}, {1}\n".format(*url) for url in PROJECT_URLS.items()] + [
  568. "Author: {0}\n".format(author),
  569. "Author-email: {0}\n".format(email),
  570. "Platform: {0}\n".format(platform),
  571. ] + ["Classifier: {0}\n".format(c) for c in METADATA['classifiers']])
  572. metadata += '\n' + DESCRIPTION.strip() + '\n'
  573. # Zip it up and name it the right thing
  574. whl = WheelFile('panda3d', version, platform)
  575. whl.lib_path = [libs_dir]
  576. if is_windows:
  577. whl.lib_path.append(ext_mod_dir)
  578. if platform.startswith("manylinux"):
  579. # On manylinux1, we pick up all libraries except for the ones specified
  580. # by the manylinux1 ABI.
  581. whl.lib_path.append("/usr/local/lib")
  582. if platform.endswith("_x86_64"):
  583. whl.lib_path += ["/lib64", "/usr/lib64"]
  584. else:
  585. whl.lib_path += ["/lib", "/usr/lib"]
  586. whl.ignore_deps.update(MANYLINUX_LIBS)
  587. # Add libpython for deployment.
  588. if is_windows:
  589. pylib_name = 'python{0}{1}.dll'.format(*sys.version_info)
  590. pylib_path = os.path.join(get_config_var('BINDIR'), pylib_name)
  591. elif is_macosx:
  592. pylib_name = 'libpython{0}.{1}.dylib'.format(*sys.version_info)
  593. pylib_path = os.path.join(get_config_var('LIBDIR'), pylib_name)
  594. else:
  595. pylib_name = get_config_var('LDLIBRARY')
  596. pylib_arch = get_config_var('MULTIARCH')
  597. libdir = get_config_var('LIBDIR')
  598. if pylib_arch and os.path.exists(os.path.join(libdir, pylib_arch, pylib_name)):
  599. pylib_path = os.path.join(libdir, pylib_arch, pylib_name)
  600. else:
  601. pylib_path = os.path.join(libdir, pylib_name)
  602. # If Python was linked statically, we don't need to include this.
  603. if not pylib_name.endswith('.a'):
  604. whl.write_file('deploy_libs/' + pylib_name, pylib_path)
  605. # Add the trees with Python modules.
  606. whl.write_directory('direct', direct_dir)
  607. # Write the panda3d tree. We use a custom empty __init__ since the
  608. # default one adds the bin directory to the PATH, which we don't have.
  609. p3d_init = """"Python bindings for the Panda3D libraries"
  610. __version__ = '{0}'
  611. """.format(version)
  612. if '27' in ABI_TAG:
  613. p3d_init += """
  614. if __debug__:
  615. if 1 / 2 == 0:
  616. raise ImportError(\"Python 2 is not supported.\")
  617. """
  618. whl.write_file_data('panda3d/__init__.py', p3d_init)
  619. # Copy the extension modules from the panda3d directory.
  620. ext_suffix = GetExtensionSuffix()
  621. for file in sorted(os.listdir(panda3d_dir)):
  622. if file == '__init__.py':
  623. pass
  624. elif file.endswith('.py') or (file.endswith(ext_suffix) and '.' not in file[:-len(ext_suffix)]):
  625. source_path = os.path.join(panda3d_dir, file)
  626. if file.endswith('.pyd') and platform.startswith('cygwin'):
  627. # Rename it to .dll for cygwin Python to be able to load it.
  628. target_path = 'panda3d/' + os.path.splitext(file)[0] + '.dll'
  629. elif file.endswith(ext_suffix) and platform.startswith('android'):
  630. # Strip the extension suffix on Android.
  631. target_path = 'panda3d/' + file[:-len(ext_suffix)] + '.so'
  632. else:
  633. target_path = 'panda3d/' + file
  634. whl.write_file(target_path, source_path)
  635. # And copy the extension modules from the Python installation into the
  636. # deploy_libs directory, for use by deploy-ng.
  637. ext_suffix = '.pyd' if is_windows else '.so'
  638. for file in sorted(os.listdir(ext_mod_dir)):
  639. if file.endswith(ext_suffix):
  640. if file.startswith('_tkinter.'):
  641. # Tkinter is supplied in a separate wheel.
  642. continue
  643. source_path = os.path.join(ext_mod_dir, file)
  644. if file.endswith('.pyd') and platform.startswith('cygwin'):
  645. # Rename it to .dll for cygwin Python to be able to load it.
  646. target_path = 'deploy_libs/' + os.path.splitext(file)[0] + '.dll'
  647. else:
  648. target_path = 'deploy_libs/' + file
  649. whl.write_file(target_path, source_path)
  650. # Add plug-ins.
  651. for lib in PLUGIN_LIBS:
  652. plugin_name = 'lib' + lib
  653. if is_windows:
  654. plugin_name += '.dll'
  655. elif is_macosx:
  656. plugin_name += '.dylib'
  657. else:
  658. plugin_name += '.so'
  659. plugin_path = os.path.join(libs_dir, plugin_name)
  660. if os.path.isfile(plugin_path):
  661. whl.write_file('panda3d/' + plugin_name, plugin_path)
  662. if platform.startswith('android'):
  663. deploy_stub_path = os.path.join(libs_dir, 'libdeploy-stubw.so')
  664. if os.path.isfile(deploy_stub_path):
  665. whl.write_file('deploy_libs/libdeploy-stubw.so', deploy_stub_path)
  666. classes_dex_path = os.path.join(output_dir, 'classes.dex')
  667. if os.path.isfile(classes_dex_path):
  668. whl.write_file('deploy_libs/classes.dex', classes_dex_path)
  669. # Add the .data directory, containing additional files.
  670. data_dir = 'panda3d-{0}.data'.format(version)
  671. #whl.write_directory(data_dir + '/data/etc', etc_dir)
  672. #whl.write_directory(data_dir + '/data/models', models_dir)
  673. # Actually, let's not. That seems to install the files to the strangest
  674. # places in the user's filesystem. Let's instead put them in panda3d.
  675. whl.write_directory('panda3d/etc', etc_dir)
  676. whl.write_directory('panda3d/models', models_dir)
  677. # Add the pandac tree for backward compatibility.
  678. for file in sorted(os.listdir(pandac_dir)):
  679. if file.endswith('.py'):
  680. whl.write_file('pandac/' + file, os.path.join(pandac_dir, file))
  681. # Let's also add the interrogate databases.
  682. input_dir = os.path.join(pandac_dir, 'input')
  683. if os.path.isdir(input_dir):
  684. for file in sorted(os.listdir(input_dir)):
  685. if file.endswith('.in'):
  686. whl.write_file('pandac/input/' + file, os.path.join(input_dir, file))
  687. # Add a panda3d-tools directory containing the executables.
  688. entry_points = '[console_scripts]\n'
  689. entry_points += 'eggcacher = direct.directscripts.eggcacher:main\n'
  690. entry_points += 'pfreeze = direct.dist.pfreeze:main\n'
  691. tools_init = ''
  692. for file in sorted(os.listdir(bin_dir)):
  693. basename = os.path.splitext(file)[0]
  694. if basename in ('eggcacher', 'packpanda'):
  695. continue
  696. source_path = os.path.join(bin_dir, file)
  697. if is_executable(source_path):
  698. # Put the .exe files inside the panda3d-tools directory.
  699. whl.write_file('panda3d_tools/' + file, source_path)
  700. if basename.endswith('_bin'):
  701. # These tools won't be invoked by the user directly.
  702. continue
  703. # Tell pip to create a wrapper script.
  704. funcname = basename.replace('-', '_')
  705. entry_points += '{0} = panda3d_tools:{1}\n'.format(basename, funcname)
  706. tools_init += '{0} = lambda: _exec_tool({1!r})\n'.format(funcname, file)
  707. entry_points += '[distutils.commands]\n'
  708. entry_points += 'build_apps = direct.dist.commands:build_apps\n'
  709. entry_points += 'bdist_apps = direct.dist.commands:bdist_apps\n'
  710. whl.write_file_data('panda3d_tools/__init__.py', PANDA3D_TOOLS_INIT.format(tools_init))
  711. # Add the dist-info directory last.
  712. info_dir = 'panda3d-{0}.dist-info'.format(version)
  713. whl.write_file_data(info_dir + '/entry_points.txt', entry_points)
  714. whl.write_file_data(info_dir + '/metadata.json', json.dumps(METADATA, indent=4, separators=(',', ': ')))
  715. whl.write_file_data(info_dir + '/METADATA', metadata)
  716. whl.write_file_data(info_dir + '/WHEEL', WHEEL_DATA.format(PY_VERSION, ABI_TAG, platform))
  717. whl.write_file(info_dir + '/LICENSE.txt', license_src)
  718. whl.write_file(info_dir + '/README.md', readme_src)
  719. whl.write_file_data(info_dir + '/top_level.txt', 'direct\npanda3d\npandac\npanda3d_tools\n')
  720. whl.close()
  721. if __name__ == "__main__":
  722. version = GetMetadataValue('version')
  723. parser = OptionParser()
  724. parser.add_option('', '--version', dest = 'version', help = 'Panda3D version number (default: %s)' % (version), default = version)
  725. parser.add_option('', '--outputdir', dest = 'outputdir', help = 'Makepanda\'s output directory (default: built)', default = 'built')
  726. parser.add_option('', '--verbose', dest = 'verbose', help = 'Enable verbose output', action = 'store_true', default = False)
  727. parser.add_option('', '--platform', dest = 'platform', help = 'Override platform tag', default = None)
  728. (options, args) = parser.parse_args()
  729. SetVerbose(options.verbose)
  730. makewheel(options.version, options.outputdir, options.platform)