Ticket #7231: extra-join.diff

django/db/models/sql/query.py

@@ -82 +82 @@
         self.extra_where = ()
         self.extra_params = ()
         self.extra_order_by = ()
+        self.extra_join = ()
 
     def __str__(self):
         """
@@ -182 +183 @@
         obj.extra_where = self.extra_where
         obj.extra_params = self.extra_params
         obj.extra_order_by = self.extra_order_by
+        obj.extra_join = self.extra_join
         obj.__dict__.update(kwargs)
         if hasattr(obj, '_setup_query'):
             obj._setup_query()
@@ -261 +263 @@
         result.append('FROM')
         result.extend(from_)
         params.extend(f_params)
+        
+        if self.extra_join:
+            result.append(' '.join(self.extra_join))
 
         if where:
             result.append('WHERE %s' % where)
@@ -1379 +1384 @@
         self.related_select_cols = []
         self.related_select_fields = []
 
-    def add_extra(self, select, select_params, where, params, tables, order_by):
+    def add_extra(self, select, select_params, where, params, tables, order_by, join):
         """
         Adds data to the various extra_* attributes for user-created additions
         to the query.
@@ -1401 +1406 @@
             self.extra_tables += tuple(tables)
         if order_by:
             self.extra_order_by = order_by
+        if join:
+            self.extra_join += tuple(join)
 
     def trim_extra_select(self, names):
         """

django/db/models/query.py

@@ -448 +448 @@
         return obj
 
     def extra(self, select=None, where=None, params=None, tables=None,
-            order_by=None, select_params=None):
+            order_by=None, select_params=None, join=None):
         """
         Add extra SQL fragments to the query.
         """
         assert self.query.can_filter(), \
                 "Cannot change a query once a slice has been taken"
         clone = self._clone()
-        clone.query.add_extra(select, select_params, where, params, tables, order_by)
+        clone.query.add_extra(select, select_params, where, params, tables, order_by, join)
         return clone
 
     def reverse(self):