Method _parseFromStatement

server/src/parser/parser.ts:899–956

()

    // import_as_names: import_as_name (',' import_as_name)* [',']
    // import_as_name: NAME ['as' NAME]
    private _parseFromStatement(): ImportFromNode {
        let fromToken = this._getKeywordToken(KeywordType.From);

        let modName = this._parseDottedModuleName(true);
        let importFromNode = new ImportFromNode(fromToken, modName);

        if (!this._consumeTokenIfKeyword(KeywordType.Import)) {
            this._addError('Expected "import"', this._peekToken());
        } else {
            // Look for "*" token.
            if (!this._consumeTokenIfOperator(OperatorType.Multiply)) {
                let inParen = this._consumeTokenIfType(TokenType.OpenParenthesis);

                while (true) {
                    let importName = this._getTokenIfIdentifier();
                    if (!importName) {
                        break;
                    }

                    let importFromAsNode = new ImportFromAsNode(new NameNode(importName));

                    if (this._consumeTokenIfKeyword(KeywordType.As)) {
                        let aliasName = this._getTokenIfIdentifier();
                        if (!aliasName) {
                            this._addError('Expected alias symbol name', this._peekToken());
                        } else {
                            importFromAsNode.alias = new NameNode(aliasName);
                            importFromAsNode.extend(aliasName);
                        }
                    }

                    importFromNode.imports.push(importFromAsNode);
                    importFromNode.extend(importFromAsNode);

                    if (!this._consumeTokenIfType(TokenType.Comma)) {
                        break;
                    }
                }

                if (importFromNode.imports.length === 0) {
                    this._addError('Expected imported symbol name', this._peekToken());
                }

                if (inParen) {
                    let nextToken = this._peekToken();
                    if (!this._consumeTokenIfType(TokenType.CloseParenthesis)) {
                        this._addError('Expected ")"', this._peekToken());
                    } else {
                        importFromNode.extend(nextToken);
                    }
                }
            }
        }

        // TODO - need to process __future__ imports

        return importFromNode;
    }