39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import os
|
|
|
|
class FilePathExtractor:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"test": ("STRING", {"default": ""})
|
|
}
|
|
}
|
|
|
|
CATEGORY = "Custom Nodes"
|
|
RETURN_TYPES = ("STRING", "STRING", "STRING", "STRING")
|
|
RETURN_NAMES = ("File Name", "File Extension", "Folder Name", "Folder Path")
|
|
FUNCTION = "process"
|
|
|
|
@staticmethod
|
|
def IS_CHANGED():
|
|
return True
|
|
|
|
def process(self, test):
|
|
if os.path.isfile(test):
|
|
file_name = os.path.basename(test)
|
|
file_extension = os.path.splitext(test)[1][1:]
|
|
folder_path = os.path.dirname(test)
|
|
folder_name = os.path.basename(folder_path)
|
|
elif os.path.isdir(test):
|
|
file_name = ""
|
|
file_extension = ""
|
|
folder_path = os.path.abspath(test)
|
|
folder_name = os.path.basename(folder_path)
|
|
else:
|
|
file_name = ""
|
|
file_extension = ""
|
|
folder_path = ""
|
|
folder_name = ""
|
|
|
|
return file_name, file_extension, folder_name, folder_path
|