Source code for robot.utils.misc

#  Copyright 2008-2015 Nokia Networks
#  Copyright 2016-     Robot Framework Foundation
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

import re

from .robottypes import is_integer
from .unic import safe_str


[docs]def printable_name(string, code_style=False): """Generates and returns printable name from the given string. Examples: 'simple' -> 'Simple' 'name with spaces' -> 'Name With Spaces' 'more spaces' -> 'More Spaces' 'Cases AND spaces' -> 'Cases AND Spaces' '' -> '' If 'code_style' is True: 'mixedCAPSCamel' -> 'Mixed CAPS Camel' 'camelCaseName' -> 'Camel Case Name' 'under_score_name' -> 'Under Score Name' 'under_and space' -> 'Under And Space' 'miXed_CAPS_nAMe' -> 'MiXed CAPS NAMe' '' -> '' """ if code_style and '_' in string: string = string.replace('_', ' ') parts = string.split() if code_style and len(parts) == 1 \ and not (string.isalpha() and string.islower()): parts = _split_camel_case(parts[0]) return ' '.join(part[0].upper() + part[1:] for part in parts)
def _split_camel_case(string): tokens = [] token = [] for prev, char, next in zip(' ' + string, string, string[1:] + ' '): if _is_camel_case_boundary(prev, char, next): if token: tokens.append(''.join(token)) token = [char] else: token.append(char) if token: tokens.append(''.join(token)) return tokens def _is_camel_case_boundary(prev, char, next): if prev.isdigit(): return not char.isdigit() if char.isupper(): return next.islower() or prev.isalpha() and not prev.isupper() return char.isdigit()
[docs]def plural_or_not(item): count = item if is_integer(item) else len(item) return '' if count in (1, -1) else 's'
[docs]def seq2str(sequence, quote="'", sep=', ', lastsep=' and '): """Returns sequence in format `'item 1', 'item 2' and 'item 3'`.""" sequence = [f'{quote}{safe_str(item)}{quote}' for item in sequence] if not sequence: return '' if len(sequence) == 1: return sequence[0] last_two = lastsep.join(sequence[-2:]) return sep.join(sequence[:-2] + [last_two])
[docs]def seq2str2(sequence): """Returns sequence in format `[ item 1 | item 2 | ... ]`.""" if not sequence: return '[ ]' return '[ %s ]' % ' | '.join(safe_str(item) for item in sequence)
[docs]def test_or_task(text: str, rpa: bool): """Replace 'test' with 'task' in the given `text` depending on `rpa`. If given text is `test`, `test` or `task` is returned directly. Otherwise, pattern `{test}` is searched from the text and occurrences replaced with `test` or `task`. In both cases matching the word `test` is case-insensitive and the returned `test` or `task` has exactly same case as the original. """ def replace(test): if not rpa: return test upper = [c.isupper() for c in test] return ''.join(c.upper() if up else c for c, up in zip('task', upper)) if text.upper() == 'TEST': return replace(text) return re.sub('{(test)}', lambda m: replace(m.group(1)), text, flags=re.IGNORECASE)
[docs]def isatty(stream): # first check if buffer was detached if hasattr(stream, 'buffer') and stream.buffer is None: return False if not hasattr(stream, 'isatty'): return False try: return stream.isatty() except ValueError: # Occurs if file is closed. return False
[docs]def parse_re_flags(flags=None): result = 0 if not flags: return result for flag in flags.split('|'): try: re_flag = getattr(re, flag.upper().strip()) except AttributeError: raise ValueError(f'Unknown regexp flag: {flag}') else: if isinstance(re_flag, re.RegexFlag): result |= re_flag else: raise ValueError(f'Unknown regexp flag: {flag}') return result
[docs]class classproperty(property): """Property that works with classes in addition to instances. Only supports getters. Setters and deleters cannot work with classes due to how the descriptor protocol works, and they are thus explicitly disabled. Metaclasses must be used if they are needed. """ def __init__(self, fget, fset=None, fdel=None, doc=None): if fset: self.setter(fset) if fdel: self.deleter(fset) super().__init__(fget) if doc: self.__doc__ = doc def __get__(self, instance, owner): return self.fget(owner)
[docs] def setter(self, fset): raise TypeError('Setters are not supported.')
[docs] def deleter(self, fset): raise TypeError('Deleters are not supported.')