Mercurial > p > roundup > code
annotate roundup/indexer.py @ 833:b80aaedba3db
Only save the index if the thing is loaded and changed.
Also, don't load the index just for a save.
| author | Richard Jones <richard@users.sourceforge.net> |
|---|---|
| date | Tue, 09 Jul 2002 21:38:43 +0000 |
| parents | 0a2c1f5e0e5a |
| children | 568eed5fb4fd |
| rev | line source |
|---|---|
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
1 # |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
2 # This module is derived from the module described at: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
3 # http://gnosis.cx/publish/programming/charming_python_15.txt |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
4 # |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
5 # Author: David Mertz (mertz@gnosis.cx) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
6 # Thanks to: Pat Knight (p.knight@ktgroup.co.uk) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
7 # Gregory Popovitch (greg@gpy.com) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
8 # |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
9 # The original module was released under this license, and remains under |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
10 # it: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
11 # |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
12 # This file is released to the public domain. I (dqm) would |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
13 # appreciate it if you choose to keep derived works under terms |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
14 # that promote freedom, but obviously am giving up any rights |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
15 # to compel such. |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
16 # |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
17 #$Id: indexer.py,v 1.7 2002-07-09 21:38:43 richard Exp $ |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
18 ''' |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
19 This module provides an indexer class, RoundupIndexer, that stores text |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
20 indices in a roundup instance. This class makes searching the content of |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
21 messages and text files possible. |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
22 ''' |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
23 import os, shutil, re, mimetypes, marshal, zlib, errno |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
24 |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
25 class Indexer: |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
26 ''' Indexes information from roundup's hyperdb to allow efficient |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
27 searching. |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
28 |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
29 Three structures are created by the indexer: |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
30 files {identifier: (fileid, wordcount)} |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
31 words {word: {fileid: count}} |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
32 fileids {fileid: identifier} |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
33 ''' |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
34 def __init__(self, db_path): |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
35 self.indexdb_path = os.path.join(db_path, 'indexes') |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
36 self.indexdb = os.path.join(self.indexdb_path, 'index.db') |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
37 self.reindex = 0 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
38 self.quiet = 9 |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
39 self.changed = 0 |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
40 |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
41 # see if we need to reindex because of a change in code |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
42 if (not os.path.exists(self.indexdb_path) or |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
43 not os.path.exists(os.path.join(self.indexdb_path, 'version'))): |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
44 # TODO: if the version file exists (in the future) we'll want to |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
45 # check the value in it - for now the file itself is a flag |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
46 self.force_reindex() |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
47 |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
48 def force_reindex(self): |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
49 '''Force a reindex condition |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
50 ''' |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
51 if os.path.exists(self.indexdb_path): |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
52 shutil.rmtree(self.indexdb_path) |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
53 os.makedirs(self.indexdb_path) |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
54 os.chmod(self.indexdb_path, 0775) |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
55 open(os.path.join(self.indexdb_path, 'version'), 'w').write('1\n') |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
56 self.reindex = 1 |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
57 self.changed = 1 |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
58 |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
59 def should_reindex(self): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
60 '''Should we reindex? |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
61 ''' |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
62 return self.reindex |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
63 |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
64 def add_text(self, identifier, text, mime_type='text/plain'): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
65 ''' Add some text associated with the (classname, nodeid, property) |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
66 identifier. |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
67 ''' |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
68 # make sure the index is loaded |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
69 self.load_index() |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
70 |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
71 # remove old entries for this identifier |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
72 if self.files.has_key(identifier): |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
73 self.purge_entry(identifier) |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
74 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
75 # split into words |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
76 words = self.splitter(text, mime_type) |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
77 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
78 # Find new file index, and assign it to identifier |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
79 # (_TOP uses trick of negative to avoid conflict with file index) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
80 self.files['_TOP'] = (self.files['_TOP'][0]-1, None) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
81 file_index = abs(self.files['_TOP'][0]) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
82 self.files[identifier] = (file_index, len(words)) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
83 self.fileids[file_index] = identifier |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
84 |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
85 # find the unique words |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
86 filedict = {} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
87 for word in words: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
88 if filedict.has_key(word): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
89 filedict[word] = filedict[word]+1 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
90 else: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
91 filedict[word] = 1 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
92 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
93 # now add to the totals |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
94 for word in filedict.keys(): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
95 # each word has a dict of {identifier: count} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
96 if self.words.has_key(word): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
97 entry = self.words[word] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
98 else: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
99 # new word |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
100 entry = {} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
101 self.words[word] = entry |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
102 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
103 # make a reference to the file for this word |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
104 entry[file_index] = filedict[word] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
105 |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
106 # save needed |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
107 self.changed = 1 |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
108 |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
109 def splitter(self, text, ftype): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
110 ''' Split the contents of a text string into a list of 'words' |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
111 ''' |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
112 if ftype == 'text/plain': |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
113 words = self.text_splitter(text) |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
114 else: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
115 return [] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
116 return words |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
117 |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
118 def text_splitter(self, text): |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
119 """Split text/plain string into a list of words |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
120 """ |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
121 # case insensitive |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
122 text = text.upper() |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
123 |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
124 # Split the raw text, losing anything longer than 25 characters |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
125 # since that'll be gibberish (encoded text or somesuch) or shorter |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
126 # than 3 characters since those short words appear all over the |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
127 # place |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
128 return re.findall(r'\b\w{2,25}\b', text) |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
129 |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
130 def search(self, search_terms, klass, ignore={}, |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
131 dre=re.compile(r'([^\d]+)(\d+)')): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
132 ''' Display search results looking for [search, terms] associated |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
133 with the hyperdb Class "klass". Ignore hits on {class: property}. |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
134 |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
135 "dre" is a helper, not an argument. |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
136 ''' |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
137 # do the index lookup |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
138 hits = self.find(search_terms) |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
139 if not hits: |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
140 return {} |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
141 |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
142 # this is specific to "issue" klass ... eugh |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
143 designator_propname = {'msg': 'messages', 'file': 'files'} |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
144 |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
145 # build a dictionary of nodes and their associated messages |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
146 # and files |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
147 nodeids = {} |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
148 for classname, nodeid, property in hits.values(): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
149 # skip this result if we don't care about this class/property |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
150 if ignore.has_key((classname, property)): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
151 continue |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
152 |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
153 # if it's a property on klass, it's easy |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
154 if classname == klass.classname: |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
155 if not nodeids.has_key(nodeid): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
156 nodeids[nodeid] = {} |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
157 continue |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
158 |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
159 # it's a linked class - find the klass entries that are |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
160 # linked to it |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
161 linkprop = designator_propname[classname] |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
162 for resid in klass.find(**{linkprop: nodeid}): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
163 resid = str(resid) |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
164 if not nodeids.has_key(id): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
165 nodeids[resid] = {} |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
166 |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
167 # update the links for this klass nodeid |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
168 node_dict = nodeids[resid] |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
169 if not node_dict.has_key(linkprop): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
170 node_dict[linkprop] = [nodeid] |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
171 elif node_dict.has_key(linkprop): |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
172 node_dict[linkprop].append(nodeid) |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
173 return nodeids |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
174 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
175 # we override this to ignore not 2 < word < 25 and also to fix a bug - |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
176 # the (fail) case. |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
177 def find(self, wordlist): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
178 ''' Locate files that match ALL the words in wordlist |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
179 ''' |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
180 if not hasattr(self, 'words'): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
181 self.load_index() |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
182 self.load_index(wordlist=wordlist) |
|
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
183 entries = {} |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
184 hits = None |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
185 for word in wordlist: |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
186 if not 2 < len(word) < 25: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
187 # word outside the bounds of what we index - ignore |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
188 continue |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
189 word = word.upper() |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
190 entry = self.words.get(word) # For each word, get index |
|
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
191 entries[word] = entry # of matching files |
|
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
192 if not entry: # Nothing for this one word (fail) |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
193 return {} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
194 if hits is None: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
195 hits = {} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
196 for k in entry.keys(): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
197 hits[k] = self.fileids[k] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
198 else: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
199 # Eliminate hits for every non-match |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
200 for fileid in hits.keys(): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
201 if not entry.has_key(fileid): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
202 del hits[fileid] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
203 if hits is None: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
204 return {} |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
205 return hits |
|
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
206 |
|
827
0a2c1f5e0e5a
We're indexing numbers now, and _underscore words
Richard Jones <richard@users.sourceforge.net>
parents:
826
diff
changeset
|
207 segments = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ#_-!" |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
208 def load_index(self, reload=0, wordlist=None): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
209 # Unless reload is indicated, do not load twice |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
210 if self.index_loaded() and not reload: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
211 return 0 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
212 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
213 # Ok, now let's actually load it |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
214 db = {'WORDS': {}, 'FILES': {'_TOP':(0,None)}, 'FILEIDS': {}} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
215 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
216 # Identify the relevant word-dictionary segments |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
217 if not wordlist: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
218 segments = self.segments |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
219 else: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
220 segments = ['-','#'] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
221 for word in wordlist: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
222 segments.append(word[0].upper()) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
223 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
224 # Load the segments |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
225 for segment in segments: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
226 try: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
227 f = open(self.indexdb + segment, 'rb') |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
228 except IOError, error: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
229 if error.errno != errno.ENOENT: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
230 raise |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
231 else: |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
232 pickle_str = zlib.decompress(f.read()) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
233 f.close() |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
234 dbslice = marshal.loads(pickle_str) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
235 if dbslice.get('WORDS'): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
236 # if it has some words, add them |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
237 for word, entry in dbslice['WORDS'].items(): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
238 db['WORDS'][word] = entry |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
239 if dbslice.get('FILES'): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
240 # if it has some files, add them |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
241 db['FILES'] = dbslice['FILES'] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
242 if dbslice.get('FILEIDS'): |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
243 # if it has fileids, add them |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
244 db['FILEIDS'] = dbslice['FILEIDS'] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
245 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
246 self.words = db['WORDS'] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
247 self.files = db['FILES'] |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
248 self.fileids = db['FILEIDS'] |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
249 self.changed = 0 |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
250 |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
251 def save_index(self): |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
252 # only save if the index is loaded and changed |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
253 if not self.index_loaded() or not self.changed: |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
254 return |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
255 |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
256 # brutal space saver... delete all the small segments |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
257 for segment in self.segments: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
258 try: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
259 os.remove(self.indexdb + segment) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
260 except OSError: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
261 # probably just nonexistent segment index file |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
262 # TODO: make sure it's an EEXIST |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
263 pass |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
264 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
265 # First write the much simpler filename/fileid dictionaries |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
266 dbfil = {'WORDS':None, 'FILES':self.files, 'FILEIDS':self.fileids} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
267 open(self.indexdb+'-','wb').write(zlib.compress(marshal.dumps(dbfil))) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
268 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
269 # The hard part is splitting the word dictionary up, of course |
|
827
0a2c1f5e0e5a
We're indexing numbers now, and _underscore words
Richard Jones <richard@users.sourceforge.net>
parents:
826
diff
changeset
|
270 letters = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ#_" |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
271 segdicts = {} # Need batch of empty dicts |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
272 for segment in letters: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
273 segdicts[segment] = {} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
274 for word, entry in self.words.items(): # Split into segment dicts |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
275 initchar = word[0].upper() |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
276 segdicts[initchar][word] = entry |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
277 |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
278 # save |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
279 for initchar in letters: |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
280 db = {'WORDS':segdicts[initchar], 'FILES':None, 'FILEIDS':None} |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
281 pickle_str = marshal.dumps(db) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
282 filename = self.indexdb + initchar |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
283 pickle_fh = open(filename, 'wb') |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
284 pickle_fh.write(zlib.compress(pickle_str)) |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
285 os.chmod(filename, 0664) |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
286 |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
287 # save done |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
288 self.changed = 0 |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
289 |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
290 def purge_entry(self, identifier): |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
291 ''' Remove a file from file index and word index |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
292 ''' |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
293 if not self.files.has_key(identifier): |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
294 return |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
295 |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
296 file_index = self.files[identifier][0] |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
297 del self.files[identifier] |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
298 del self.fileids[file_index] |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
299 |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
300 # The much harder part, cleanup the word index |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
301 for key, occurs in self.words.items(): |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
302 if occurs.has_key(file_index): |
|
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
303 del occurs[file_index] |
|
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
304 |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
305 # save needed |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
306 self.changed = 1 |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
307 |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
308 def index_loaded(self): |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
309 return (hasattr(self,'fileids') and hasattr(self,'files') and |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
310 hasattr(self,'words')) |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
311 |
|
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
312 # |
|
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
313 #$Log: not supported by cvs2svn $ |
|
833
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
314 #Revision 1.6 2002/07/09 04:26:44 richard |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
315 #We're indexing numbers now, and _underscore words |
|
b80aaedba3db
Only save the index if the thing is loaded and changed.
Richard Jones <richard@users.sourceforge.net>
parents:
827
diff
changeset
|
316 # |
|
827
0a2c1f5e0e5a
We're indexing numbers now, and _underscore words
Richard Jones <richard@users.sourceforge.net>
parents:
826
diff
changeset
|
317 #Revision 1.5 2002/07/09 04:19:09 richard |
|
0a2c1f5e0e5a
We're indexing numbers now, and _underscore words
Richard Jones <richard@users.sourceforge.net>
parents:
826
diff
changeset
|
318 #Added reindex command to roundup-admin. |
|
0a2c1f5e0e5a
We're indexing numbers now, and _underscore words
Richard Jones <richard@users.sourceforge.net>
parents:
826
diff
changeset
|
319 #Fixed reindex on first access. |
|
0a2c1f5e0e5a
We're indexing numbers now, and _underscore words
Richard Jones <richard@users.sourceforge.net>
parents:
826
diff
changeset
|
320 #Also fixed reindexing of entries that change. |
|
0a2c1f5e0e5a
We're indexing numbers now, and _underscore words
Richard Jones <richard@users.sourceforge.net>
parents:
826
diff
changeset
|
321 # |
|
826
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
322 #Revision 1.4 2002/07/09 03:02:52 richard |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
323 #More indexer work: |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
324 #- all String properties may now be indexed too. Currently there's a bit of |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
325 # "issue" specific code in the actual searching which needs to be |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
326 # addressed. In a nutshell: |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
327 # + pass 'indexme="yes"' as a String() property initialisation arg, eg: |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
328 # file = FileClass(db, "file", name=String(), type=String(), |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
329 # comment=String(indexme="yes")) |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
330 # + the comment will then be indexed and be searchable, with the results |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
331 # related back to the issue that the file is linked to |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
332 #- as a result of this work, the FileClass has a default MIME type that may |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
333 # be overridden in a subclass, or by the use of a "type" property as is |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
334 # done in the default templates. |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
335 #- the regeneration of the indexes (if necessary) is done once the schema is |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
336 # set up in the dbinit. |
|
6d7a45c8464a
Added reindex command to roundup-admin.
Richard Jones <richard@users.sourceforge.net>
parents:
825
diff
changeset
|
337 # |
|
825
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
338 #Revision 1.3 2002/07/08 06:58:15 richard |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
339 #cleaned up the indexer code: |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
340 # - it splits more words out (much simpler, faster splitter) |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
341 # - removed code we'll never use (roundup.roundup_indexer has the full |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
342 # implementation, and replaces roundup.indexer) |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
343 # - only index text/plain and rfc822/message (ideas for other text formats to |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
344 # index are welcome) |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
345 # - added simple unit test for indexer. Needs more tests for regression. |
|
0779ea9f1f18
More indexer work:
Richard Jones <richard@users.sourceforge.net>
parents:
818
diff
changeset
|
346 # |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
347 #Revision 1.2 2002/05/25 07:16:24 rochecompaan |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
348 #Merged search_indexing-branch with HEAD |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
349 # |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
350 #Revision 1.1.2.3 2002/05/02 11:52:12 rochecompaan |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
351 #Fixed small bug that prevented indexes from being generated. |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
352 # |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
353 #Revision 1.1.2.2 2002/04/19 19:54:42 rochecompaan |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
354 #cgi_client.py |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
355 # removed search link for the time being |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
356 # moved rendering of matches to htmltemplate |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
357 #hyperdb.py |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
358 # filtering of nodes on full text search incorporated in filter method |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
359 #roundupdb.py |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
360 # added paramater to call of filter method |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
361 #roundup_indexer.py |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
362 # added search method to RoundupIndexer class |
|
749
51c425129b35
Merged search_indexing-branch with HEAD
Roche Compaan <rochecompaan@users.sourceforge.net>
parents:
diff
changeset
|
363 # |
|
818
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
364 #Revision 1.1.2.1 2002/04/03 11:55:57 rochecompaan |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
365 # . Added feature #526730 - search for messages capability |
|
254b8d112eec
cleaned up the indexer code:
Richard Jones <richard@users.sourceforge.net>
parents:
749
diff
changeset
|
366 # |
