mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-10 06:10:50 +08:00
1b2ea61345
- Run comfyui workflows directly inside other python applications using EmbeddedComfyClient. - Optional telemetry in prompts and models using anonymity preserving Plausible self-hosted or hosted. - Better OpenAPI schema - Basic support for distributed ComfyUI backends. Limitations: no progress reporting, no easy way to start your own distributed backend, requires RabbitMQ as a message broker.
271 lines
11 KiB
Python
271 lines
11 KiB
Python
# coding: utf-8
|
|
|
|
"""
|
|
comfyui
|
|
No description provided (generated by Openapi JSON Schema Generator https://github.com/openapi-json-schema-tools/openapi-json-schema-generator) # noqa: E501
|
|
The version of the OpenAPI document: 0.0.1
|
|
Generated by: https://github.com/openapi-json-schema-tools/openapi-json-schema-generator
|
|
"""
|
|
|
|
import logging
|
|
import ssl
|
|
from urllib.parse import urlencode
|
|
import typing
|
|
|
|
import certifi # type: ignore[import]
|
|
import urllib3
|
|
from urllib3 import fields
|
|
from urllib3 import exceptions as urllib3_exceptions
|
|
from urllib3._collections import HTTPHeaderDict
|
|
|
|
from comfy.api import exceptions
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
_TYPE_FIELD_VALUE = typing.Union[str, bytes]
|
|
|
|
|
|
class RequestField(fields.RequestField):
|
|
def __init__(
|
|
self,
|
|
name: str,
|
|
data: _TYPE_FIELD_VALUE,
|
|
filename: typing.Optional[str] = None,
|
|
headers: typing.Optional[typing.Mapping[str, typing.Union[str, None]]] = None,
|
|
header_formatter: typing.Optional[typing.Callable[[str, _TYPE_FIELD_VALUE], str]] = None,
|
|
):
|
|
super().__init__(name, data, filename, headers, header_formatter) # type: ignore
|
|
|
|
def __eq__(self, other):
|
|
if not isinstance(other, fields.RequestField):
|
|
return False
|
|
return self.__dict__ == other.__dict__
|
|
|
|
|
|
class RESTClientObject(object):
|
|
|
|
def __init__(self, configuration, pools_size=4, maxsize=None):
|
|
# urllib3.PoolManager will pass all kw parameters to connectionpool
|
|
# https://github.com/shazow/urllib3/blob/f9409436f83aeb79fbaf090181cd81b784f1b8ce/urllib3/poolmanager.py#L75 # noqa: E501
|
|
# https://github.com/shazow/urllib3/blob/f9409436f83aeb79fbaf090181cd81b784f1b8ce/urllib3/connectionpool.py#L680 # noqa: E501
|
|
# maxsize is the number of requests to host that are allowed in parallel # noqa: E501
|
|
# Custom SSL certificates and client certificates: http://urllib3.readthedocs.io/en/latest/advanced-usage.html # noqa: E501
|
|
|
|
# cert_reqs
|
|
if configuration.verify_ssl:
|
|
cert_reqs = ssl.CERT_REQUIRED
|
|
else:
|
|
cert_reqs = ssl.CERT_NONE
|
|
|
|
# ca_certs
|
|
if configuration.ssl_ca_cert:
|
|
ca_certs = configuration.ssl_ca_cert
|
|
else:
|
|
# if not set certificate file, use Mozilla's root certificates.
|
|
ca_certs = certifi.where()
|
|
|
|
addition_pool_args = {}
|
|
if configuration.assert_hostname is not None:
|
|
addition_pool_args['assert_hostname'] = configuration.assert_hostname # noqa: E501
|
|
|
|
if configuration.retries is not None:
|
|
addition_pool_args['retries'] = configuration.retries
|
|
|
|
if configuration.socket_options is not None:
|
|
addition_pool_args['socket_options'] = configuration.socket_options
|
|
|
|
if maxsize is None:
|
|
if configuration.connection_pool_maxsize is not None:
|
|
maxsize = configuration.connection_pool_maxsize
|
|
else:
|
|
maxsize = 4
|
|
|
|
# https pool manager
|
|
if configuration.proxy:
|
|
self.pool_manager = urllib3.ProxyManager(
|
|
num_pools=pools_size,
|
|
maxsize=maxsize,
|
|
cert_reqs=cert_reqs,
|
|
ca_certs=ca_certs,
|
|
cert_file=configuration.cert_file,
|
|
key_file=configuration.key_file,
|
|
proxy_url=configuration.proxy,
|
|
proxy_headers=configuration.proxy_headers,
|
|
**addition_pool_args
|
|
)
|
|
else:
|
|
self.pool_manager = urllib3.PoolManager(
|
|
num_pools=pools_size,
|
|
maxsize=maxsize,
|
|
cert_reqs=cert_reqs,
|
|
ca_certs=ca_certs,
|
|
cert_file=configuration.cert_file,
|
|
key_file=configuration.key_file,
|
|
**addition_pool_args
|
|
)
|
|
|
|
def request(
|
|
self,
|
|
method: str,
|
|
url: str,
|
|
headers: typing.Optional[HTTPHeaderDict] = None,
|
|
fields: typing.Optional[typing.Tuple[RequestField, ...]] = None,
|
|
body: typing.Optional[typing.Union[str, bytes]] = None,
|
|
stream: bool = False,
|
|
timeout: typing.Optional[typing.Union[int, float, typing.Tuple]] = None,
|
|
) -> urllib3.HTTPResponse:
|
|
"""Perform requests.
|
|
|
|
:param method: http request method
|
|
:param url: http request url
|
|
:param headers: http request headers
|
|
:param body: request body, for other types
|
|
:param fields: request parameters for
|
|
`application/x-www-form-urlencoded`
|
|
or `multipart/form-data`
|
|
:param stream: if True, the urllib3.HTTPResponse object will
|
|
be returned without reading/decoding response
|
|
data. Default is False.
|
|
:param timeout: timeout setting for this request. If one
|
|
number provided, it will be total request
|
|
timeout. It can also be a pair (tuple) of
|
|
(connection, read) timeouts.
|
|
"""
|
|
assert method in ['GET', 'HEAD', 'DELETE', 'POST', 'PUT',
|
|
'PATCH', 'OPTIONS']
|
|
|
|
if fields and body:
|
|
raise exceptions.ApiValueError(
|
|
"body parameter cannot be used with fields parameter."
|
|
)
|
|
|
|
headers = headers or HTTPHeaderDict()
|
|
|
|
used_timeout: typing.Optional[urllib3.Timeout] = None
|
|
if timeout:
|
|
if isinstance(timeout, (int, float)):
|
|
used_timeout = urllib3.Timeout(total=timeout)
|
|
elif (isinstance(timeout, tuple) and
|
|
len(timeout) == 2):
|
|
used_timeout = urllib3.Timeout(connect=timeout[0], read=timeout[1])
|
|
|
|
try:
|
|
# For `POST`, `PUT`, `PATCH`, `OPTIONS`, `DELETE`
|
|
if method in {'POST', 'PUT', 'PATCH', 'OPTIONS', 'DELETE'}:
|
|
if 'Content-Type' not in headers and body is None:
|
|
r = self.pool_manager.request(
|
|
method,
|
|
url,
|
|
preload_content=not stream,
|
|
timeout=used_timeout,
|
|
headers=headers
|
|
)
|
|
elif headers['Content-Type'] == 'application/x-www-form-urlencoded': # noqa: E501
|
|
r = self.pool_manager.request(
|
|
method, url,
|
|
body=body,
|
|
encode_multipart=False,
|
|
preload_content=not stream,
|
|
timeout=used_timeout,
|
|
headers=headers)
|
|
elif headers['Content-Type'] == 'multipart/form-data':
|
|
# must del headers['Content-Type'], or the correct
|
|
# Content-Type which generated by urllib3 will be
|
|
# overwritten.
|
|
del headers['Content-Type']
|
|
r = self.pool_manager.request(
|
|
method, url,
|
|
fields=fields,
|
|
encode_multipart=True,
|
|
preload_content=not stream,
|
|
timeout=used_timeout,
|
|
headers=headers)
|
|
# Pass a `string` parameter directly in the body to support
|
|
# other content types than Json when `body` argument is
|
|
# provided in serialized form
|
|
elif isinstance(body, str) or isinstance(body, bytes):
|
|
request_body = body
|
|
r = self.pool_manager.request(
|
|
method, url,
|
|
body=request_body,
|
|
preload_content=not stream,
|
|
timeout=used_timeout,
|
|
headers=headers)
|
|
else:
|
|
# Cannot generate the request from given parameters
|
|
msg = """Cannot prepare a request message for provided
|
|
arguments. Please check that your arguments match
|
|
declared content type."""
|
|
raise exceptions.ApiException(status=0, reason=msg)
|
|
# For `GET`, `HEAD`
|
|
else:
|
|
r = self.pool_manager.request(method, url,
|
|
preload_content=not stream,
|
|
timeout=used_timeout,
|
|
headers=headers)
|
|
except urllib3_exceptions.SSLError as e:
|
|
msg = "{0}\n{1}".format(type(e).__name__, str(e))
|
|
raise exceptions.ApiException(status=0, reason=msg)
|
|
|
|
if not stream:
|
|
# log response body
|
|
logger.debug("response body: %s", r.data)
|
|
|
|
return r
|
|
|
|
def get(self, url, headers=None, stream=False,
|
|
timeout=None, fields=None) -> urllib3.HTTPResponse:
|
|
return self.request("GET", url,
|
|
headers=headers,
|
|
stream=stream,
|
|
timeout=timeout,
|
|
fields=fields)
|
|
|
|
def head(self, url, headers=None, stream=False,
|
|
timeout=None, fields=None) -> urllib3.HTTPResponse:
|
|
return self.request("HEAD", url,
|
|
headers=headers,
|
|
stream=stream,
|
|
timeout=timeout,
|
|
fields=fields)
|
|
|
|
def options(self, url, headers=None,
|
|
body=None, stream=False, timeout=None, fields=None) -> urllib3.HTTPResponse:
|
|
return self.request("OPTIONS", url,
|
|
headers=headers,
|
|
stream=stream,
|
|
timeout=timeout,
|
|
body=body, fields=fields)
|
|
|
|
def delete(self, url, headers=None, body=None,
|
|
stream=False, timeout=None, fields=None) -> urllib3.HTTPResponse:
|
|
return self.request("DELETE", url,
|
|
headers=headers,
|
|
stream=stream,
|
|
timeout=timeout,
|
|
body=body, fields=fields)
|
|
|
|
def post(self, url, headers=None,
|
|
body=None, stream=False, timeout=None, fields=None) -> urllib3.HTTPResponse:
|
|
return self.request("POST", url,
|
|
headers=headers,
|
|
stream=stream,
|
|
timeout=timeout,
|
|
body=body, fields=fields)
|
|
|
|
def put(self, url, headers=None,
|
|
body=None, stream=False, timeout=None, fields=None) -> urllib3.HTTPResponse:
|
|
return self.request("PUT", url,
|
|
headers=headers,
|
|
stream=stream,
|
|
timeout=timeout,
|
|
body=body, fields=fields)
|
|
|
|
def patch(self, url, headers=None,
|
|
body=None, stream=False, timeout=None, fields=None) -> urllib3.HTTPResponse:
|
|
return self.request("PATCH", url,
|
|
headers=headers,
|
|
stream=stream,
|
|
timeout=timeout,
|
|
body=body, fields=fields)
|