add username, password to WebUIApi constructor. check controlnet when set_auth is called.

mix1009 · mix1009 · commit bcbd335256fd · 2023-06-01T15:03:11.000+09:00
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
 
 setup(
     name="webuiapi",
-    version="0.9.0",
+    version="0.9.1",
     description="Python API client for AUTOMATIC1111/stable-diffusion-webui",
     url="https://github.com/mix1009/sdwebuiapi",
     author="ChunKoo Park",
diff --git a/webuiapi/__init__.py b/webuiapi/__init__.py
@@ -12,7 +12,7 @@
     ControlNetUnit,
 )
 
-__version__ = "0.9.0"
+__version__ = "0.9.1"
 
 __all__ = [
     "__version__",
diff --git a/webuiapi/webuiapi.py b/webuiapi/webuiapi.py
@@ -138,6 +138,8 @@ def __init__(
         sampler="Euler a",
         steps=20,
         use_https=False,
+        username=None,
+        password=None,
     ):
         if baseurl is None:
             if use_https:
@@ -151,7 +153,10 @@ def __init__(
 
         self.session = requests.Session()
 
-        self.check_controlnet()
+        if username and password:
+            self.set_auth(username, password)
+        else:
+            self.check_controlnet()
 
     def check_controlnet(self):
         try:
@@ -162,6 +167,7 @@ def check_controlnet(self):
 
     def set_auth(self, username, password):
         self.session.auth = (username, password)
+        self.check_controlnet()
 
     def _to_api_result(self, response):
         if response.status_code != 200:
