monkey_patches.py 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. #!/usr/bin/env python
  2. # Licensed to Cloudera, Inc. under one
  3. # or more contributor license agreements. See the NOTICE file
  4. # distributed with this work for additional information
  5. # regarding copyright ownership. Cloudera, Inc. licenses this file
  6. # to you under the Apache License, Version 2.0 (the
  7. # "License"); you may not use this file except in compliance
  8. # with the License. You may obtain a copy of the License at
  9. #
  10. # http://www.apache.org/licenses/LICENSE-2.0
  11. #
  12. # Unless required by applicable law or agreed to in writing, software
  13. # distributed under the License is distributed on an "AS IS" BASIS,
  14. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15. # See the License for the specific language governing permissions and
  16. # limitations under the License.
  17. import re
  18. from django.core.validators import RegexValidator
  19. from desktop.conf import ENABLE_ORGANIZATIONS
  20. from desktop.lib.django_util import get_username_re_rule
  21. if ENABLE_ORGANIZATIONS.get():
  22. from useradmin.models import User
  23. else:
  24. from django.contrib.auth.models import User
  25. def monkey_patch_username_validator():
  26. """
  27. In 1.6, the `User.username` field gained some validation rules to check that
  28. it conformed to a particular regex. Unfortunately we use to support a more
  29. liberal username scheme. The proper solution would be to use our own custom
  30. user model, but that touches a lot of code. It's easier if we just modify the
  31. regular expression inside the username validator.
  32. """
  33. username = User._meta.get_field("username")
  34. regex = re.compile('^%s$' % get_username_re_rule())
  35. for validator in username.validators:
  36. if isinstance(validator, RegexValidator):
  37. validator.regex = regex
  38. monkey_patch_username_validator()