diff options
Diffstat (limited to 'python/psi-api/src/com/jetbrains/python/psi')
-rw-r--r-- | python/psi-api/src/com/jetbrains/python/psi/PyElementGenerator.java | 30 | ||||
-rw-r--r-- | python/psi-api/src/com/jetbrains/python/psi/PyStatementPart.java | 8 |
2 files changed, 28 insertions, 10 deletions
diff --git a/python/psi-api/src/com/jetbrains/python/psi/PyElementGenerator.java b/python/psi-api/src/com/jetbrains/python/psi/PyElementGenerator.java index daa8b3063014..539d3ab8df2e 100644 --- a/python/psi-api/src/com/jetbrains/python/psi/PyElementGenerator.java +++ b/python/psi-api/src/com/jetbrains/python/psi/PyElementGenerator.java @@ -90,8 +90,6 @@ public abstract class PyElementGenerator { @NotNull public abstract PyCallExpression createCallExpression(final LanguageLevel langLevel, String functionName); - public abstract PyImportStatement createImportStatementFromText(final LanguageLevel languageLevel, String text); - public abstract PyImportElement createImportElement(final LanguageLevel languageLevel, String name); public abstract PyFunction createProperty(final LanguageLevel languageLevel, @@ -139,4 +137,32 @@ public abstract class PyElementGenerator { */ @NotNull public abstract PsiElement createNewLine(); + + /** + * Creates import statement of form {@code from qualifier import name as alias}. + * + * @param languageLevel language level for created element + * @param qualifier from where {@code name} will be imported (module name) + * @param name text of the reference in import element + * @param alias optional alias for {@code as alias} part + * @return created {@link com.jetbrains.python.psi.PyFromImportStatement} + */ + @NotNull + public abstract PyFromImportStatement createFromImportStatement(@NotNull LanguageLevel languageLevel, + @NotNull String qualifier, + @NotNull String name, + @Nullable String alias); + + /** + * Creates import statement of form {@code import name as alias}. + * + * @param languageLevel language level for created element + * @param name text of the reference in import element (module name) + * @param alias optional alias for {@code as alias} part + * @return created {@link com.jetbrains.python.psi.PyImportStatement} + */ + @NotNull + public abstract PyImportStatement createImportStatement(@NotNull LanguageLevel languageLevel, + @NotNull String name, + @Nullable String alias); } diff --git a/python/psi-api/src/com/jetbrains/python/psi/PyStatementPart.java b/python/psi-api/src/com/jetbrains/python/psi/PyStatementPart.java index a14a91ef9f9c..90ef99cec362 100644 --- a/python/psi-api/src/com/jetbrains/python/psi/PyStatementPart.java +++ b/python/psi-api/src/com/jetbrains/python/psi/PyStatementPart.java @@ -15,8 +15,6 @@ */ package com.jetbrains.python.psi; -import org.jetbrains.annotations.NotNull; - /** * Abstract part of a multipart statement. * User: dcheryasov @@ -24,10 +22,4 @@ import org.jetbrains.annotations.NotNull; */ public interface PyStatementPart extends PyElement, PyStatementListContainer { PyStatementPart[] EMPTY_ARRAY = new PyStatementPart[0]; - - /** - * @return the body of the part. - */ - @NotNull - PyStatementList getStatementList(); } |