Skip to content

Commit

Permalink
Merge pull request #618 from ProtixIT/joinqueryset-quoted
Browse files Browse the repository at this point in the history
Remove `JoinQueryset.get_quoted_query()`
  • Loading branch information
foarsitter authored May 2, 2024
2 parents ba9ac5d + 2b2110f commit f20c394
Showing 1 changed file with 2 additions and 17 deletions.
19 changes: 2 additions & 17 deletions model_utils/managers.py
Original file line number Diff line number Diff line change
Expand Up @@ -293,21 +293,6 @@ class SoftDeletableManager(SoftDeletableManagerMixin, models.Manager):

class JoinQueryset(models.QuerySet):

def get_quoted_query(self, query):
query, params = query.sql_with_params()

# Put additional quotes around string.
params = [
f'\'{p}\''
if isinstance(p, str) else p
for p in params
]

# Cast list of parameters to tuple because I got
# "not enough format characters" otherwise.
params = tuple(params)
return query % params

def join(self, qs=None):
'''
Join one queryset together with another using a temporary table. If
Expand Down Expand Up @@ -349,7 +334,7 @@ def join(self, qs=None):
new_qs = self.model.objects.all()

TABLE_NAME = 'temp_stuff'
query = self.get_quoted_query(qs.query)
query, params = qs.query.sql_with_params()
sql = '''
DROP TABLE IF EXISTS {table_name};
DROP INDEX IF EXISTS {table_name}_id;
Expand All @@ -358,7 +343,7 @@ def join(self, qs=None):
'''.format(table_name=TABLE_NAME, fk_column=fk_column, query=str(query))

with connection.cursor() as cursor:
cursor.execute(sql)
cursor.execute(sql, params)

class TempModel(models.Model):
temp_key = models.ForeignKey(
Expand Down

0 comments on commit f20c394

Please sign in to comment.