0f21128c657b642d400e8d3c25b685c43f52c17e
[idea/community.git] / python / src / com / jetbrains / python / codeInsight / intentions / PyConvertLiteralToSetIntention.java
1 /*
2  * Copyright 2000-2015 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.jetbrains.python.codeInsight.intentions;
17
18 import com.intellij.openapi.util.Condition;
19 import com.intellij.util.containers.ContainerUtil;
20 import com.jetbrains.python.psi.*;
21 import org.jetbrains.annotations.NotNull;
22
23 /**
24  * @author Mikhail Golubev
25  */
26 public class PyConvertLiteralToSetIntention extends PyBaseConvertCollectionLiteralIntention {
27   public PyConvertLiteralToSetIntention() {
28     super(PySetLiteralExpression.class, "set", "{", "}");
29   }
30
31   @Override
32   protected boolean isAvailableForCollection(PySequenceExpression literal) {
33     return LanguageLevel.forElement(literal).isAtLeast(LanguageLevel.PYTHON27) && !isInTargetPosition(literal);
34   }
35
36   private static boolean isInTargetPosition(@NotNull final PySequenceExpression sequenceLiteral) {
37     return ContainerUtil.exists(sequenceLiteral.getElements(), new Condition<PyExpression>() {
38       @Override
39       public boolean value(PyExpression expression) {
40         return expression instanceof PyTargetExpression;
41       }
42     });
43   }
44 }