Development of an internal social media platform with personalised dashboards for students
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

develop.py 7.9KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218
  1. from distutils.util import convert_path
  2. from distutils import log
  3. from distutils.errors import DistutilsError, DistutilsOptionError
  4. import os
  5. import glob
  6. import io
  7. from setuptools.extern import six
  8. from pkg_resources import Distribution, PathMetadata, normalize_path
  9. from setuptools.command.easy_install import easy_install
  10. from setuptools import namespaces
  11. import setuptools
  12. __metaclass__ = type
  13. class develop(namespaces.DevelopInstaller, easy_install):
  14. """Set up package for development"""
  15. description = "install package in 'development mode'"
  16. user_options = easy_install.user_options + [
  17. ("uninstall", "u", "Uninstall this source package"),
  18. ("egg-path=", None, "Set the path to be used in the .egg-link file"),
  19. ]
  20. boolean_options = easy_install.boolean_options + ['uninstall']
  21. command_consumes_arguments = False # override base
  22. def run(self):
  23. if self.uninstall:
  24. self.multi_version = True
  25. self.uninstall_link()
  26. self.uninstall_namespaces()
  27. else:
  28. self.install_for_development()
  29. self.warn_deprecated_options()
  30. def initialize_options(self):
  31. self.uninstall = None
  32. self.egg_path = None
  33. easy_install.initialize_options(self)
  34. self.setup_path = None
  35. self.always_copy_from = '.' # always copy eggs installed in curdir
  36. def finalize_options(self):
  37. ei = self.get_finalized_command("egg_info")
  38. if ei.broken_egg_info:
  39. template = "Please rename %r to %r before using 'develop'"
  40. args = ei.egg_info, ei.broken_egg_info
  41. raise DistutilsError(template % args)
  42. self.args = [ei.egg_name]
  43. easy_install.finalize_options(self)
  44. self.expand_basedirs()
  45. self.expand_dirs()
  46. # pick up setup-dir .egg files only: no .egg-info
  47. self.package_index.scan(glob.glob('*.egg'))
  48. egg_link_fn = ei.egg_name + '.egg-link'
  49. self.egg_link = os.path.join(self.install_dir, egg_link_fn)
  50. self.egg_base = ei.egg_base
  51. if self.egg_path is None:
  52. self.egg_path = os.path.abspath(ei.egg_base)
  53. target = normalize_path(self.egg_base)
  54. egg_path = normalize_path(os.path.join(self.install_dir,
  55. self.egg_path))
  56. if egg_path != target:
  57. raise DistutilsOptionError(
  58. "--egg-path must be a relative path from the install"
  59. " directory to " + target
  60. )
  61. # Make a distribution for the package's source
  62. self.dist = Distribution(
  63. target,
  64. PathMetadata(target, os.path.abspath(ei.egg_info)),
  65. project_name=ei.egg_name
  66. )
  67. self.setup_path = self._resolve_setup_path(
  68. self.egg_base,
  69. self.install_dir,
  70. self.egg_path,
  71. )
  72. @staticmethod
  73. def _resolve_setup_path(egg_base, install_dir, egg_path):
  74. """
  75. Generate a path from egg_base back to '.' where the
  76. setup script resides and ensure that path points to the
  77. setup path from $install_dir/$egg_path.
  78. """
  79. path_to_setup = egg_base.replace(os.sep, '/').rstrip('/')
  80. if path_to_setup != os.curdir:
  81. path_to_setup = '../' * (path_to_setup.count('/') + 1)
  82. resolved = normalize_path(
  83. os.path.join(install_dir, egg_path, path_to_setup)
  84. )
  85. if resolved != normalize_path(os.curdir):
  86. raise DistutilsOptionError(
  87. "Can't get a consistent path to setup script from"
  88. " installation directory", resolved, normalize_path(os.curdir))
  89. return path_to_setup
  90. def install_for_development(self):
  91. if six.PY3 and getattr(self.distribution, 'use_2to3', False):
  92. # If we run 2to3 we can not do this inplace:
  93. # Ensure metadata is up-to-date
  94. self.reinitialize_command('build_py', inplace=0)
  95. self.run_command('build_py')
  96. bpy_cmd = self.get_finalized_command("build_py")
  97. build_path = normalize_path(bpy_cmd.build_lib)
  98. # Build extensions
  99. self.reinitialize_command('egg_info', egg_base=build_path)
  100. self.run_command('egg_info')
  101. self.reinitialize_command('build_ext', inplace=0)
  102. self.run_command('build_ext')
  103. # Fixup egg-link and easy-install.pth
  104. ei_cmd = self.get_finalized_command("egg_info")
  105. self.egg_path = build_path
  106. self.dist.location = build_path
  107. # XXX
  108. self.dist._provider = PathMetadata(build_path, ei_cmd.egg_info)
  109. else:
  110. # Without 2to3 inplace works fine:
  111. self.run_command('egg_info')
  112. # Build extensions in-place
  113. self.reinitialize_command('build_ext', inplace=1)
  114. self.run_command('build_ext')
  115. self.install_site_py() # ensure that target dir is site-safe
  116. if setuptools.bootstrap_install_from:
  117. self.easy_install(setuptools.bootstrap_install_from)
  118. setuptools.bootstrap_install_from = None
  119. self.install_namespaces()
  120. # create an .egg-link in the installation dir, pointing to our egg
  121. log.info("Creating %s (link to %s)", self.egg_link, self.egg_base)
  122. if not self.dry_run:
  123. with open(self.egg_link, "w") as f:
  124. f.write(self.egg_path + "\n" + self.setup_path)
  125. # postprocess the installed distro, fixing up .pth, installing scripts,
  126. # and handling requirements
  127. self.process_distribution(None, self.dist, not self.no_deps)
  128. def uninstall_link(self):
  129. if os.path.exists(self.egg_link):
  130. log.info("Removing %s (link to %s)", self.egg_link, self.egg_base)
  131. egg_link_file = open(self.egg_link)
  132. contents = [line.rstrip() for line in egg_link_file]
  133. egg_link_file.close()
  134. if contents not in ([self.egg_path],
  135. [self.egg_path, self.setup_path]):
  136. log.warn("Link points to %s: uninstall aborted", contents)
  137. return
  138. if not self.dry_run:
  139. os.unlink(self.egg_link)
  140. if not self.dry_run:
  141. self.update_pth(self.dist) # remove any .pth link to us
  142. if self.distribution.scripts:
  143. # XXX should also check for entry point scripts!
  144. log.warn("Note: you must uninstall or replace scripts manually!")
  145. def install_egg_scripts(self, dist):
  146. if dist is not self.dist:
  147. # Installing a dependency, so fall back to normal behavior
  148. return easy_install.install_egg_scripts(self, dist)
  149. # create wrapper scripts in the script dir, pointing to dist.scripts
  150. # new-style...
  151. self.install_wrapper_scripts(dist)
  152. # ...and old-style
  153. for script_name in self.distribution.scripts or []:
  154. script_path = os.path.abspath(convert_path(script_name))
  155. script_name = os.path.basename(script_path)
  156. with io.open(script_path) as strm:
  157. script_text = strm.read()
  158. self.install_script(dist, script_name, script_text, script_path)
  159. def install_wrapper_scripts(self, dist):
  160. dist = VersionlessRequirement(dist)
  161. return easy_install.install_wrapper_scripts(self, dist)
  162. class VersionlessRequirement:
  163. """
  164. Adapt a pkg_resources.Distribution to simply return the project
  165. name as the 'requirement' so that scripts will work across
  166. multiple versions.
  167. >>> dist = Distribution(project_name='foo', version='1.0')
  168. >>> str(dist.as_requirement())
  169. 'foo==1.0'
  170. >>> adapted_dist = VersionlessRequirement(dist)
  171. >>> str(adapted_dist.as_requirement())
  172. 'foo'
  173. """
  174. def __init__(self, dist):
  175. self.__dist = dist
  176. def __getattr__(self, name):
  177. return getattr(self.__dist, name)
  178. def as_requirement(self):
  179. return self.project_name