forked from lazniak/videoclipgenerator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathnumber_padder.py
44 lines (38 loc) · 1.35 KB
/
number_padder.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# number_padder.py
class NumberPadder:
def __init__(self):
self.type = "NUMBER_PADDER"
self.output_node = True
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"number": ("INT", {
"default": 42,
"min": 0, # Minimalna wartość
"max": 999999999 # Maksymalna wartość - można dostosować
}),
"padding_length": ("INT", {
"default": 4,
"min": 1, # Minimalna długość paddingu
"max": 20 # Maksymalna długość paddingu
}),
},
}
RETURN_TYPES = ("STRING",)
FUNCTION = "pad_number"
CATEGORY = "formatting"
def pad_number(self, number, padding_length):
try:
# Konwertuje liczbę na string i dodaje padding
padded_number = str(number).zfill(padding_length)
return (padded_number,)
except Exception as e:
print(f"Błąd podczas formatowania liczby: {str(e)}")
return (str(number),) # W razie błędu zwraca oryginalną liczbę jako string
NODE_CLASS_MAPPINGS = {
"NumberPadder": NumberPadder
}
NODE_DISPLAY_NAME_MAPPINGS = {
"NumberPadder": "Number Padder"
}