Function generate_alias

pgcli/pgcompleter.py:62–77

Generate a table alias. Given a table name will return an alias for that table using the first of the following options there's a match for. 1. The predefined alias for table defined in the alias_map. 2. All upper-case letters in the table name. 3. The first letter

(tbl, alias_map=None)

def generate_alias(tbl, alias_map=None):
    """Generate a table alias.

    Given a table name will return an alias for that table using the first of
    the following options there's a match for.

        1. The predefined alias for table defined in the alias_map.
        2. All upper-case letters in the table name.
        3. The first letter of the table name and all letters preceded by _

    :param tbl: unescaped name of the table to alias
    :param alias_map: optional mapping of predefined table aliases
    """
    if alias_map and tbl in alias_map:
        return alias_map[tbl]
    return "".join([l for l in tbl if l.isupper()] or [l for l, prev in zip(tbl, "_" + tbl) if prev == "_" and l != "_"])


class InvalidMapFile(ValueError):