version.py 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. from __future__ import unicode_literals
  2. import datetime
  3. import os
  4. import subprocess
  5. def get_version(version=None):
  6. "Returns a PEP 440-compliant version number from VERSION."
  7. version = get_complete_version(version)
  8. # Now build the two parts of the version number:
  9. # main = X.Y[.Z]
  10. # sub = .devN - for pre-alpha releases
  11. # | {a|b|rc}N - for alpha, beta, and rc releases
  12. main = get_main_version(version)
  13. sub = ""
  14. if version[3] == "alpha" and version[4] == 0:
  15. git_changeset = get_git_changeset()
  16. if git_changeset:
  17. sub = ".dev%s" % git_changeset
  18. else:
  19. sub = ".dev"
  20. elif version[3] != "final":
  21. mapping = {"alpha": "a", "beta": "b", "rc": "rc"}
  22. sub = mapping[version[3]] + str(version[4])
  23. return str(main + sub)
  24. def get_main_version(version=None):
  25. "Returns main version (X.Y[.Z]) from VERSION."
  26. version = get_complete_version(version)
  27. parts = 2 if version[2] == 0 else 3
  28. return ".".join(str(x) for x in version[:parts])
  29. def get_complete_version(version=None):
  30. """Returns a tuple of the promise version. If version argument is non-empty,
  31. then checks for correctness of the tuple provided.
  32. """
  33. if version is None:
  34. from promise import VERSION
  35. return VERSION
  36. else:
  37. assert len(version) == 5
  38. assert version[3] in ("alpha", "beta", "rc", "final")
  39. return version
  40. def get_docs_version(version=None):
  41. version = get_complete_version(version)
  42. if version[3] != "final":
  43. return "dev"
  44. else:
  45. return "%d.%d" % version[:2]
  46. def get_git_changeset():
  47. """Returns a numeric identifier of the latest git changeset.
  48. The result is the UTC timestamp of the changeset in YYYYMMDDHHMMSS format.
  49. This value isn't guaranteed to be unique, but collisions are very unlikely,
  50. so it's sufficient for generating the development version numbers.
  51. """
  52. repo_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  53. try:
  54. git_log = subprocess.Popen(
  55. "git log --pretty=format:%ct --quiet -1 HEAD",
  56. stdout=subprocess.PIPE,
  57. stderr=subprocess.PIPE,
  58. shell=True,
  59. cwd=repo_dir,
  60. universal_newlines=True,
  61. )
  62. timestamp = git_log.communicate()[0]
  63. timestamp = datetime.datetime.utcfromtimestamp(int(timestamp))
  64. except Exception:
  65. return None
  66. return timestamp.strftime("%Y%m%d%H%M%S")