xref: /aosp_15_r20/external/autotest/frontend/db/backends/afe_sqlite/base.py (revision 9c5db1993ded3edbeafc8092d69fe5de2ee02df7)
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