Jan Lindemann
5d1ba6e15a
Add new ruff rules and fix their fallout:
future-annotations = true
select = [ "TC", # type-checking import placement rules "FA", # future annotations rules ]This comprises:
- Streamline imports and exports in cmds.xxx.Cmd
- Import base class as "Base"- Export types Cmd and Parent via __all__- Move all types imported only for annotation below TYPE_CHECKING
- Use "from __future__ import annotations" all over the place
Signed-off-by: Jan Lindemann <jan@janware.com>
72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from ...lib.base import InputMode
|
|
from ...lib.log import NOTICE, log
|
|
from .Cmd import Cmd, Parent
|
|
from ...lib.base import Result
|
|
|
|
if TYPE_CHECKING:
|
|
from argparse import ArgumentParser, Namespace
|
|
|
|
class CmdCanonicalizeRemotes(Cmd): # export
|
|
|
|
def __rewrite_url(self, url: str) -> str:
|
|
return url.replace('/srv/git', '')
|
|
|
|
def __init__(self, parent: Parent) -> None:
|
|
super().__init__(
|
|
parent, 'canonicalize-remotes', help = 'Streamline janware Git remotes'
|
|
)
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
super().add_arguments(parser)
|
|
parser.add_argument(
|
|
'-n',
|
|
'--dry-run',
|
|
default = False,
|
|
action = 'store_true',
|
|
help = 'Only log what would be done',
|
|
)
|
|
|
|
async def _run(self, args: Namespace) -> None:
|
|
|
|
async def git(cmd: list[str], ro = False, throw = True) -> Result:
|
|
cmd = ['/usr/bin/git', *cmd]
|
|
log(NOTICE, f'-- {" ".join(cmd)}')
|
|
if ro or not args.dry_run:
|
|
return await self.app.exec_context.run(
|
|
cmd, cmd_input = InputMode.NonInteractive, throw = throw
|
|
)
|
|
return Result(b'', None, 0)
|
|
|
|
remotes: dict[str, dict[str, str | list[str]]] = {}
|
|
result = await git(['remote', '-v'], ro = True)
|
|
for line in result.stdout_str.splitlines():
|
|
name, url, fp = line.split()
|
|
remote = remotes.setdefault(name, {})
|
|
key = 'url' if fp == '(fetch)' else 'pushurl'
|
|
fpurls = remote.setdefault(key, [])
|
|
assert isinstance(fpurls, list)
|
|
fpurls.append(url)
|
|
for name, remote in remotes.items():
|
|
dirty_keys: set[str] = set()
|
|
for key, urls in remote.items():
|
|
for url in urls:
|
|
if url != self.__rewrite_url(url):
|
|
dirty_keys.add(key)
|
|
for key in dirty_keys:
|
|
urls = remote[key]
|
|
await git(
|
|
['config', '--unset-all', f'remote.{name}.{key}'], throw = False
|
|
)
|
|
for url in urls:
|
|
await git(
|
|
[
|
|
'config',
|
|
'--add',
|
|
f'remote.{name}.{key}',
|
|
self.__rewrite_url(url),
|
|
]
|
|
)
|