···

       
1
       
1
       
 
       
#! /usr/bin/env nix-shell

     

       
2
       
2
       
 
       
#! nix-shell -i "python3 -I" -p "python3.withPackages(p: with p; [ aiohttp rich structlog ])"

     

       
3
       
3
       
 
       


     

       
4
       
4
       
-
       
from argparse import ArgumentParser

     

       
4
       
4
       
+
       
from argparse import ArgumentParser, Namespace

     

       
5
       
5
       
 
       
from collections import defaultdict

     

       
6
       
6
       
+
       
from collections.abc import Mapping, Sequence

     

       
6
       
7
       
 
       
from enum import IntEnum

     

       
7
       
8
       
 
       
from http import HTTPStatus

     

       
8
       
9
       
 
       
from pathlib import Path

     

       
10
       
10
       
+
       
from typing import Optional

     

       
9
       
11
       
 
       
import asyncio, json, logging

     

       
10
       
12
       
 
       


     

       
11
       
13
       
 
       
import aiohttp, structlog

     
···

       
24
       
26
       
 
       
    HTTPStatus.NOT_FOUND,

     

       
25
       
27
       
 
       
))

     

       
26
       
28
       
 
       


     

       
27
       
27
       
-
       
async def check(session, manpage: str, url: str) -> HTTPStatus:

     

       
29
       
29
       
+
       
async def check(session: aiohttp.ClientSession, manpage: str, url: str) -> HTTPStatus:

     

       
28
       
30
       
 
       
    with log_context(manpage=manpage, url=url):

     

       
29
       
31
       
 
       
        logger.debug("Checking")

     

       
30
       
32
       
 
       
        async with session.head(url) as resp:

     
···

       
41
       
43
       
 
       


     

       
42
       
44
       
 
       
            return st

     

       
43
       
45
       
 
       


     

       
44
       
44
       
-
       
async def main(urls_path):

     

       
46
       
46
       
+
       
async def main(urls_path: Path) -> Mapping[HTTPStatus, int]:

     

       
45
       
47
       
 
       
    logger.info(f"Parsing {urls_path}")

     

       
46
       
48
       
 
       
    with urls_path.open() as urls_file:

     

       
47
       
49
       
 
       
        urls = json.load(urls_file)

     

       
48
       
50
       
 
       


     

       
49
       
49
       
-
       
    count = defaultdict(lambda: 0)

     

       
51
       
51
       
+
       
    count: defaultdict[HTTPStatus, int] = defaultdict(lambda: 0)

     

       
50
       
52
       
 
       


     

       
51
       
53
       
 
       
    logger.info(f"Checking URLs from {urls_path}")

     

       
52
       
54
       
 
       
    async with aiohttp.ClientSession() as session:

     
···

       
65
       
67
       
 
       
    return count

     

       
66
       
68
       
 
       


     

       
67
       
69
       
 
       


     

       
68
       
68
       
-
       
def parse_args(args=None):

     

       
70
       
70
       
+
       
def parse_args(args: Optional[Sequence[str]] = None) -> Namespace:

     

       
69
       
71
       
 
       
    parser = ArgumentParser(

     

       
70
       
72
       
 
       
        prog = 'check-manpage-urls',

     

       
71
       
73
       
 
       
        description = 'Check the validity of the manpage URLs linked in the nixpkgs manual',