Files
Auto-GPT/scripts/check_requirements.py
2023-04-12 02:18:07 +03:00

28 lines
808 B
Python

import pkg_resources
import sys
def main():
requirements_file = sys.argv[1]
with open(requirements_file, 'r') as f:
required_packages = [line.strip().split('#')[0].strip() for line in f.readlines()]
installed_packages = [package.key for package in pkg_resources.working_set]
missing_packages = []
for package in required_packages:
if not package: # Skip empty lines
continue
package_name = package.strip().split('==')[0]
if package_name.lower() not in installed_packages:
missing_packages.append(package_name)
if missing_packages:
print('Missing packages:')
print(', '.join(missing_packages))
sys.exit(1)
else:
print('All packages are installed.')
if __name__ == '__main__':
main()