1*9c5db199SXin Liimport logging 2*9c5db199SXin Liimport os 3*9c5db199SXin Li 4*9c5db199SXin Liimport common 5*9c5db199SXin Li 6*9c5db199SXin Lios.environ.setdefault('DJANGO_SETTINGS_MODULE', 'autotest_lib.frontend.settings') 7*9c5db199SXin Li 8*9c5db199SXin Lidef _enable_autocommit_by_name(name): 9*9c5db199SXin Li """Enable autocommit for the connection with matching name. 10*9c5db199SXin Li 11*9c5db199SXin Li @param name: Name of the connection. 12*9c5db199SXin Li """ 13*9c5db199SXin Li from django.db import connections 14*9c5db199SXin Li # ensure a connection is open 15*9c5db199SXin Li logging.info('Enabling autocommit for %s(%s)', 16*9c5db199SXin Li name, connections[name].settings_dict['HOST']) 17*9c5db199SXin Li connections[name].cursor() 18*9c5db199SXin Li connections[name].connection.autocommit(True) 19*9c5db199SXin Li 20*9c5db199SXin Li 21*9c5db199SXin Lidef enable_autocommit(): 22*9c5db199SXin Li """Enable autocommit for default and global connection. 23*9c5db199SXin Li """ 24*9c5db199SXin Li _enable_autocommit_by_name('default') 25*9c5db199SXin Li _enable_autocommit_by_name('global') 26