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.

102 lines
3.0 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
6 years ago
7 years ago
6 years ago
7 years ago
6 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,
  26. response_body=None):
  27. Exception.__init__(self, error_message)
  28. self.response_code = response_code
  29. self.response_body = response_body
  30. self.error_message = error_message
  31. def __str__(self):
  32. if self.response_code is not None:
  33. return "{0}: {1}".format(self.response_code, self.error_message)
  34. else:
  35. return "{0}".format(self.error_message)
  36. class KeycloakAuthenticationError(KeycloakError):
  37. pass
  38. class KeycloakConnectionError(KeycloakError):
  39. pass
  40. class KeycloakOperationError(KeycloakError):
  41. pass
  42. class KeycloakGetError(KeycloakOperationError):
  43. pass
  44. class KeycloakSecretNotFound(KeycloakOperationError):
  45. pass
  46. class KeycloakRPTNotFound(KeycloakOperationError):
  47. pass
  48. class KeycloakAuthorizationConfigError(KeycloakOperationError):
  49. pass
  50. class KeycloakInvalidTokenError(KeycloakOperationError):
  51. pass
  52. def raise_error_from_response(response, error, expected_code=200, skip_exists=False):
  53. if expected_code == response.status_code:
  54. if expected_code == requests.codes.no_content:
  55. return {}
  56. try:
  57. return response.json()
  58. except ValueError:
  59. return response.content
  60. if skip_exists and response.status_code == 409:
  61. return {"Already exists"}
  62. try:
  63. message = response.json()['message']
  64. except (KeyError, ValueError):
  65. message = response.content
  66. if isinstance(error, dict):
  67. error = error.get(response.status_code, KeycloakOperationError)
  68. else:
  69. if response.status_code == 401:
  70. error = KeycloakAuthenticationError
  71. raise error(error_message=message,
  72. response_code=response.status_code,
  73. response_body=response.content)