blob: 99714155876b18825536ed0a8e0804350b4f7765 [file] [log] [blame]
from django.db.backends.mysql import compiler as mysql_compiler
from autotest_lib.frontend.afe.model_logic import _quote_name
class SQLCompiler(mysql_compiler.SQLCompiler):
def get_from_clause(self):
from_, params = super(SQLCompiler, self).get_from_clause()
if hasattr(self.query, "_custom_joins"):
for join_dict in self.query._custom_joins:
from_.append('%s %s AS %s ON (%s)'
% (join_dict['join_type'],
_quote_name(join_dict['table']),
_quote_name(join_dict['alias']),
join_dict['condition']))
params.extend(join_dict['condition_values'])
return from_, params
class SQLInsertCompiler(mysql_compiler.SQLInsertCompiler, SQLCompiler):
pass
class SQLDeleteCompiler(mysql_compiler.SQLDeleteCompiler, SQLCompiler):
pass
class SQLUpdateCompiler(mysql_compiler.SQLUpdateCompiler, SQLCompiler):
pass
class SQLAggregateCompiler(mysql_compiler.SQLAggregateCompiler, SQLCompiler):
pass
class SQLDateCompiler(mysql_compiler.SQLDateCompiler, SQLCompiler):
pass