aquafire/paginate_tags.py

paginate_tags.py · 2.1 KiB · Python Raw

async def get_artifacts(
    client: HarborAsyncClient,
    project_name: str,
    repository_name: str,
    page_number: int,
    page_size: int,
) -> List[str]:
    # номер страницы с артефактами в Harbor
    harbor_page_number = 1
    # номер страницы с тегами (RIM)
    rim_page_number = page_number  # используем входной page_number
    # буфер под все собранные теги
    all_tags = []
    
    # собираем ВСЕ теги из Harbor
    while True:
        # считываем из Harbor артефакты
        artifacts = await client.get_artifacts(
            project_name=project_name,
            repository_name=repository_name,
            page=harbor_page_number,
            page_size=10,
            limit=10,
        )

print(f"Harbor page number: {harbor_page_number}")

# обрабатываем артефакты
        for artifact in artifacts:
            # проверяем есть ли тэги
            if artifact.tags:
                # перебираем все ссылки на тэги
                for tag_reference in artifact.tags:
                    # путь до артефакта
                    artifact_name = (
                        f"{project_name}/{repository_name}:{tag_reference.name}"
                    )
                    all_tags.append(artifact_name)

# если нет артефактов - выходим
        if not artifacts:
            break

# увеличичиваем номер страницы в harbor
        harbor_page_number += 1
    
    # применяем пагинацию к собранным тегам
    start_idx = (rim_page_number - 1) * page_size
    end_idx = start_idx + page_size
    
    # проверяем валидность запрошенной страницы
    if start_idx >= len(all_tags):
        return []  # несуществующая страница - пустой результат
    
    return all_tags[start_idx:end_idx]

1	async def get_artifacts(
2	client: HarborAsyncClient,
3	project_name: str,
4	repository_name: str,
5	page_number: int,
6	page_size: int,
7	) -> List[str]:
8	# номер страницы с артефактами в Harbor
9	harbor_page_number = 1
10	# номер страницы с тегами (RIM)
11	rim_page_number = page_number # используем входной page_number
12	# буфер под все собранные теги
13	all_tags = []
14
15	# собираем ВСЕ теги из Harbor
16	while True:
17	# считываем из Harbor артефакты
18	artifacts = await client.get_artifacts(
19	project_name=project_name,
20	repository_name=repository_name,
21	page=harbor_page_number,
22	page_size=10,
23	limit=10,
24	)
25
26	print(f"Harbor page number: {harbor_page_number}")
27
28	# обрабатываем артефакты
29	for artifact in artifacts:
30	# проверяем есть ли тэги
31	if artifact.tags:
32	# перебираем все ссылки на тэги
33	for tag_reference in artifact.tags:
34	# путь до артефакта
35	artifact_name = (
36	f"{project_name}/{repository_name}:{tag_reference.name}"
37	)
38	all_tags.append(artifact_name)
39
40	# если нет артефактов - выходим
41	if not artifacts:
42	break
43
44	# увеличичиваем номер страницы в harbor
45	harbor_page_number += 1
46
47	# применяем пагинацию к собранным тегам
48	start_idx = (rim_page_number - 1) * page_size
49	end_idx = start_idx + page_size
50
51	# проверяем валидность запрошенной страницы
52	if start_idx >= len(all_tags):
53	return [] # несуществующая страница - пустой результат
54
55	return all_tags[start_idx:end_idx]