Coverage for dak/dakdb/update131.py: 77%

1""" 

2Add stayofexecution to the suite table 

3 

4@contact: Debian FTP Master <ftpmaster@debian.org> 

5@copyright: 2025 Jochen Sprickerhof <jspricke@debian.org> 

6@copyright: 2025 James Addison <jay@jp-hosting.net> 

7@license: GNU General Public License version 2 or later 

8""" 

9 

10# This program is free software; you can redistribute it and/or modify 

11# it under the terms of the GNU General Public License as published by 

12# the Free Software Foundation; either version 2 of the License, or 

13# (at your option) any later version. 

14 

15# This program is distributed in the hope that it will be useful, 

16# but WITHOUT ANY WARRANTY; without even the implied warranty of 

17# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

18# GNU General Public License for more details. 

19 

20# You should have received a copy of the GNU General Public License 

21# along with this program; if not, write to the Free Software 

22# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 

23 

24################################################################################ 

25 

26import psycopg2 

27 

28from daklib.dak_exceptions import DBUpdateError 

29 

30################################################################################ 

31 

32 

33def do_update(self): 

34 print(__doc__) 

35 try: 

36 c = self.db.cursor() 

37 

38 c.execute( 

39 "ALTER TABLE suite ADD COLUMN stayofexecution INTERVAL NOT NULL DEFAULT INTERVAL '0 days'", 

40 ) 

41 c.execute( 

42 "UPDATE suite SET stayofexecution = INTERVAL '1 day' WHERE id IN (SELECT suite FROM suite_build_queue_copy)" 

43 ) 

44 

45 c.execute("UPDATE config SET value = '131' WHERE name = 'db_revision'") 

46 self.db.commit() 

47 

48 except psycopg2.ProgrammingError as msg: 

49 self.db.rollback() 

50 raise DBUpdateError( 

51 "Unable to apply sick update 131, rollback issued. Error message: {0}".format( 

52 msg 

53 ) 

54 )