devve1 commited on
Commit
9b7025c
1 Parent(s): cd3eef1

Update ordered_multimap.py

Browse files
Files changed (1) hide show
  1. ordered_multimap.py +3 -10
ordered_multimap.py CHANGED
@@ -1,21 +1,14 @@
1
- from weak_list import WeakList
2
  from collections import defaultdict
3
 
4
- class DictWrapper:
5
- def __init__(self, data: str):
6
- self.data = data
7
-
8
- def get_dict(self):
9
- return self.data
10
-
11
  class OrderedMultiIndexMapWeakRef:
12
  def __init__(self):
13
  self.data = []
14
- self.index_list = WeakList()
15
 
16
  def insert(self, text, title: str, sub_title: str):
17
  self.data.append(text)
18
- self.index_list.append(DictWrapper({'title': title, 'sub_title': sub_title}))
19
 
20
  def get(self, key):
21
  if key:
 
1
+ from weak_list import WeakValueDictionnary
2
  from collections import defaultdict
3
 
 
 
 
 
 
 
 
4
  class OrderedMultiIndexMapWeakRef:
5
  def __init__(self):
6
  self.data = []
7
+ self.index_list = []
8
 
9
  def insert(self, text, title: str, sub_title: str):
10
  self.data.append(text)
11
+ self.index_list.append(WeakValueDictionnary({'title': title, 'sub_title': sub_title}))
12
 
13
  def get(self, key):
14
  if key: