makewheel.py 36 KB

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