You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

128 lines
3.4 KiB

7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
6 years ago
7 years ago
6 years ago
6 years ago
7 years ago
7 years ago
  1. # -*- coding: utf-8 -*-
  2. #
  3. # The MIT License (MIT)
  4. #
  5. # Copyright (C) 2017 Marcos Pereira <marcospereira.mpj@gmail.com>
  6. #
  7. # Permission is hereby granted, free of charge, to any person obtaining a copy of
  8. # this software and associated documentation files (the "Software"), to deal in
  9. # the Software without restriction, including without limitation the rights to
  10. # use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
  11. # the Software, and to permit persons to whom the Software is furnished to do so,
  12. # subject to the following conditions:
  13. #
  14. # The above copyright notice and this permission notice shall be included in all
  15. # copies or substantial portions of the Software.
  16. #
  17. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18. # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
  19. # FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
  20. # COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
  21. # IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  22. # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23. import requests
  24. class KeycloakError(Exception):
  25. def __init__(self, error_message="", response_code=None, response_body=None):
  26. Exception.__init__(self, error_message)
  27. self.response_code = response_code
  28. self.response_body = response_body
  29. self.error_message = error_message
  30. def __str__(self):
  31. if self.response_code is not None:
  32. return "{0}: {1}".format(self.response_code, self.error_message)
  33. else:
  34. return "{0}".format(self.error_message)
  35. class KeycloakAuthenticationError(KeycloakError):
  36. pass
  37. class KeycloakConnectionError(KeycloakError):
  38. pass
  39. class KeycloakOperationError(KeycloakError):
  40. pass
  41. class KeycloakDeprecationError(KeycloakError):
  42. pass
  43. class KeycloakGetError(KeycloakOperationError):
  44. pass
  45. class KeycloakPostError(KeycloakOperationError):
  46. pass
  47. class KeycloakPutError(KeycloakOperationError):
  48. pass
  49. class KeycloakDeleteError(KeycloakOperationError):
  50. pass
  51. class KeycloakSecretNotFound(KeycloakOperationError):
  52. pass
  53. class KeycloakRPTNotFound(KeycloakOperationError):
  54. pass
  55. class KeycloakAuthorizationConfigError(KeycloakOperationError):
  56. pass
  57. class KeycloakInvalidTokenError(KeycloakOperationError):
  58. pass
  59. class KeycloakPermissionFormatError(KeycloakOperationError):
  60. pass
  61. class PermissionDefinitionError(Exception):
  62. pass
  63. def raise_error_from_response(response, error, expected_codes=None, skip_exists=False):
  64. if expected_codes is None:
  65. expected_codes = [200, 201, 204]
  66. if response.status_code in expected_codes:
  67. if response.status_code == requests.codes.no_content:
  68. return {}
  69. try:
  70. return response.json()
  71. except ValueError:
  72. return response.content
  73. if skip_exists and response.status_code == 409:
  74. return {"msg": "Already exists"}
  75. try:
  76. message = response.json()["message"]
  77. except (KeyError, ValueError):
  78. message = response.content
  79. if isinstance(error, dict):
  80. error = error.get(response.status_code, KeycloakOperationError)
  81. else:
  82. if response.status_code == 401:
  83. error = KeycloakAuthenticationError
  84. raise error(
  85. error_message=message, response_code=response.status_code, response_body=response.content
  86. )