Added INPUT_TYPES class method to PublicGitRepo class to define required input types for the function, and modified validate_git_repo_url function to accept keyword arguments and return the git_url.

This commit is contained in:
Tasha Upchurch
2023-07-07 02:25:43 -04:00
parent b3510c8c36
commit 7748b4b142
+12 -9
View File
@@ -3,18 +3,21 @@ import giturlparse
class PublicGitRepo:
"""replaces the text in the given text string with a given other text at some key positions"""
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"url": ("STRING", {"multiline": False}), # Added url input
"email": ("STRING", {"multiline": False}) # Added email input
}
}
CATEGORY = "text"
RETURN_TYPES = ("STRING",)
FUNCTION = "validate_git_repo_url"
def validate_git_repo_url(self, text: str, replacement, key1, **kwargs):
"""
>>> PublicGitRepo().validate_git_repo_url("hello world", "universe", "world", git_url="https://github.com/example/repo.git")
'hello universe'
"""
git_url = kwargs.get("git_url")
def validate_git_repo_url(self,**kwargs):
git_url = kwargs.get("url", None)
parsed_url = giturlparse.parse(git_url)
if not parsed_url.valid:
raise ValueError("Invalid Git repository URL")
return (text.replace(key1, replacement),)
return (git_url,)