Upgrade to typed NamedTuples

This commit is contained in:
awsr
2026-03-28 16:57:59 -07:00
parent 1e9bef8d56
commit 8ef8074467
3 changed files with 42 additions and 17 deletions
+20 -3
View File
@@ -1,14 +1,19 @@
from __future__ import annotations
import os
import re
import sys
import time
from collections import namedtuple
from dataclasses import dataclass
from typing import TYPE_CHECKING, NamedTuple
import gradio as gr
from modules import paths, script_callbacks, extensions, script_loading, scripts_postprocessing, errors, timer
from modules.logger import log
from installer import control_extensions
if TYPE_CHECKING:
from types import ModuleType
AlwaysVisible = object()
time_component = {}
@@ -211,10 +216,22 @@ def basedir():
return current_basedir
ScriptFile = namedtuple("ScriptFile", ["basedir", "filename", "path", "priority"])
class ScriptFile(NamedTuple):
basedir: str
filename: str
path: str
priority: str
class ScriptClassData(NamedTuple):
script_class: type[Script] | type[scripts_postprocessing.ScriptPostprocessing]
path: str
basedir: str
module: ModuleType
scripts_data = []
postprocessing_scripts_data = []
ScriptClassData = namedtuple("ScriptClassData", ["script_class", "path", "basedir", "module"])
def list_scripts(scriptdirname, extension):