help.py 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. import json
  2. import platform
  3. import sys
  4. from typing import Dict
  5. from . import __version__ as pyjwt_version
  6. try:
  7. import cryptography
  8. cryptography_version = cryptography.__version__
  9. except ModuleNotFoundError:
  10. cryptography_version = ""
  11. def info() -> Dict[str, Dict[str, str]]:
  12. """
  13. Generate information for a bug report.
  14. Based on the requests package help utility module.
  15. """
  16. try:
  17. platform_info = {
  18. "system": platform.system(),
  19. "release": platform.release(),
  20. }
  21. except OSError:
  22. platform_info = {"system": "Unknown", "release": "Unknown"}
  23. implementation = platform.python_implementation()
  24. if implementation == "CPython":
  25. implementation_version = platform.python_version()
  26. elif implementation == "PyPy":
  27. pypy_version_info = sys.pypy_version_info # type: ignore[attr-defined]
  28. implementation_version = (
  29. f"{pypy_version_info.major}."
  30. f"{pypy_version_info.minor}."
  31. f"{pypy_version_info.micro}"
  32. )
  33. if pypy_version_info.releaselevel != "final":
  34. implementation_version = "".join(
  35. [
  36. implementation_version,
  37. pypy_version_info.releaselevel,
  38. ]
  39. )
  40. else:
  41. implementation_version = "Unknown"
  42. return {
  43. "platform": platform_info,
  44. "implementation": {
  45. "name": implementation,
  46. "version": implementation_version,
  47. },
  48. "cryptography": {"version": cryptography_version},
  49. "pyjwt": {"version": pyjwt_version},
  50. }
  51. def main() -> None:
  52. """Pretty-print the bug information as JSON."""
  53. print(json.dumps(info(), sort_keys=True, indent=2))
  54. if __name__ == "__main__":
  55. main()