1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
|
# sql.py
# Copyright (C) 2005 Michael Bayer mike_mp@zzzcomputing.com
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""defines the base components of SQL expression trees."""
import sqlalchemy.schema as schema
import sqlalchemy.util as util
import sqlalchemy.types as types
import string
__ALL__ = ['textclause', 'select', 'join', 'and_', 'or_', 'union', 'desc', 'asc', 'outerjoin', 'alias', 'subquery', 'bindparam', 'sequence']
def desc(column):
"""returns a descending ORDER BY clause element"""
return CompoundClause(None, column, "DESC")
def asc(column):
"""returns an ascending ORDER BY clause element"""
return CompoundClause(None, column, "ASC")
def outerjoin(left, right, onclause, **kwargs):
"""returns an OUTER JOIN clause element, given the left and right hand expressions,
as well as the ON condition's expression. When chaining joins together, the previous JOIN
expression should be specified as the left side of this JOIN expression."""
return Join(left, right, onclause, isouter = True, **kwargs)
def join(left, right, onclause, **kwargs):
"""returns a JOIN clause element (regular inner join), given the left and right hand expressions,
as well as the ON condition's expression. When chaining joins together, the previous JOIN
expression should be specified as the left side of this JOIN expression."""
return Join(left, right, onclause, **kwargs)
def select(columns, whereclause = None, from_obj = [], **kwargs):
"""returns a SELECT clause element, given a list of columns and/or selectable items to select
columns from, an optional expression for the WHERE clause, an optional list of "FROM" objects
to select from, and additional parameters."""
return Select(columns, whereclause = whereclause, from_obj = from_obj, **kwargs)
def insert(table, values = None, **kwargs):
"""returns an INSERT clause element.
'table' is the table to be inserted into.
'values' is a dictionary which specifies the column specifications of the INSERT, and is optional.
If left as None, the
column specifications are determined from the bind parameters used during the compile phase of the
INSERT statement. If the bind parameters also are None during the compile phase, then the column
specifications will be generated from the full list of table columns.
If both 'values' and compile-time bind parameters are present, the compile-time bind parameters
override the information specified within 'values' on a per-key basis.
The keys within 'values' can be either Column objects or their string identifiers.
Each key may reference one of: a literal data value (i.e. string, number, etc.), a Column object,
or a SELECT statement. If a SELECT statement is specified which references this INSERT statement's
table, the statement will be correlated against the INSERT statement.
"""
return Insert(table, values, **kwargs)
def update(table, whereclause = None, values = None, **kwargs):
"""returns an UPDATE clause element.
'table' is the table to be updated.
'whereclause' is a ClauseElement describing the WHERE condition of the UPDATE statement.
'values' is a dictionary which specifies the SET conditions of the UPDATE, and is optional.
If left as None, the
SET conditions are determined from the bind parameters used during the compile phase of the
UPDATE statement. If the bind parameters also are None during the compile phase, then the SET
conditions will be generated from the full list of table columns.
If both 'values' and compile-time bind parameters are present, the compile-time bind parameters
override the information specified within 'values' on a per-key basis.
The keys within 'values' can be either Column objects or their string identifiers.
Each key may reference one of: a literal data value (i.e. string, number, etc.), a Column object,
or a SELECT statement. If a SELECT statement is specified which references this UPDATE statement's
table, the statement will be correlated against the UPDATE statement.
"""
return Update(table, whereclause, values, **kwargs)
def delete(table, whereclause = None, **kwargs):
return Delete(table, whereclause, **kwargs)
def and_(*clauses):
return _compound_clause('AND', *clauses)
def or_(*clauses):
clause = _compound_clause('OR', *clauses)
return clause
def exists(*args, **params):
s = select(*args, **params)
return BinaryClause(TextClause("EXISTS"), s, None)
def union(*selects, **params):
return _compound_select('UNION', *selects, **params)
def alias(*args, **params):
return Alias(*args, **params)
def subquery(alias, *args, **params):
return Alias(Select(*args, **params), alias)
def bindparam(key, value = None, type=None):
if isinstance(key, schema.Column):
return BindParamClause(key.name, value, type=key.type)
else:
return BindParamClause(key, value, type=type)
def text(text):
return TextClause(text)
def null():
return Null()
def sequence():
return Sequence()
def _compound_clause(keyword, *clauses):
return CompoundClause(keyword, *clauses)
def _compound_select(keyword, *selects, **params):
if len(selects) == 0:
return None
s = selects[0]
for n in selects[1:]:
s.append_clause(keyword, n)
if params.get('order_by', None) is not None:
s.order_by(*params['order_by'])
return s
def _is_literal(element):
return not isinstance(element, ClauseElement) and not isinstance(element, schema.SchemaItem)
class ClauseVisitor(schema.SchemaVisitor):
"""builds upon SchemaVisitor to define the visiting of SQL statement elements in
addition to Schema elements."""
def visit_columnclause(self, column):pass
def visit_fromclause(self, fromclause):pass
def visit_bindparam(self, bindparam):pass
def visit_textclause(self, textclause):pass
def visit_compound(self, compound):pass
def visit_binary(self, binary):pass
def visit_alias(self, alias):pass
def visit_select(self, select):pass
def visit_join(self, join):pass
def visit_null(self, null):pass
class Compiled(ClauseVisitor):
"""represents a compiled SQL expression. the __str__ method of the Compiled object
should produce the actual text of the statement. Compiled objects are specific to the database
library that created them, and also may or may not be specific to the columns referenced
within a particular set of bind parameters. In no case should the Compiled object be dependent
on the actual values of those bind parameters, even though it may reference those values
as defaults."""
def __init__(self, engine, statement, bindparams):
self.engine = engine
self.bindparams = bindparams
self.statement = statement
def __str__(self):
"""returns the string text of the generated SQL statement."""
raise NotImplementedError()
def get_params(self, **params):
"""returns the bind params for this compiled object, with values overridden by
those given in the **params dictionary"""
raise NotImplementedError()
def execute(self, *multiparams, **params):
"""executes this compiled object using the underlying SQLEngine"""
if len(multiparams):
params = [self.get_params(**m) for m in multiparams]
else:
params = self.get_params(**params)
return self.engine.execute(str(self), params, compiled = self, typemap = self.typemap)
class ClauseElement(object):
"""base class for elements of a programmatically constructed SQL expression.
includes a list of 'from objects' which collects items to be placed
in the FROM clause of a SQL statement.
when many ClauseElements are attached together, the from objects and bind
parameters are scooped up into the enclosing-most ClauseElement.
"""
def hash_key(self):
"""returns a string that uniquely identifies the concept this ClauseElement represents.
two ClauseElements can have the same value for hash_key() iff they both correspond to the
exact same generated SQL. This allows the hash_key() values of a collection of ClauseElements
to be constructed into a larger identifying string for the purpose of caching a SQL expression.
Note that since ClauseElements may be mutable, the hash_key() value is subject to change
if the underlying structure of the ClauseElement changes."""
raise NotImplementedError(repr(self))
def _get_from_objects(self):
raise NotImplementedError(repr(self))
def accept_visitor(self, visitor):
raise NotImplementedError(repr(self))
def copy_container(self):
"""should return a copy of this ClauseElement, iff this ClauseElement contains other
ClauseElements. Otherwise, it should be left alone to return self. This is used to create
copies of expression trees that still reference the same "leaf nodes". The new structure
can then be restructured without affecting the original."""
return self
def compile(self, engine = None, bindparams = None):
"""compiles this SQL expression using its underlying SQLEngine to produce
a Compiled object. If no engine can be found, an ansisql engine is used.
bindparams is a dictionary representing the default bind parameters to be used with
the statement. """
if engine is None:
for f in self._get_from_objects():
engine = f.engine
if engine is not None: break
else:
import sqlalchemy.ansisql as ansisql
engine = ansisql.engine()
#raise "no engine supplied, and no engine could be located within the clauses!"
return engine.compile(self, bindparams = bindparams)
def __str__(self):
return str(self.compile())
def execute(self, *multiparams, **params):
"""compiles and executes this SQL expression using its underlying SQLEngine.
the given **params are used as bind parameters when compiling and executing the expression.
the DBAPI cursor object is returned."""
e = self.engine
if len(multiparams):
bindparams = multiparams[0]
else:
bindparams = params
c = self.compile(e, bindparams = bindparams)
return c.execute(*multiparams, **params)
def result(self, **params):
"""the same as execute(), except a RowProxy object is returned instead of a DBAPI cursor."""
raise NotImplementedError()
class ColumnClause(ClauseElement):
"""represents a textual column clause in a SQL statement."""
def __init__(self, text, selectable):
self.text = text
self.table = selectable
self._impl = ColumnSelectable(self)
self.type = types.NullTypeEngine()
columns = property(lambda self: [self])
name = property(lambda self:self.text)
key = property(lambda self:self.text)
label = property(lambda self:self.text)
fullname = property(lambda self:self.text)
def accept_visitor(self, visitor):
visitor.visit_columnclause(self)
def hash_key(self):
return "ColumnClause(%s, %s)" % (self.text, self.table.hash_key())
def _get_from_objects(self):
return []
def _make_proxy(self, selectable, name = None):
c = ColumnClause(self.text or name, selectable)
selectable.columns[c.key] = c
c._impl = ColumnSelectable(c)
return c
class FromClause(ClauseElement):
"""represents a FROM clause element in a SQL statement."""
def __init__(self, from_name = None, from_key = None):
self.from_name = from_name
self.id = from_key or from_name
def _get_from_objects(self):
# this could also be [self], at the moment it doesnt matter to the Select object
return []
engine = property(lambda s: None)
def hash_key(self):
return "FromClause(%s, %s)" % (repr(self.id), repr(self.from_name))
def accept_visitor(self, visitor):
visitor.visit_fromclause(self)
class BindParamClause(ClauseElement):
def __init__(self, key, value, shortname = None, type = None):
self.key = key
self.value = value
self.shortname = shortname
self.type = type or types.NULLTYPE
def accept_visitor(self, visitor):
visitor.visit_bindparam(self)
def _get_from_objects(self):
return []
def hash_key(self):
return "BindParam(%s, %s, %s)" % (repr(self.key), repr(self.value), repr(self.shortname))
def typeprocess(self, value):
return self.type.convert_bind_param(value)
class TextClause(ClauseElement):
"""represents any plain text WHERE clause or full SQL statement"""
def __init__(self, text = ""):
self.text = text
self.parens = False
def accept_visitor(self, visitor):
visitor.visit_textclause(self)
def hash_key(self):
return "TextClause(%s)" % repr(self.text)
def _get_from_objects(self):
return []
class Null(ClauseElement):
def accept_visitor(self, visitor):
visitor.visit_null(self)
def _get_from_objects(self):
return []
def hash_key(self):
return "Null"
class CompoundClause(ClauseElement):
"""represents a list of clauses joined by an operator"""
def __init__(self, operator, *clauses):
self.operator = operator
self.fromobj = []
self.clauses = []
self.parens = False
for c in clauses:
if c is None: continue
self.append(c)
def copy_container(self):
clauses = [clause.copy_container() for clause in self.clauses]
return CompoundClause(self.operator, *clauses)
def append(self, clause):
if _is_literal(clause):
clause = TextClause(str(clause))
elif isinstance(clause, CompoundClause):
clause.parens = True
self.clauses.append(clause)
self.fromobj += clause._get_from_objects()
def accept_visitor(self, visitor):
for c in self.clauses:
c.accept_visitor(visitor)
visitor.visit_compound(self)
def _get_from_objects(self):
return self.fromobj
def hash_key(self):
return string.join([c.hash_key() for c in self.clauses], self.operator)
class ClauseList(ClauseElement):
def __init__(self, *clauses):
self.clauses = clauses
def accept_visitor(self, visitor):
for c in self.clauses:
c.accept_visitor(visitor)
visitor.visit_clauselist(self)
def _get_from_objects(self):
return []
class BinaryClause(ClauseElement):
"""represents two clauses with an operator in between"""
def __init__(self, left, right, operator):
self.left = left
self.right = right
self.operator = operator
self.parens = False
def copy_container(self):
return BinaryClause(self.left.copy_container(), self.right.copy_container(), self.operator)
def _get_from_objects(self):
return self.left._get_from_objects() + self.right._get_from_objects()
def hash_key(self):
return self.left.hash_key() + self.operator + self.right.hash_key()
def accept_visitor(self, visitor):
self.left.accept_visitor(visitor)
self.right.accept_visitor(visitor)
visitor.visit_binary(self)
def swap(self):
c = self.left
self.left = self.right
self.right = c
class Selectable(FromClause):
"""represents a column list-holding object, like a table or subquery. can be used anywhere
a Table is used."""
c = property(lambda self: self.columns)
def accept_visitor(self, visitor):
raise NotImplementedError()
def select(self, whereclauses = None, **params):
raise NotImplementedError()
def join(self, right, *args, **kwargs):
return Join(self, right, *args, **kwargs)
def outerjoin(self, right, *args, **kwargs):
return Join(self, right, isouter = True, *args, **kwargs)
def group_parenthesized(self):
"""indicates if this Selectable requires parenthesis when grouped into a compound statement"""
return True
class Join(Selectable):
# TODO: put "using" + "natural" concepts in here and make "onclause" optional
def __init__(self, left, right, onclause, isouter = False, allcols = True):
self.left = left
self.right = right
self.id = self.left.id + "_" + self.right.id
self.allcols = allcols
if allcols:
self.columns = [c for c in self.left.columns] + [c for c in self.right.columns]
else:
self.columns = self.right.columns
# TODO: if no onclause, do NATURAL JOIN
self.onclause = onclause
self.isouter = isouter
primary_keys = property (lambda self: [c for c in self.left.columns if c.primary_key] + [c for c in self.right.columns if c.primary_key])
def group_parenthesized(self):
"""indicates if this Selectable requires parenthesis when grouped into a compound statement"""
return False
def hash_key(self):
return "Join(%s, %s, %s, %s)" % (repr(self.left.hash_key()), repr(self.right.hash_key()), repr(self.onclause.hash_key()), repr(self.isouter))
def select(self, whereclauses = None, **params):
return select([self.left, self.right], and_(self.onclause, whereclauses), **params)
def accept_visitor(self, visitor):
self.left.accept_visitor(visitor)
self.right.accept_visitor(visitor)
self.onclause.accept_visitor(visitor)
visitor.visit_join(self)
engine = property(lambda s:s.left.engine or s.right.engine)
def _get_from_objects(self):
m = {}
for x in self.onclause._get_from_objects():
m[x.id] = x
result = [self] + [FromClause(from_key = c.id) for c in self.left._get_from_objects() + self.right._get_from_objects()]
for x in result:
m[x.id] = x
result = m.values()
return result
class Alias(Selectable):
def __init__(self, selectable, alias = None):
self.selectable = selectable
self.columns = util.OrderedProperties()
if alias is None:
alias = id(self)
self.name = alias
self.id = self.name
self.count = 0
for co in selectable.columns:
co._make_proxy(self)
primary_keys = property (lambda self: [c for c in self.columns if c.primary_key])
def hash_key(self):
return "Alias(%s, %s)" % (repr(self.selectable.hash_key()), repr(self.name))
def accept_visitor(self, visitor):
self.selectable.accept_visitor(visitor)
visitor.visit_alias(self)
def _get_from_objects(self):
return [self]
def group_parenthesized(self):
return False
engine = property(lambda s: s.selectable.engine)
def select(self, whereclauses = None, **params):
return select([self], whereclauses, **params)
class ColumnSelectable(Selectable):
"""Selectable implementation that gets attached to a schema.Column object."""
def __init__(self, column):
self.column = column
self.name = column.name
self.columns = [self.column]
if column.table.name:
self.label = column.table.name + "_" + self.column.name
self.fullname = column.table.name + "." + self.column.name
else:
self.label = self.column.name
self.fullname = self.column.name
def copy_container(self):
return self.column
def group_parenthesized(self):
return False
def _get_from_objects(self):
return [self.column.table]
def _compare(self, operator, obj):
if _is_literal(obj):
if obj is None:
if operator != '=':
raise "Only '=' operator can be used with NULL"
return BinaryClause(self.column, null(), 'IS')
elif self.column.table.name is None:
obj = BindParamClause(self.name, obj, shortname = self.name, type = self.column.type)
else:
obj = BindParamClause(self.column.table.name + "_" + self.name, obj, shortname = self.name, type = self.column.type)
return BinaryClause(self.column, obj, operator)
def __lt__(self, other):
return self._compare('<', other)
def __le__(self, other):
return self._compare('<=', other)
def __eq__(self, other):
return self._compare('=', other)
def __ne__(self, other):
return self._compare('!=', other)
def __gt__(self, other):
return self._compare('>', other)
def __ge__(self, other):
return self._compare('>=', other)
def like(self, other):
return self._compare('LIKE', other)
def in_(self, *other):
if _is_literal(other[0]):
return self._compare('IN', CompoundClause(',', other))
else:
return self._compare('IN', union(*other))
def startswith(self, other):
return self._compare('LIKE', str(other) + "%")
def endswith(self, other):
return self._compare('LIKE', "%" + str(other))
class TableImpl(Selectable):
"""attached to a schema.Table to provide it with a Selectable interface
as well as other functions
"""
def __init__(self, table):
self.table = table
self.id = self.table.name
def get_from_text(self):
return self.table.name
def group_parenthesized(self):
return False
def join(self, right, *args, **kwargs):
return Join(self.table, right, *args, **kwargs)
def outerjoin(self, right, *args, **kwargs):
return Join(self.table, right, isouter = True, *args, **kwargs)
def select(self, whereclauses = None, **params):
return select([self.table], whereclauses, **params)
def insert(self, select = None):
return insert(self.table, select = select)
def update(self, whereclause = None, values = None):
return update(self.table, whereclause, values)
def delete(self, whereclause = None):
return delete(self.table, whereclause)
columns = property(lambda self: self.table.columns)
def _get_from_objects(self):
return [self.table]
def create(self, **params):
self.table.engine.create(self.table)
def drop(self, **params):
self.table.engine.drop(self.table)
class Select(Selectable):
"""finally, represents a SELECT statement, with appendable clauses, as well as
the ability to execute itself and return a result set."""
def __init__(self, columns, whereclause = None, from_obj = [], group_by = None, order_by = None, use_labels = False, engine = None):
self.columns = util.OrderedProperties()
self.froms = util.OrderedDict()
self.use_labels = use_labels
self.id = id(self)
self.name = None
self.whereclause = whereclause
self._engine = engine
# indicates if this select statement is a subquery inside of a WHERE clause
# note this is different from a subquery inside the FROM list
self.issubquery = False
self._text = None
self._raw_columns = []
self._clauses = []
for c in columns:
self.append_column(c)
if whereclause is not None:
self.set_whereclause(whereclause)
for f in from_obj:
self.append_from(f)
if group_by:
self.append_clause("GROUP_BY", group_by)
if order_by:
self.order_by(*order_by)
def append_column(self, column):
if _is_literal(column):
column = ColumnClause(str(column), self)
self._raw_columns.append(column)
for f in column._get_from_objects():
self.froms.setdefault(f.id, f)
for co in column.columns:
if self.use_labels:
co._make_proxy(self, name = co.label)
else:
co._make_proxy(self)
def set_whereclause(self, whereclause):
if type(whereclause) == str:
self.whereclause = TextClause(whereclause)
class CorrelatedVisitor(ClauseVisitor):
def visit_select(s, select):
for f in self.froms.keys():
select.clear_from(f)
select.issubquery = True
self.whereclause.accept_visitor(CorrelatedVisitor())
for f in self.whereclause._get_from_objects():
self.froms.setdefault(f.id, f)
def clear_from(self, id):
self.append_from(FromClause(from_name = None, from_key = id))
def append_from(self, fromclause):
if type(fromclause) == str:
fromclause = FromClause(from_name = fromclause)
self.froms[fromclause.id] = fromclause
for r in fromclause._get_from_objects():
self.froms[r.id] = r
def append_clause(self, keyword, clause):
if type(clause) == str:
clause = TextClause(clause)
self._clauses.append((keyword, clause))
def compile(self, engine = None, bindparams = None):
if engine is None:
if self.engine is None:
for f in self.froms.values():
self.engine = f.engine
if self.engine is not None: break
engine = self.engine
if engine is None:
raise "no engine supplied, and no engine could be located within the clauses!"
return engine.compile(self, bindparams)
def accept_visitor(self, visitor):
for f in self.froms.values():
f.accept_visitor(visitor)
if self.whereclause is not None:
self.whereclause.accept_visitor(visitor)
for tup in self._clauses:
tup[1].accept_visitor(visitor)
visitor.visit_select(self)
def order_by(self, *clauses):
self.append_clause("ORDER BY", ClauseList(*clauses))
def select(self, whereclauses = None, **params):
return select([self], whereclauses, **params)
def _find_engine(self):
"""tries to return a SQLEngine, either explicitly set in this object, or searched
within the from clauses for one"""
if self._engine:
return self._engine
for f in self.froms.values():
e = f.engine
if e:
return e
return None
engine = property(lambda s: s._find_engine())
def _get_from_objects(self):
if self.issubquery:
return []
else:
return [self]
class UpdateBase(ClauseElement):
"""forms the base for INSERT, UPDATE, and DELETE statements.
Deals with the special needs of INSERT and UPDATE parameter lists -
these statements have two separate lists of parameters, those
defined when the statement is constructed, and those specified at compile time."""
def _process_colparams(self, parameters):
if parameters is None:
return None
for key in parameters.keys():
value = parameters[key]
if isinstance(value, Select):
value.clear_from(self.table.id)
elif _is_literal(value):
if _is_literal(key):
col = self.table.c[key]
else:
col = key
try:
parameters[key] = bindparam(col, value)
except KeyError:
del parameters[key]
return parameters
def get_colparams(self, parameters):
# case one: no parameters in the statement, no parameters in the
# compiled params - just return binds for all the table columns
if parameters is None and self.parameters is None:
return [(c, bindparam(c.name, type=c.type)) for c in self.table.columns]
# if we have statement parameters - set defaults in the
# compiled params
if parameters is None:
parameters = {}
else:
parameters = parameters.copy()
if self.parameters is not None:
for k, v in self.parameters.iteritems():
parameters.setdefault(k, v)
# now go thru compiled params, get the Column object for each key
d = {}
for key, value in parameters.iteritems():
if isinstance(key, schema.Column):
d[key] = value
else:
try:
d[self.table.columns[str(key)]] = value
except AttributeError:
pass
# create a list of column assignment clauses as tuples
values = []
for c in self.table.columns:
if d.has_key(c):
value = d[c]
if _is_literal(value):
value = bindparam(c.name, value, type=c.type)
values.append((c, value))
return values
def compile(self, engine = None, bindparams = None):
if engine is None:
engine = self.engine
if engine is None:
raise "no engine supplied, and no engine could be located within the clauses!"
return engine.compile(self, bindparams)
class Insert(UpdateBase):
def __init__(self, table, parameters = None, **params):
self.table = table
self.select = None
self.parameters = self._process_colparams(parameters)
self.engine = self.table.engine
def accept_visitor(self, visitor):
if self.select is not None:
self.select.accept_visitor(visitor)
visitor.visit_insert(self)
class Update(UpdateBase):
def __init__(self, table, whereclause, parameters = None, **params):
self.table = table
self.whereclause = whereclause
self.parameters = self._process_colparams(parameters)
self.engine = self.table.engine
def accept_visitor(self, visitor):
if self.whereclause is not None:
self.whereclause.accept_visitor(visitor)
visitor.visit_update(self)
class Delete(UpdateBase):
def __init__(self, table, whereclause, **params):
self.table = table
self.whereclause = whereclause
self.engine = self.table.engine
def accept_visitor(self, visitor):
if self.whereclause is not None:
self.whereclause.accept_visitor(visitor)
visitor.visit_delete(self)
class Sequence(BindParamClause):
def __init__(self):
BindParamClause.__init__(self, 'sequence')
def accept_visitor(self, visitor):
visitor.visit_sequence(self)
|