| Alex Miller | 47d6128 | 2013-04-17 13:53:58 -0700 | [diff] [blame] | 1 | import os |
| showard | 364fe86 | 2008-10-17 02:01:16 +0000 | [diff] [blame] | 2 | |
| Alex Miller | 47d6128 | 2013-04-17 13:53:58 -0700 | [diff] [blame] | 3 | import common |
| 4 | |
| 5 | os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'autotest_lib.frontend.settings') |
| showard | fa8629c | 2008-11-04 16:51:23 +0000 | [diff] [blame] | 6 | |
| Jakob Juelich | 8a764d1 | 2014-10-14 19:24:21 -0700 | [diff] [blame] | 7 | def _enable_autocommit_by_name(name): |
| 8 | """Enable autocommit for the connection with matching name. |
| 9 | |
| 10 | @param name: Name of the connection. |
| 11 | """ |
| 12 | from django.db import connections |
| 13 | # ensure a connection is open |
| 14 | connections[name].cursor() |
| 15 | connections[name].connection.autocommit(True) |
| 16 | |
| 17 | |
| showard | fa8629c | 2008-11-04 16:51:23 +0000 | [diff] [blame] | 18 | def enable_autocommit(): |
| Jakob Juelich | 8a764d1 | 2014-10-14 19:24:21 -0700 | [diff] [blame] | 19 | """Enable autocommit for default and global connection. |
| 20 | """ |
| 21 | _enable_autocommit_by_name('default') |
| 22 | _enable_autocommit_by_name('global') |