mirror of
https://github.com/langgenius/dify.git
synced 2026-01-14 06:07:33 +08:00
d5ff89f6d3
- Initialized `response` variable in `trigger.py` to ensure proper handling in the trigger endpoint. - Updated `http_parser.py` to conditionally set `CONTENT_TYPE` and `CONTENT_LENGTH` headers for improved robustness. - Changed `datetime.utcnow()` to `datetime.now(UTC)` in `sqlalchemy_workflow_trigger_log_repository.py` and `rate_limiter.py` for consistent time zone handling. - Refactored `async_workflow_service.py` to use the public method `get_tenant_owner_timezone` for better encapsulation. - Simplified subscription retrieval logic in `plugin_parameter_service.py` for clarity. These changes improve code reliability and maintainability while ensuring accurate time management and request processing.
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import logging
|
|
import re
|
|
|
|
from flask import jsonify, request
|
|
from werkzeug.exceptions import NotFound
|
|
|
|
from controllers.trigger import bp
|
|
from services.trigger.trigger_subscription_builder_service import TriggerSubscriptionBuilderService
|
|
from services.trigger_service import TriggerService
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
UUID_PATTERN = r"^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$"
|
|
UUID_MATCHER = re.compile(UUID_PATTERN)
|
|
|
|
|
|
@bp.route("/plugin/<string:endpoint_id>", methods=["GET", "POST", "PUT", "PATCH", "DELETE", "HEAD", "OPTIONS"])
|
|
def trigger_endpoint(endpoint_id: str):
|
|
"""
|
|
Handle endpoint trigger calls.
|
|
"""
|
|
# endpoint_id must be UUID
|
|
if not UUID_MATCHER.match(endpoint_id):
|
|
raise NotFound("Invalid endpoint ID")
|
|
handling_chain = [
|
|
TriggerService.process_endpoint,
|
|
TriggerSubscriptionBuilderService.process_builder_validation_endpoint,
|
|
]
|
|
response = None
|
|
try:
|
|
for handler in handling_chain:
|
|
response = handler(endpoint_id, request)
|
|
if response:
|
|
break
|
|
if not response:
|
|
raise NotFound("Endpoint not found")
|
|
return response
|
|
except ValueError as e:
|
|
raise NotFound(str(e))
|
|
except Exception as e:
|
|
logger.exception("Webhook processing failed for {endpoint_id}")
|
|
return jsonify({"error": "Internal server error", "message": str(e)}), 500
|