PersonalPublic/different
1
0
Fork 0
mirror of https://github.com/Balshgit/different synced 2025-09-12 01:20:43 +03:00
Code Issues Projects Releases Wiki Activity

add lock for pretty print

Browse Source
This commit is contained in:
Dmitry Afanasyev 2023-10-14 18:18:57 +03:00
parent 62c303916c
commit 9dde5371b4
1 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
dockerhub.py
View File

@ -2,7 +2,7 @@ import asyncio
import re import re
import sys import sys
from logging import Logger from logging import Logger
from multiprocessing import Process from multiprocessing import Process, Lock
from typing import Any from typing import Any
import httpx import httpx
@ -155,12 +155,14 @@ class DockerHubScanner:
return services_tags return services_tags
def print_data(self, service_name: str, service_component: dict[str, str]) -> None: def print_data(self, service_name: str, service_component: dict[str, str], lock: Lock) -> None:
component_name = service_component['name'] component_name = service_component['name']
component_version = service_component['version'] component_version = service_component['version']
data = self.get_data(service_name, service_component) data = self.get_data(service_name, service_component)
with lock:
print( print(
f"Service: {colored(service_name, color='light_grey')}", f"Service: {colored(service_name, color='light_grey')}",
f"\nComponent: {colored(component_name, color='light_blue')}", f"\nComponent: {colored(component_name, color='light_blue')}",
@ -201,7 +203,9 @@ class DockerHubScanner:
if __name__ == '__main__': if __name__ == '__main__':
print('Services'.center(50, '-'), '\n') print(colored('Services'.center(50, '-', ), color='white'), '\n')
global_lock = Lock()
dockerhub_scanner = DockerHubScanner() dockerhub_scanner = DockerHubScanner()
processes = [] processes = []
@ -212,10 +216,13 @@ if __name__ == '__main__':
continue continue
process = Process( process = Process(
target=dockerhub_scanner.print_data, target=dockerhub_scanner.print_data,
kwargs={'service_name': service, 'service_component': component} kwargs={'service_name': service, 'service_component': component, 'lock': global_lock}
) )
processes.append(process) processes.append(process)
process.start() process.start()
for process in processes: for process in processes:
process.join() process.join()
print(colored("All jobs done", color='white'))