From 628528a258c8dbd44afbefb082de6b315f04d502 Mon Sep 17 00:00:00 2001
From: AbyssYuan0 <jiaxiangchang0@gmail.com>
Date: Tue, 16 Jan 2024 22:32:30 +0800
Subject: [PATCH] =?UTF-8?q?=E8=8E=B7=E5=8F=96=E6=96=87=E4=BB=B6=E5=A4=B9?=
 =?UTF-8?q?=E5=90=8D=E7=A7=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 __init__.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/__init__.py b/__init__.py
index 917a6e0..d9896cf 100644
--- a/__init__.py
+++ b/__init__.py
@@ -973,6 +973,25 @@ class GetUUID:
         return (result,)
 
 
+class GetDirName:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "dir_path": ("STRING", {"default": ""}),
+            },
+        }
+
+    CATEGORY = "badger"
+
+    RETURN_TYPES = ("STRING",)
+    FUNCTION = "get_dir_name"
+
+    def get_dir_name(self, dir_path):
+        folder_name = os.path.basename(os.path.dirname(dir_path))
+        return (folder_name,)
+
+
 NODE_CLASS_MAPPINGS = {
     "ImageOverlap-badger": ImageOverlap,
     "FloatToInt-badger": FloatToInt,
@@ -996,7 +1015,8 @@ NODE_CLASS_MAPPINGS = {
     "FindThickLinesFromCanny-badger": FindThickLinesFromCanny,
     "TrimTransparentEdges-badger": TrimTransparentEdges,
     "ExpandImageWithColor-badger": ExpandImageWithColor,
-    "GetUUID-badger": GetUUID
+    "GetUUID-badger": GetUUID,
+    "GetDirName-badger": GetDirName,
 
 }