check.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. def check_requirements(installed_dists):
  2. missing_reqs_dict = {}
  3. incompatible_reqs_dict = {}
  4. for dist in installed_dists:
  5. key = '%s==%s' % (dist.project_name, dist.version)
  6. missing_reqs = list(get_missing_reqs(dist, installed_dists))
  7. if missing_reqs:
  8. missing_reqs_dict[key] = missing_reqs
  9. incompatible_reqs = list(get_incompatible_reqs(
  10. dist, installed_dists))
  11. if incompatible_reqs:
  12. incompatible_reqs_dict[key] = incompatible_reqs
  13. return (missing_reqs_dict, incompatible_reqs_dict)
  14. def get_missing_reqs(dist, installed_dists):
  15. """Return all of the requirements of `dist` that aren't present in
  16. `installed_dists`.
  17. """
  18. installed_names = set(d.project_name.lower() for d in installed_dists)
  19. missing_requirements = set()
  20. for requirement in dist.requires():
  21. if requirement.project_name.lower() not in installed_names:
  22. missing_requirements.add(requirement)
  23. yield requirement
  24. def get_incompatible_reqs(dist, installed_dists):
  25. """Return all of the requirements of `dist` that are present in
  26. `installed_dists`, but have incompatible versions.
  27. """
  28. installed_dists_by_name = {}
  29. for installed_dist in installed_dists:
  30. installed_dists_by_name[installed_dist.project_name] = installed_dist
  31. for requirement in dist.requires():
  32. present_dist = installed_dists_by_name.get(requirement.project_name)
  33. if present_dist and present_dist not in requirement:
  34. yield (requirement, present_dist)