119 lines
5.2 KiB
Python
119 lines
5.2 KiB
Python
from typing import Dict, Type, TypeVar
|
|
from .modules.cda.cda_json_create import CDAJSONCreate
|
|
from .modules.matplotlib.mpl_bar import MPLBar
|
|
from .modules.matplotlib.mpl_line import MPLLine
|
|
from .modules.matplotlib.mpl_scatter import MPLScatter
|
|
from .modules.pandas_wrapper.pandas_columns import PandasColumns
|
|
from .modules.pandas_wrapper.pandas_create import PandasCreate
|
|
from .modules.pandas_wrapper.pandas_create_with_index import PandasCreateWithIndex
|
|
from .modules.pandas_wrapper.pandas_dropna import PandasDropNA
|
|
from .modules.pandas_wrapper.pandas_group_by import PandasGroupBy
|
|
from .modules.pandas_wrapper.pandas_head import PandasHead
|
|
from .modules.pandas_wrapper.pandas_iloc_row_series import PandasIlocRowSeries
|
|
from .modules.pandas_wrapper.pandas_iloc_rows_dataframe import PandasIlocRowsDataFrame
|
|
from .modules.pandas_wrapper.pandas_index import PandasIndex
|
|
from .modules.pandas_wrapper.pandas_isna import PandasIsNA
|
|
from .modules.pandas_wrapper.pandas_join import PandasJoin
|
|
from .modules.pandas_wrapper.pandas_load_csv import PandasLoadCSV
|
|
from .modules.pandas_wrapper.pandas_load_csv_with_encoding import PandasLoadCSVWithEncoding
|
|
from .modules.pandas_wrapper.pandas_load_csv_with_index import PandasLoadCSVWithIndex
|
|
from .modules.pandas_wrapper.pandas_load_json import PandasLoadJSON
|
|
from .modules.pandas_wrapper.pandas_loc_cell_str import PandasLocCellStr
|
|
from .modules.pandas_wrapper.pandas_loc_row_series import PandasLocRowSeries
|
|
from .modules.pandas_wrapper.pandas_save_csv import PandasSaveCSV
|
|
from .modules.pandas_wrapper.pandas_save_json import PandasSaveJSON
|
|
from .modules.pandas_wrapper.pandas_select_columns import PandasSelectColumns
|
|
from .modules.pandas_wrapper.pandas_select_rows import PandasSelectRows
|
|
from .modules.pandas_wrapper.pandas_series_to_string import PandasSeriesToString
|
|
from .modules.pandas_wrapper.pandas_show_dataframe import PandasShowDataFrame
|
|
from .modules.pandas_wrapper.pandas_show_text import PandasShowText
|
|
from .modules.pandas_wrapper.pandas_sort import PandasSort
|
|
from .modules.pandas_wrapper.pandas_summary import PandasSummary
|
|
from .modules.pandas_wrapper.pandas_to_string import PandasToString
|
|
|
|
T = TypeVar("T")
|
|
|
|
NODE_CLASS_MAPPINGS: Dict[str, Type[T]] = {
|
|
"CDAJSONCreate": CDAJSONCreate,
|
|
"MPLBar": MPLBar,
|
|
"MPLLine": MPLLine,
|
|
"MPLScatter": MPLScatter,
|
|
"PandasColumns": PandasColumns,
|
|
"PandasCreate": PandasCreate,
|
|
"PandasCreateWithIndex": PandasCreateWithIndex,
|
|
"PandasDropNA": PandasDropNA,
|
|
"PandasGroupBy": PandasGroupBy,
|
|
"PandasHead": PandasHead,
|
|
"PandasIlocRowSeries": PandasIlocRowSeries,
|
|
"PandasIlocRowsDataFrame": PandasIlocRowsDataFrame,
|
|
"PandasIndex": PandasIndex,
|
|
"PandasIsNA": PandasIsNA,
|
|
"PandasJoin": PandasJoin,
|
|
"PandasLoadCSV": PandasLoadCSV,
|
|
"PandasLoadCSVWithEncoding": PandasLoadCSVWithEncoding,
|
|
"PandasLoadCSVWithIndex": PandasLoadCSVWithIndex,
|
|
"PandasLoadJSON": PandasLoadJSON,
|
|
"PandasLocCellStr": PandasLocCellStr,
|
|
"PandasLocRowSeries": PandasLocRowSeries,
|
|
"PandasSaveCSV": PandasSaveCSV,
|
|
"PandasSaveJSON": PandasSaveJSON,
|
|
"PandasSelectColumns": PandasSelectColumns,
|
|
"PandasSelectRows": PandasSelectRows,
|
|
"PandasSeriesToString": PandasSeriesToString,
|
|
"PandasShowDataFrame": PandasShowDataFrame,
|
|
"PandasShowText": PandasShowText,
|
|
"PandasSort": PandasSort,
|
|
"PandasSummary": PandasSummary,
|
|
"PandasToString": PandasToString,
|
|
}
|
|
"""
|
|
NODE_CLASS_MAPPINGS (Dict[str, Type[T]]):
|
|
A dictionary mapping node names to their corresponding class implementations.
|
|
"""
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS: Dict[str, str] = {
|
|
"CDAJSONCreate": "CDA JSON Create",
|
|
"MPLBar": "MPL Bar Chart",
|
|
"MPLLine": "MPL Line Plot",
|
|
"MPLScatter": "MPL Scatter Plot",
|
|
"PandasColumns": "Pandas Columns",
|
|
"PandasCreate": "Pandas Create",
|
|
"PandasCreateWithIndex": "Pandas Create With Index",
|
|
"PandasDropNA": "Pandas Drop NA",
|
|
"PandasGroupBy": "Pandas Group By",
|
|
"PandasHead": "Pandas Head",
|
|
"PandasIlocRowSeries": "Pandas Iloc Row Series",
|
|
"PandasIlocRowsDataFrame": "Pandas Iloc Rows DataFrame",
|
|
"PandasIndex": "Pandas Index",
|
|
"PandasIsNA": "Pandas Is NA",
|
|
"PandasJoin": "Pandas Join",
|
|
"PandasLoadCSV": "Pandas Load CSV",
|
|
"PandasLoadCSVWithEncoding": "Pandas Load CSV With Encoding",
|
|
"PandasLoadCSVWithIndex": "Pandas Load CSV With Index",
|
|
"PandasLoadJSON": "Pandas Load JSON",
|
|
"PandasLocCellStr": "Pandas Loc Cell Str",
|
|
"PandasLocRowSeries": "Pandas Loc Row Series",
|
|
"PandasSaveCSV": "Pandas Save CSV",
|
|
"PandasSaveJSON": "Pandas Save JSON",
|
|
"PandasSelectColumns": "Pandas Select Columns",
|
|
"PandasSelectRows": "Pandas Select Rows",
|
|
"PandasSeriesToString": "Pandas Series To String",
|
|
"PandasShowDataFrame": "Pandas Show DataFrame",
|
|
"PandasShowText": "Pandas Show Text",
|
|
"PandasSort": "Pandas Sort",
|
|
"PandasSummary": "Pandas Summary",
|
|
"PandasToString": "Pandas To String",
|
|
}
|
|
"""
|
|
NODE_DISPLAY_NAME_MAPPINGS (Dict[str, str]):
|
|
A dictionary mapping node names to user-friendly display names.
|
|
"""
|
|
|
|
"""
|
|
Below two lines were taken from:
|
|
https://github.com/pythongosssss/ComfyUI-Custom-Scripts/blob/main/__init__.py
|
|
See credit/credit.md for the full license.
|
|
"""
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|