devve1 commited on
Commit
fb37baa
1 Parent(s): 52cb13a

Update ordered_multimap.py

Browse files
Files changed (1) hide show
  1. ordered_multimap.py +5 -9
ordered_multimap.py CHANGED
@@ -9,20 +9,16 @@ class StringWrapper:
9
  class OrderedMultiIndexMapWeakRef:
10
  def __init__(self):
11
  self.data = []
 
12
  self.index_list = []
13
 
14
  def insert(self, id, text, title: str, sub_title: str):
15
  self.data.append(text)
16
- self.index_list.append((id, WeakValueDictionary({'title': StringWrapper(title), 'sub_title': StringWrapper(sub_title)})))
 
17
 
18
- def get(self, key):
19
- if key:
20
- return [self.data[i][1] for i in self.index_map[key]]
21
- else:
22
- return [self.data[i][1] for i in self.keyless_values]
23
-
24
- def all_titles(self):
25
- return self.index_list
26
 
27
  def all_texts(self):
28
  return self.data
 
9
  class OrderedMultiIndexMapWeakRef:
10
  def __init__(self):
11
  self.data = []
12
+ self.ids = []
13
  self.index_list = []
14
 
15
  def insert(self, id, text, title: str, sub_title: str):
16
  self.data.append(text)
17
+ self.ids.append(id)
18
+ self.index_list.append(WeakValueDictionary({'title': StringWrapper(title), 'sub_title': StringWrapper(sub_title)}))
19
 
20
+ def all_ids(self):
21
+ return self.ids
 
 
 
 
 
 
22
 
23
  def all_texts(self):
24
  return self.data