Django forms and formset support added
[idea/community.git] / python / psi-api / src / com / jetbrains / python / codeInsight / PyCustomMemberTypeInfo.java
1 /*
2  * Copyright 2000-2016 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;
17
18 import com.intellij.openapi.util.Key;
19 import com.intellij.openapi.util.Pair;
20 import com.intellij.openapi.util.UserDataHolder;
21 import org.jetbrains.annotations.NotNull;
22
23 import java.util.Collections;
24 import java.util.HashMap;
25 import java.util.Map;
26
27 /**
28  * Info to add to type of custom member.
29  *
30  * @author Ilya.Kazakevich
31  */
32 public class PyCustomMemberTypeInfo<K> {
33   @NotNull
34   private final Map<Key<K>, K> myCustomInfo = new HashMap<Key<K>, K>();
35
36   public PyCustomMemberTypeInfo(@NotNull final Key<K> key, @NotNull final K value) {
37     this(Collections.singleton(Pair.create(key, value)));
38   }
39
40   public PyCustomMemberTypeInfo(@NotNull final Iterable<Pair<Key<K>, K>> customInfo) {
41     for (final Pair<Key<K>, K> pair : customInfo) {
42       myCustomInfo.put(pair.first, pair.second);
43     }
44   }
45
46   public PyCustomMemberTypeInfo(@NotNull final Map<Key<K>, K> customInfo) {
47     myCustomInfo.putAll(customInfo);
48   }
49
50   void fill(@NotNull final UserDataHolder typeToFill) {
51     for (final Map.Entry<Key<K>, K> entry : myCustomInfo.entrySet()) {
52       typeToFill.putUserData(entry.getKey(), entry.getValue());
53     }
54   }
55 }