1"""
2Autotest frontend SQLite based compiler.
3"""
4from django.db.backends.sqlite3.base import DatabaseOperations as SQLiteDatabaseOperations
5from django.db.backends.sqlite3.base import DatabaseWrapper as SQLiteDatabaseWrapper
6
7
8class DatabaseOperations(SQLiteDatabaseOperations):
9    compiler_module = "autotest_lib.frontend.db.backends.afe_sqlite.compiler"
10
11
12class DatabaseWrapper(SQLiteDatabaseWrapper):
13    def __init__(self, *args, **kwargs):
14        super(DatabaseWrapper, self).__init__(*args, **kwargs)
15        self.ops = DatabaseOperations(self.connection)
16