endpoint.py 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133
  1. from flask_login import current_user
  2. from flask_restful import Resource, reqparse
  3. from werkzeug.exceptions import Forbidden
  4. from controllers.console import api
  5. from controllers.console.setup import setup_required
  6. from controllers.console.wraps import account_initialization_required
  7. from libs.login import login_required
  8. from services.plugin.endpoint_service import EndpointService
  9. class EndpointCreateApi(Resource):
  10. @setup_required
  11. @login_required
  12. @account_initialization_required
  13. def post(self):
  14. user = current_user
  15. if not user.is_admin_or_owner:
  16. raise Forbidden()
  17. parser = reqparse.RequestParser()
  18. parser.add_argument("plugin_unique_identifier", type=str, required=True)
  19. parser.add_argument("settings", type=dict, required=True)
  20. args = parser.parse_args()
  21. plugin_unique_identifier = args["plugin_unique_identifier"]
  22. settings = args["settings"]
  23. return EndpointService.create_endpoint(
  24. tenant_id=user.current_tenant_id,
  25. user_id=user.id,
  26. plugin_unique_identifier=plugin_unique_identifier,
  27. settings=settings,
  28. )
  29. class EndpointListApi(Resource):
  30. @setup_required
  31. @login_required
  32. @account_initialization_required
  33. def get(self):
  34. user = current_user
  35. return EndpointService.list_endpoints(
  36. tenant_id=user.current_tenant_id,
  37. user_id=user.id,
  38. )
  39. class EndpointDeleteApi(Resource):
  40. @setup_required
  41. @login_required
  42. @account_initialization_required
  43. def post(self):
  44. user = current_user
  45. parser = reqparse.RequestParser()
  46. parser.add_argument("endpoint_id", type=str, required=True)
  47. args = parser.parse_args()
  48. endpoint_id = args["endpoint_id"]
  49. return EndpointService.delete_endpoint(
  50. tenant_id=user.current_tenant_id, user_id=user.id, endpoint_id=endpoint_id
  51. )
  52. class EndpointUpdateApi(Resource):
  53. @setup_required
  54. @login_required
  55. @account_initialization_required
  56. def post(self):
  57. user = current_user
  58. parser = reqparse.RequestParser()
  59. parser.add_argument("endpoint_id", type=str, required=True)
  60. parser.add_argument("settings", type=dict, required=True)
  61. args = parser.parse_args()
  62. endpoint_id = args["endpoint_id"]
  63. settings = args["settings"]
  64. return EndpointService.update_endpoint(
  65. tenant_id=user.current_tenant_id,
  66. user_id=user.id,
  67. endpoint_id=endpoint_id,
  68. settings=settings,
  69. )
  70. class EndpointEnableApi(Resource):
  71. @setup_required
  72. @login_required
  73. @account_initialization_required
  74. def post(self):
  75. user = current_user
  76. parser = reqparse.RequestParser()
  77. parser.add_argument("endpoint_id", type=str, required=True)
  78. args = parser.parse_args()
  79. endpoint_id = args["endpoint_id"]
  80. return EndpointService.enable_endpoint(
  81. tenant_id=user.current_tenant_id, user_id=user.id, endpoint_id=endpoint_id
  82. )
  83. class EndpointDisableApi(Resource):
  84. @setup_required
  85. @login_required
  86. @account_initialization_required
  87. def post(self):
  88. user = current_user
  89. parser = reqparse.RequestParser()
  90. parser.add_argument("endpoint_id", type=str, required=True)
  91. args = parser.parse_args()
  92. endpoint_id = args["endpoint_id"]
  93. return EndpointService.disable_endpoint(
  94. tenant_id=user.current_tenant_id, user_id=user.id, endpoint_id=endpoint_id
  95. )
  96. api.add_resource(EndpointCreateApi, "/workspaces/current/endpoints/create")
  97. api.add_resource(EndpointListApi, "/workspaces/current/endpoints/list")
  98. api.add_resource(EndpointDeleteApi, "/workspaces/current/endpoints/delete")
  99. api.add_resource(EndpointUpdateApi, "/workspaces/current/endpoints/update")
  100. api.add_resource(EndpointEnableApi, "/workspaces/current/endpoints/enable")
  101. api.add_resource(EndpointDisableApi, "/workspaces/current/endpoints/disable")