mirror of
https://github.com/tubearchivist/tubearchivist.git
synced 2024-12-26 19:50:12 +00:00
118 lines
3.4 KiB
Python
Executable File
118 lines
3.4 KiB
Python
Executable File
#!/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()
|