Function _emit_on

mycli/packages/completion_engine.py:308–335

(ctx: SuggestContext)

def _emit_on(ctx: SuggestContext) -> list[Suggestion]:
    tables = _tables(ctx)  # [(schema, table, alias), ...]
    parent = _parent_name(ctx)
    if parent:
        # "ON parent.<suggestion>"
        # parent can be either a schema name or table alias
        # todo recognize and separate schema and table suggestions
        # todo remove function suggestions here
        tables = [t for t in tables if identifies(parent, *t)]
        return [
            {'type': 'column', 'tables': tables},
            {'type': 'table', 'schema': parent},
            {'type': 'view', 'schema': parent},
            {'type': 'function', 'schema': parent},
        ]

    # ON <suggestion>
    # Use table alias if there is one, otherwise the table name
    aliases = _aliases(tables)
    suggest: list[Suggestion] = [{'type': 'fk_join', 'tables': tables}, {'type': 'alias', 'aliases': aliases}]

    # The lists of 'aliases' could be empty if we're trying to complete
    # a GRANT query. eg: GRANT SELECT, INSERT ON <tab>
    # In that case we just suggest all schemata and all tables.
    if not aliases:
        suggest.append({'type': 'database'})
        suggest.append({'type': 'table', 'schema': parent})
    return suggest


def _emit_database(_ctx: SuggestContext) -> list[Suggestion]: