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