Ticket #6161: oracle_qsrf_patch_v2.diff

branches/queryset-refactor/django/db/backends/oracle/base.py

@@ -93 +93 @@
-        return 30
-
-    def query_set_class(self, DefaultQuerySet):
-from django.db import connection
-from django.db.models.query import EmptyResultSet, GET_ITERATOR_CHUNK_SIZE, quote_only_if_word
+# Getting the DefaultQuery class.
+DefaultQuery = DefaultQuerySet().query.__class__
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+
-            def resolve_columns(self, row, fields=()):
-                from django.db.models.fields import DateField, DateTimeField, \
-
+ 
-                values = []
-                for value, field in map(None, row, fields):
-                    if isinstance(value, Database.LOB):
@@ -325 +143 @@
-                    values.append(value)
-                return values
-
+            def as_sql(self, with_limits=True):
+                """
+                Creates the SQL for this query. Returns the SQL string and list of
+                parameters.  This is overriden from the original Query class to
+                accommodate Oracle's limit/offset SQL.
+                
+                If 'with_limits' is False, any limit/offset information is not included
+                in the query.
+                """
+                # The `do_offset` flag indicates whether we need to construct the
+                # SQL needed to use limit/offset w/Oracle.
+                do_offset = with_limits and (self.high_mark or self.low_mark)
+
+                # If no offsets, just return the result of the base class `as_sql`.
+                if not do_offset:
+                    return super(OracleQuery, self).as_sql(with_limits=False)
+
+                # `get_columns` needs to be called before `get_ordering` to populate
+                # `_select_alias`.
+                self.pre_sql_setup()
+                out_cols = self.get_columns()
+                ordering = self.get_ordering()
+
+                # Getting the "ORDER BY" SQL for the ROW_NUMBER() result.
+                if ordering:
+                    rn_orderby = ', '.join(ordering)
+                else:
+                    # Oracle's ROW_NUMBER() function always requires an order-by clause.
+                    # So we need to define a default order-by, since none was provided.
+                    qn = self.quote_name_unless_alias
+                    opts = self.model._meta
+                    rn_orderby = '%s.%s' % (qn(opts.db_table), qn(opts.fields[0].db_column or opts.fields[0].column))
+
+                # Getting the selection SQL and the params, which has the `rn`
+                # extra selection SQL; we pop `rn` after this completes so we do
+                # not get the attribute on the returned models.
+                self.extra_select['rn'] = 'ROW_NUMBER() OVER (ORDER BY %s )' % rn_orderby
+                sql, params= super(OracleQuery, self).as_sql(with_limits=False)
+                self.extra_select.pop('rn')
+
+                # Constructing the result SQL, using the initial select SQL
+                # obtained above.
+                result = ['SELECT * FROM (%s)' % sql]
+                
+                # Place WHERE condition on `rn` for the desired range.
+                result.append('WHERE rn > %d' % self.low_mark)
+                if self.high_mark:
+                    result.append('AND rn <= %d' % self.high_mark)
+
+                # Returning the SQL w/params.
+                return ' '.join(result), params
+              
+        from django.db import connection
+        class OracleQuerySet(DefaultQuerySet):
+            "The OracleQuerySet is overriden to use OracleQuery."
+            def __init__(self, model=None, query=None):
+                super(OracleQuerySet, self).__init__(model=model, query=query)
+                self.query = query or OracleQuery(self.model, connection)
+
-        return OracleQuerySet
-
-    def quote_name(self, name):
@@ -446 +323 @@
-    charset = 'utf-8'
-
-    def _format_params(self, params):
+        sz_kwargs, result = {}, {}
-        if isinstance(params, dict):
-            result = {}
-            charset = self.charset
-            for key, value in params.items():
-                result[smart_str(key, charset)] = smart_str(value, charset)
-return result
+    if hasattr(value, 'oracle_type'): sz_kwargs[key] = value.oracle_type()
-        else:
-
+
+
+
+
-
+        # If any of the parameters had an `oracle_type` method, then we set
+        # the inputsizes for those parameters using the returned type
+        if sz_kwargs: self.setinputsizes(**sz_kwargs)
+        return result
+
-    def execute(self, query, params=None):
-        if params is None:
-            params = []
@@ -472 +357 @@
-
-    def executemany(self, query, params=None):
-        try:
-
+  
-        except (IndexError, TypeError):
-
-
+  
+  
-        # cx_Oracle wants no trailing ';' for SQL statements.  For PL/SQL, it
-        # it does want a trailing ';' but not a trailing '/'.  However, these
-        # characters must be included in the original query in case the query