176 # disable notification during migration |
176 # disable notification during migration |
177 with self.cnx.allow_all_hooks_but('notification'): |
177 with self.cnx.allow_all_hooks_but('notification'): |
178 super(ServerMigrationHelper, self).migrate(vcconf, toupgrade, options) |
178 super(ServerMigrationHelper, self).migrate(vcconf, toupgrade, options) |
179 |
179 |
180 def cmd_process_script(self, migrscript, funcname=None, *args, **kwargs): |
180 def cmd_process_script(self, migrscript, funcname=None, *args, **kwargs): |
181 with self.cnx.ensure_cnx_set: |
181 try: |
182 try: |
182 return super(ServerMigrationHelper, self).cmd_process_script( |
183 return super(ServerMigrationHelper, self).cmd_process_script( |
183 migrscript, funcname, *args, **kwargs) |
184 migrscript, funcname, *args, **kwargs) |
184 except ExecutionError as err: |
185 except ExecutionError as err: |
185 sys.stderr.write("-> %s\n" % err) |
186 sys.stderr.write("-> %s\n" % err) |
186 except BaseException: |
187 except BaseException: |
187 self.rollback() |
188 self.rollback() |
188 raise |
189 raise |
|
190 |
189 |
191 # Adjust docstring |
190 # Adjust docstring |
192 cmd_process_script.__doc__ = MigrationHelper.cmd_process_script.__doc__ |
191 cmd_process_script.__doc__ = MigrationHelper.cmd_process_script.__doc__ |
193 |
192 |
194 # server specific migration methods ######################################## |
193 # server specific migration methods ######################################## |