]> git.donarmstrong.com Git - dak.git/blob - dak/update_db.py
fix update script
[dak.git] / dak / update_db.py
1 #!/usr/bin/env python
2
3 """ Database Update Main Script
4
5 @contact: Debian FTP Master <ftpmaster@debian.org>
6 # Copyright (C) 2008  Michael Casadevall <mcasadevall@debian.org>
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
26 # <Ganneff> when do you have it written?
27 # <NCommander> Ganneff, after you make my debian account
28 # <Ganneff> blackmail wont work
29 # <NCommander> damn it
30
31 ################################################################################
32
33 import psycopg2
34 import sys
35 import fcntl
36 import os
37 import apt_pkg
38 import time
39 import errno
40 from daklib import database
41 from daklib import utils
42 from daklib.dak_exceptions import DBUpdateError
43
44 ################################################################################
45
46 Cnf = None
47 projectB = None
48 required_database_schema = 10
49
50 ################################################################################
51
52 class UpdateDB:
53     def usage (self, exit_code=0):
54         print """Usage: dak update-db
55 Updates dak's database schema to the lastest version. You should disable crontabs while this is running
56
57   -h, --help                show this help and exit."""
58         sys.exit(exit_code)
59
60
61 ################################################################################
62
63     def update_db_to_zero(self):
64         """ This function will attempt to update a pre-zero database schema to zero """
65
66         # First, do the sure thing, and create the configuration table
67         try:
68             print "Creating configuration table ..."
69             c = self.db.cursor()
70             c.execute("""CREATE TABLE config (
71                                   id SERIAL PRIMARY KEY NOT NULL,
72                                   name TEXT UNIQUE NOT NULL,
73                                   value TEXT
74                                 );""")
75             c.execute("INSERT INTO config VALUES ( nextval('config_id_seq'), 'db_revision', '0')")
76             self.db.commit()
77
78         except psycopg2.ProgrammingError:
79             self.db.rollback()
80             print "Failed to create configuration table."
81             print "Can the projectB user CREATE TABLE?"
82             print ""
83             print "Aborting update."
84             sys.exit(-255)
85
86 ################################################################################
87
88     def get_db_rev(self):
89         global projectB
90
91         # We keep database revision info the config table
92         # Try and access it
93
94         try:
95             c = self.db.cursor()
96             q = c.execute("SELECT value FROM config WHERE name = 'db_revision';")
97             return c.fetchone()[0]
98
99         except psycopg2.ProgrammingError:
100             # Whoops .. no config table ...
101             self.db.rollback()
102             print "No configuration table found, assuming dak database revision to be pre-zero"
103             return -1
104
105 ################################################################################
106
107     def update_db(self):
108         # Ok, try and find the configuration table
109         print "Determining dak database revision ..."
110
111         try:
112             # Build a connect string
113             connect_str = "dbname=%s"% (Cnf["DB::Name"])
114             if Cnf["DB::Host"] != '': connect_str += " host=%s" % (Cnf["DB::Host"])
115             if Cnf["DB::Port"] != '-1': connect_str += " port=%d" % (int(Cnf["DB::Port"]))
116
117             self.db = psycopg2.connect(connect_str)
118
119         except:
120             print "FATAL: Failed connect to database"
121             pass
122
123         database_revision = int(self.get_db_rev())
124
125         if database_revision == -1:
126             print "dak database schema predates update-db."
127             print ""
128             print "This script will attempt to upgrade it to the lastest, but may fail."
129             print "Please make sure you have a database backup handy. If you don't, press Ctrl-C now!"
130             print ""
131             print "Continuing in five seconds ..."
132             time.sleep(5)
133             print ""
134             print "Attempting to upgrade pre-zero database to zero"
135
136             self.update_db_to_zero()
137             database_revision = 0
138
139         print "dak database schema at " + str(database_revision)
140         print "dak version requires schema " + str(required_database_schema)
141
142         if database_revision == required_database_schema:
143             print "no updates required"
144             sys.exit(0)
145
146         for i in range (database_revision, required_database_schema):
147             print "updating database schema from " + str(database_revision) + " to " + str(i+1)
148             try:
149                 dakdb = __import__("dakdb", globals(), locals(), ['update'+str(i+1)])
150                 update_module = getattr(dakdb, "update"+str(i+1))
151                 update_module.do_update(self)
152             except DBUpdateError, e:
153                 # Seems the update did not work.
154                 print "Was unable to update database schema from %s to %s." % (str(database_revision), str(i+1))
155                 print "The error message received was %s" % (e)
156                 utils.fubar("DB Schema upgrade failed")
157             database_revision += 1
158
159 ################################################################################
160
161     def init (self):
162         global Cnf, projectB
163
164         Cnf = utils.get_conf()
165         arguments = [('h', "help", "Update-DB::Options::Help")]
166         for i in [ "help" ]:
167             if not Cnf.has_key("Update-DB::Options::%s" % (i)):
168                 Cnf["Update-DB::Options::%s" % (i)] = ""
169
170         arguments = apt_pkg.ParseCommandLine(Cnf, arguments, sys.argv)
171
172         options = Cnf.SubTree("Update-DB::Options")
173         if options["Help"]:
174             self.usage()
175         elif arguments:
176             utils.warn("dak update-db takes no arguments.")
177             self.usage(exit_code=1)
178
179
180         self.update_db()
181
182         try:
183             lock_fd = os.open(Cnf["Dinstall::LockFile"], os.O_RDWR | os.O_CREAT)
184             fcntl.lockf(lock_fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
185         except IOError, e:
186             if errno.errorcode[e.errno] == 'EACCES' or errno.errorcode[e.errno] == 'EAGAIN':
187                 utils.fubar("Couldn't obtain lock; assuming another 'dak process-unchecked' is already running.")
188
189
190 ################################################################################
191
192 if __name__ == '__main__':
193     app = UpdateDB()
194     app.init()
195
196 def main():
197     app = UpdateDB()
198     app.init()