Mercurial > hg > ZDBInterface
annotate ZDBInterfaceFolder.py @ 7:917e28a08c58
search case insensitive. max and skip options.
author | casties |
---|---|
date | Tue, 15 Feb 2011 21:14:03 +0100 |
parents | ca30cf0e810d |
children | 17b19345d011 |
rev | line source |
---|---|
3 | 1 ''' |
2 Created on 14.2.2011 | |
3 | |
4 @author: casties | |
5 ''' | |
6 | |
7 from OFS.Folder import Folder | |
8 from Products.PageTemplates.PageTemplateFile import PageTemplateFile | |
9 from AccessControl import getSecurityManager, Unauthorized | |
10 import logging | |
11 import re | |
12 | |
13 from DBInterface import * | |
14 | |
15 | |
16 class ZDBInterfaceFolder(DBInterface, Folder): | |
17 """Folder for database queries | |
18 """ | |
19 | |
20 meta_type="ZDBInterfaceFolder" | |
21 manage_options=Folder.manage_options+( | |
22 {'label':'Config','action':'manage_editZDBInterfaceFolderForm'}, | |
23 ) | |
24 | |
25 # management templates | |
26 manage_editZDBInterfaceFolderForm=PageTemplateFile('zpt/editZDBInterfaceFolder',globals()) | |
27 | |
28 | |
29 def __init__(self, id, title, connection_id=None): | |
30 """init""" | |
31 self.id = id | |
32 self.title = title | |
33 # database connection id | |
34 self.connection_id = connection_id | |
35 | |
36 | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
37 def getSQLQuery(self, table, argv): |
4 | 38 """returns dict with SQL query string and args""" |
39 logging.debug("getSQLquery table=%s argv=%s"%(table,argv)) | |
40 args = [] | |
41 select = "*" | |
42 order = None | |
43 wheres = [] | |
44 whereOp = "AND" | |
45 done = {} | |
7 | 46 limit = None |
47 offset = None | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
48 |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
49 def doOp(op, param, val): |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
50 # handles comparison operations in WHERE clause |
7 | 51 logging.debug("doop op=%s param=%s val=%s"%(op,param,val)) |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
52 if isinstance(val, list): |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
53 # join multiple parameters with spaces (makes sense with checkbox and -op=all) |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
54 val = " ".join(val) |
7 | 55 |
56 # string comparisons are case-insensitive | |
57 if isinstance(val, basestring): | |
58 param = "LOWER(%s)"%param | |
59 val = val.lower() | |
60 | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
61 if op == "eq": |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
62 wheres.append(param + " = %s") |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
63 args.append(val) |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
64 elif op == "lt": |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
65 wheres.append(param + " < %s") |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
66 args.append(val) |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
67 elif op == "gt": |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
68 wheres.append(param + " > %s") |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
69 args.append(val) |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
70 elif op == "bw": |
7 | 71 wheres.append(param + " ILIKE %s") |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
72 args.append("%" + val) |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
73 elif op == "ew": |
7 | 74 wheres.append(param + " ILIKE %s") |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
75 args.append(val + "%") |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
76 elif op == "ct": |
7 | 77 wheres.append(param + " ILIKE %s") |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
78 args.append("%" + val + "%") |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
79 elif op == "all": |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
80 words = [] |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
81 for word in val.split(" "): |
7 | 82 words.append(param + " ILIKE %s") |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
83 args.append("%" + word + "%") |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
84 wheres.append("(" + " AND ".join(words) + ")") |
7 | 85 elif op == "one": |
86 words = [] | |
87 for word in val.split(" "): | |
88 words.append(param + " ILIKE %s") | |
89 args.append("%" + word + "%") | |
90 wheres.append("(" + " OR ".join(words) + ")") | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
91 else: |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
92 logging.error("getSQLquery: unknown op=%s!"%op) |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
93 |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
94 return |
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
95 |
4 | 96 # evaluate argv |
97 for (key, val) in argv.items(): | |
98 if key in done: | |
99 # parameter has been processed | |
100 continue | |
101 | |
102 # beginning of a command should always be "_" | |
103 if key[0] == "-": | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
104 key = "_" + key[1:] |
4 | 105 |
106 if key == "_select": | |
107 # SELECT expression | |
108 select = sqlName(val, more="*,") | |
109 elif key == "_sort": | |
110 # sort i.e. ORDER BY expression | |
111 order = sqlName(val, more=",") | |
112 elif key == "_lop": | |
113 # logical operation joining WHERE clauses | |
114 whereOp = sqlName(val) | |
7 | 115 elif key == "max": |
116 # number of results | |
117 limit = sqlName(val) | |
118 elif key == "skip": | |
119 # start at result number | |
120 offset = sqlName(val) | |
4 | 121 elif key[:3] == "_op": |
122 # operation parameters _op_param=option | |
123 param = sqlName(key[4:]) | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
124 logging.debug("param=%s key=%s val=%s"%(param,key,val)) |
4 | 125 if param in argv: |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
126 doOp(val, param, argv[param]) |
4 | 127 done[param] = True |
128 else: | |
129 # no corresponding parameter | |
130 logging.error("getSQLquery: param=%s for op not found!"%param) | |
131 | |
132 # process remaining parameters (without _op) | |
133 for (key, val) in argv.items(): | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
134 if key not in done and key[0] not in ("_", "-"): |
4 | 135 param = sqlName(key) |
136 # default is begins-with | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
137 doOp("bw", param, val) |
4 | 138 |
139 # join it all | |
140 query = "SELECT %s FROM %s"%(select, table) | |
141 if wheres: | |
142 query += " WHERE " + (" " + whereOp + " ").join(wheres) | |
143 | |
144 if order: | |
145 query += " ORDER BY " + order | |
7 | 146 |
147 if limit: | |
148 query += " LIMIT " + limit | |
149 | |
150 if offset: | |
151 query += " OFFSET " + offset | |
4 | 152 |
153 return {'query' : query, 'args' : args} | |
154 | |
155 def ZDBInlineSearch(self, _table=None, **argv): | |
156 """returns result set from search""" | |
5
ca30cf0e810d
fix bugs in ZDBInlineSearch.
root@xserve09.mpiwg-berlin.mpg.de
parents:
4
diff
changeset
|
157 query = self.getSQLQuery(_table, argv) |
4 | 158 result = self.executeZSQL(query['query'], query['args']) |
159 return result | |
3 | 160 |
161 def manage_editZDBInterfaceFolder(self, title=None, connection_id=None, | |
162 REQUEST=None): | |
163 """Change the object""" | |
164 if title is not None: | |
165 self.title = title | |
166 | |
167 if connection_id is not None: | |
168 self.connection_id = connection_id | |
169 | |
170 #checkPermission=getSecurityManager().checkPermission | |
171 REQUEST.RESPONSE.redirect('manage_main') | |
172 | |
173 | |
174 manage_addZDBInterfaceFolderForm=PageTemplateFile('zpt/addZDBInterfaceFolder',globals()) | |
175 | |
176 def manage_addZDBInterfaceFolder(self, id, title='', label='', description='', | |
177 createPublic=0, | |
178 createUserF=0, | |
179 REQUEST=None): | |
180 """Add a new object with id *id*.""" | |
181 | |
182 ob=ZDBInterfaceFolder(str(id),title) | |
183 self._setObject(id, ob) | |
184 | |
185 #checkPermission=getSecurityManager().checkPermission | |
186 REQUEST.RESPONSE.redirect('manage_main') | |
187 | |
188 |