strings.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. # Licensed to the Apache Software Foundation (ASF) under one
  2. # or more contributor license agreements. See the NOTICE file
  3. # distributed with this work for additional information
  4. # regarding copyright ownership. The ASF licenses this file
  5. # to you under the Apache License, Version 2.0 (the
  6. # "License"); you may not use this file except in compliance
  7. # with the License. You may obtain a copy of the License at
  8. #
  9. # http://www.apache.org/licenses/LICENSE-2.0
  10. #
  11. # Unless required by applicable law or agreed to in writing,
  12. # software distributed under the License is distributed on an
  13. # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  14. # KIND, either express or implied. See the License for the
  15. # specific language governing permissions and limitations
  16. # under the License.
  17. """Common utility functions with strings."""
  18. from __future__ import annotations
  19. import random
  20. import string
  21. def get_random_string(length=8, choices=string.ascii_letters + string.digits):
  22. """Generate random string."""
  23. return "".join(random.choices(choices, k=length))
  24. TRUE_LIKE_VALUES = {"on", "t", "true", "y", "yes", "1"}
  25. def to_boolean(astring: str | None) -> bool:
  26. """Convert a string to a boolean."""
  27. if astring is None:
  28. return False
  29. if astring.lower() in TRUE_LIKE_VALUES:
  30. return True
  31. return False