97 lines
4.0 KiB
Python
97 lines
4.0 KiB
Python
"""
|
|
@author: Manny Gonzalez
|
|
@title: 🐯 YFG Comical Nodes
|
|
@nickname: 🐯 YFG Comical Nodes
|
|
@description: Utility custom nodes for special effects, image manipulation and quality of life tools.
|
|
"""
|
|
|
|
## Based on original code by Jordan Thompson (WASasquatch) https://github.com/WASasquatch/was-node-suite-comfyui ##
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to
|
|
# deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
# and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
# THE SOFTWARE.
|
|
|
|
import requests
|
|
import json
|
|
import hashlib
|
|
|
|
class RandomOrgTrueRandomNumber:
|
|
"""True Random Number Generator via random.org"""
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"api_key": ("STRING", {"default": "00000000-0000-0000-0000-000000000000", "multiline": False}),
|
|
"minimum": ("FLOAT", {"default": 0, "min": -9223372036854775808, "max": 9223372036854775807}),
|
|
"maximum": ("FLOAT", {"default": 10, "min": -9223372036854775808, "max": 9223372036854775807}),
|
|
"mode": (["random", "fixed"],),
|
|
}
|
|
}
|
|
|
|
# Emit Number, Float, and Integer outputs to match dependent nodes
|
|
RETURN_TYPES = ("NUMBER", "FLOAT", "INT")
|
|
FUNCTION = "return_true_random_number"
|
|
CATEGORY = "🐯 YFG/🔢 Numbers"
|
|
|
|
def return_true_random_number(self, api_key, minimum, maximum, mode="random"):
|
|
"""Generate one true random integer and return as NUMBER, FLOAT, and INT"""
|
|
nums = self.get_random_numbers(
|
|
api_key=api_key,
|
|
amount=1,
|
|
minimum=minimum,
|
|
maximum=maximum,
|
|
mode=mode
|
|
)
|
|
number = nums[0]
|
|
return (number, float(number), int(number))
|
|
|
|
def get_random_numbers(self, api_key, amount=1, minimum=0, maximum=10, mode="random"):
|
|
"""Fetches `amount` true random integers from random.org."""
|
|
if not api_key or api_key == "00000000-0000-0000-0000-000000000000":
|
|
print("Error: A valid RANDOM.ORG API key is required.")
|
|
return [0]
|
|
|
|
payload = {
|
|
"jsonrpc": "2.0",
|
|
"method": "generateIntegers",
|
|
"params": {
|
|
"apiKey": api_key,
|
|
"n": amount,
|
|
"min": int(minimum),
|
|
"max": int(maximum),
|
|
"replacement": True,
|
|
"base": 10
|
|
},
|
|
"id": 1
|
|
}
|
|
|
|
response = requests.post(
|
|
"https://api.random.org/json-rpc/2/invoke",
|
|
headers={"Content-Type": "application/json"},
|
|
data=json.dumps(payload)
|
|
)
|
|
if response.status_code == 200:
|
|
data = response.json()
|
|
if "result" in data and "random" in data["result"]:
|
|
return data["result"]["random"]["data"]
|
|
return [0]
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, api_key, mode, **kwargs):
|
|
"""Used by ComfyUI to detect when `mode=='fixed'` and keep output constant"""
|
|
m = hashlib.sha256()
|
|
m.update(api_key.encode('utf-8'))
|
|
if mode == 'fixed':
|
|
return m.hexdigest()
|
|
return float('nan')
|