tubearchivist/version_check.py

118 lines
3.4 KiB
Python
Raw Normal View History

#!/usr/bin/env python
""" check requirements.txt for outdated packages """
import sys
import pathlib
import requests
class Requirements:
""" handle requirements.txt """
FILE_PATH = 'tubearchivist/requirements.txt'
LOCK = '/tmp/tubearchivist-requirements.lock'
def __init__(self):
self.exists = self.checked_today()
self.all_requirements = False
self.all_updates = False
def checked_today(self):
""" skip requirements check when lock file exists """
exists = pathlib.Path(self.LOCK).exists()
return exists
def look_for_updates(self):
""" look through requirements and check for updates """
self.all_requirements = self.get_dependencies()
self.all_updates = self.check_packages()
def get_dependencies(self):
""" read out requirements.txt """
all_requirements = []
with open(self.FILE_PATH, 'r', encoding='utf-8') as f:
dependencies = f.readlines()
for dependency in dependencies:
package, version = dependency.split('==')
all_requirements.append((package, version.strip()))
all_requirements.sort(key = lambda x: x[0].lower())
return all_requirements
def check_packages(self):
""" compare installed with remote version """
total = len(self.all_requirements)
print(f'checking versions for {total} packages...')
all_updates = {}
for dependency in self.all_requirements:
package, version_installed = dependency
url = f'https://pypi.org/pypi/{package}/json'
response = requests.get(url).json()
version_remote = response['info']['version']
homepage = response['info']['home_page']
if version_remote != version_installed:
to_update = {
package: {
"from": version_installed,
"to": version_remote
}
}
all_updates.update(to_update)
message = (f'update {package} {version_installed}' +
f'==> {version_remote}\n {homepage}')
print(message)
if not all_updates:
print('no updates found')
# remember that
pathlib.Path(self.LOCK).touch()
return all_updates
def apply_updates(self):
""" update requirements.txt file with new versions """
to_write = []
for requirement in self.all_requirements:
package, old_version = requirement
if package in self.all_updates.keys():
package_version = self.all_updates[package]['to']
else:
package_version = old_version
to_write.append(f'{package}=={package_version}\n')
with open(self.FILE_PATH, 'w', encoding='utf-8') as f:
f.writelines(to_write)
print('requirements.txt updates')
def main():
""" main to check for updates """
handler = Requirements()
if handler.exists:
return
handler.look_for_updates()
if handler.all_updates:
input_response = input('\nupdate requirements.txt? [y/n] ')
if input_response == 'y':
handler.apply_updates()
else:
print('cancle update...')
sys.exit(1)
if __name__ == "__main__":
main()