242 |
result.append(row[0]) |
result.append(row[0]) |
243 |
return result |
return result |
244 |
|
|
245 |
|
def Width(self, col): |
246 |
|
"""Return the maximum width of values in the column |
247 |
|
|
248 |
|
The return value is the the maximum length of string representation |
249 |
|
of the values in the column (represented by index or name).""" |
250 |
|
max = 0 |
251 |
|
|
252 |
|
type = self.column_map[col].type |
253 |
|
iname = self.column_map[col].internal_name |
254 |
|
cursor = self.db.cursor() |
255 |
|
cursor.execute("SELECT %s FROM %s;" % (iname, self.tablename)) |
256 |
|
values = [ i[0] for i in cursor.fetchall()] |
257 |
|
if not values: |
258 |
|
return None |
259 |
|
|
260 |
|
if type == sql_type_map[table.FIELDTYPE_DOUBLE]: |
261 |
|
format = "%.12f" |
262 |
|
elif type == sql_type_map[table.FIELDTYPE_INT]: |
263 |
|
format = "%d" |
264 |
|
else: |
265 |
|
format = "%s" |
266 |
|
for value in values: |
267 |
|
if value is None: continue |
268 |
|
l = len(format % value) |
269 |
|
if l > max: |
270 |
|
max = l |
271 |
|
|
272 |
|
return max |
273 |
|
|
274 |
def SimpleQuery(self, left, comparison, right): |
def SimpleQuery(self, left, comparison, right): |
275 |
"""Return the indices of all rows that matching a condition. |
"""Return the indices of all rows that matching a condition. |
276 |
|
|
410 |
|
|
411 |
self.right_table.ensure_index(self.right_field) |
self.right_table.ensure_index(self.right_field) |
412 |
|
|
413 |
|
# determine the internal column names to join on before |
414 |
|
# coalescing the column information because if the external |
415 |
|
# column names are the same they will be mapped to the same |
416 |
|
# internal name afterwards. |
417 |
|
internal_left_col = self.left_table.orig_to_internal[self.left_field] |
418 |
|
internal_right_col = self.right_table.orig_to_internal[self.right_field] |
419 |
|
|
420 |
# Coalesce the column information |
# Coalesce the column information |
421 |
visited = {} |
visited = {} |
422 |
columns = [] |
columns = [] |
423 |
for col in self.left_table.columns + self.right_table.columns: |
newcolumns = [] |
424 |
|
for table, col in ( |
425 |
|
[ (self.left_table.tablename, c) for c in self.left_table.columns ] |
426 |
|
+ [ (self.right_table.tablename, c) for c in self.right_table.columns]): |
427 |
if col.name in visited: |
if col.name in visited: |
428 |
# We can't allow multiple columns with the same original |
if col.name == self.left_field: |
429 |
# name, so omit this one. FIXME: There should be a |
continue |
430 |
# better solution. |
else: |
431 |
continue |
# We can't allow multiple columns with the same original |
432 |
columns.append(col) |
# name, so append '_' to this one until it is unique. |
433 |
TransientTableBase.create(self, columns) |
# FIXME: There should be a better solution. |
434 |
|
while col.name in visited: |
435 |
|
col.name = col.name + '_' |
436 |
|
columns.append((table, col)) |
437 |
|
newcol = ColumnReference(col.name, col.type, |
438 |
|
"Col%03d" % (len(newcolumns)+1)) |
439 |
|
newcolumns.append(newcol) |
440 |
|
visited[col.name] = 1 |
441 |
|
TransientTableBase.create(self, newcolumns) |
442 |
|
|
443 |
# Copy the joined data to the table. |
# Copy the joined data to the table. |
444 |
internal_names = [col.internal_name for col in self.columns] |
newinternal_names = [col.internal_name for col in self.columns] |
445 |
|
internal_references = ["%s.%s" % (table, col.internal_name) |
446 |
|
for table, col in columns] |
447 |
if self.outer_join: |
if self.outer_join: |
448 |
join_operator = 'LEFT OUTER JOIN' |
join_operator = 'LEFT OUTER JOIN' |
449 |
else: |
else: |
450 |
join_operator = 'JOIN' |
join_operator = 'JOIN' |
451 |
stmt = ("INSERT INTO %s (id, %s) SELECT %s.id, %s FROM %s" |
stmt = ("INSERT INTO %s (id, %s) SELECT %s.id, %s FROM %s" |
452 |
" %s %s ON %s = %s;" |
" %s %s ON %s.%s = %s.%s;" |
453 |
% (self.tablename, |
% (self.tablename, |
454 |
", ".join(internal_names), |
", ".join(newinternal_names), |
455 |
self.left_table.tablename, |
self.left_table.tablename, |
456 |
", ".join(internal_names), |
", ".join(internal_references), |
457 |
self.left_table.tablename, |
self.left_table.tablename, |
458 |
join_operator, |
join_operator, |
459 |
self.right_table.tablename, |
self.right_table.tablename, |
460 |
self.orig_to_internal[self.left_field], |
self.left_table.tablename, |
461 |
self.orig_to_internal[self.right_field])) |
internal_left_col, |
462 |
|
self.right_table.tablename, |
463 |
|
internal_right_col)) |
464 |
self.db.execute(stmt) |
self.db.execute(stmt) |
465 |
|
|
466 |
def Dependencies(self): |
def Dependencies(self): |
556 |
def Dependencies(self): |
def Dependencies(self): |
557 |
"""Return a tuple containing the original table""" |
"""Return a tuple containing the original table""" |
558 |
return (self.table,) |
return (self.table,) |
559 |
|
|
560 |
|
def Width(self, col): |
561 |
|
return self.table.Width(col) |