From 61ee9bde0befeef86b873a1164c30f86bbbcd41e Mon Sep 17 00:00:00 2001 From: Eyan Au Date: Wed, 24 Nov 2021 18:40:52 +0800 Subject: pref(script): use heapq to sort and add minor changes in check-aur-dependencies.py (#22) Signed-off-by: 67au --- utils/aur/check-aur-dependencies.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'utils/aur') diff --git a/utils/aur/check-aur-dependencies.py b/utils/aur/check-aur-dependencies.py index 6749f69..f3e9e31 100755 --- a/utils/aur/check-aur-dependencies.py +++ b/utils/aur/check-aur-dependencies.py @@ -18,6 +18,7 @@ # along with this program. If not, see . import argparse import asyncio +import heapq import logging import re from pathlib import Path @@ -34,7 +35,7 @@ async def check_pkg(session: aiohttp.ClientSession, num: int, pkg: str) -> Tuple async def main() -> int: - parser = argparse.ArgumentParser(description='check-aur.py') + parser = argparse.ArgumentParser(description='check-aur-dependencies.py') parser.add_argument('paths', metavar='sub_path', type=str, nargs='*', help='sub-path for repo root') args = parser.parse_args() @@ -60,13 +61,13 @@ async def main() -> int: for task in asyncio.as_completed(tasks): n, r = await task if r: - yay_dep_nums.append(n) + heapq.heappush(yay_dep_nums, n) yaydeps_dir = repo_dir.parent.joinpath('.yaydeps') if len(yay_dep_nums) > 0: async with aiofiles.open(yaydeps_dir.joinpath(repo_dir.stem), 'w', encoding='utf-8') as f: - await f.write('\n'.join([*(deps[n] for n in sorted(yay_dep_nums)), ''])) + await f.write('\n'.join([*(deps[heapq.heappop(yay_dep_nums)] for _ in range(len(yay_dep_nums))), ''])) return 0 @@ -74,5 +75,4 @@ async def main() -> int: if __name__ == '__main__': logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(funcName)s - %(lineno)d - %(message)s') - loop = asyncio.get_event_loop() - exit(loop.run_until_complete(main())) + exit(asyncio.run(main())) -- cgit v1.3.1