ruff n ruff

Author: jamadeo

src/ai_migrate/progress.py

@@ -1,9 +1,8 @@
 import asyncio
-import io
 import sys
 import itertools
 from enum import StrEnum
-from typing import Dict, Literal
+from typing import Dict
 import shutil
 
 
@@ -22,7 +21,7 @@ def __init__(self, line_limit):
 
     def write(self, s: str):
         self.lines.extend(s.removesuffix("\n").splitlines())
-        self.lines = self.lines[-self.line_limit:]
+        self.lines = self.lines[-self.line_limit :]
 
     def flush(self):
         pass
@@ -60,15 +59,23 @@ def render(self) -> str:
 
         name_part = f"{self.name}: "
         padding = terminal_width - len(name_part) - len(right_part)
-        logs = [
-            f"    {line}"[:terminal_width] for line in self.logger.getvalue().splitlines()
-        ] if self.status == Status.RUNNING else []
-        return f"\r{name_part}{' ' * max(0, padding)}{right_part}" + (f"\n{"\n".join(logs)}" if logs else "")
+        logs = (
+            [
+                f"    {line}"[:terminal_width]
+                for line in self.logger.getvalue().splitlines()
+            ]
+            if self.status == Status.RUNNING
+            else []
+        )
+        return f"\r{name_part}{' ' * max(0, padding)}{right_part}" + (
+            f"\n{'\n'.join(logs)}" if logs else ""
+        )
 
     def get_logger(self, header: str):
         self.logger.header = header
         return self.logger
 
+
 class StatusManager:
     def __init__(self):
         self.bars: Dict[str, StatusBar] = {}
@@ -101,7 +108,10 @@ async def render(self):
 
         self._last_render_lines = 0
         bars = list(self.bars.values())
-        bars = itertools.groupby(sorted(bars, key=lambda bar: (bar.status, bar.name)), key=lambda bar: bar.status)
+        bars = itertools.groupby(
+            sorted(bars, key=lambda bar: (bar.status, bar.name)),
+            key=lambda bar: bar.status,
+        )
         for status, bars in bars:
             bars = [*bars]
             if status == Status.WAITING:
@@ -113,7 +123,6 @@ async def render(self):
                     print(rendered)
                     self._last_render_lines += len(rendered.splitlines())
 
-
     async def mark_with_status(self, name: str, status: Status):
         async with self.lock:
             if name in self.bars:
@@ -137,4 +146,3 @@ async def stop(self):
 
     def get_logger(self, name: str, header: str = ""):
         return self.bars[name].get_logger(header=header)
-

src/ai_migrate/projects.py

@@ -87,6 +87,7 @@ def flush(self):
         for f in self.files:
             f.flush()
 
+
 async def run(
     project_dir: str,
     logs_dir: str | Path,
@@ -145,7 +146,10 @@ async def process_one_fileset(index, files: FileGroup, task_name: str):
         log_file.parent.mkdir(parents=True, exist_ok=True)
         log_buffer = open(log_file, "w")
 
-        logger = Tee(status_manager.get_logger(task_name, header=f"==> {log_file} <=="), log_buffer)
+        logger = Tee(
+            status_manager.get_logger(task_name, header=f"==> {log_file} <=="),
+            log_buffer,
+        )
 
         try:
             await run_migration(