X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=db_access.py;h=2fe45177bce8022a91ddbaf994d1305254c39370;hb=c30e23562021f4ead23579ed4206f7acb3201204;hp=0b038862688ae2c7f9eaa48c64da1aff28679d53;hpb=e0e87158dc6558cce66c233a27713009a3d6bec8;p=dak.git diff --git a/db_access.py b/db_access.py index 0b038862..2fe45177 100644 --- a/db_access.py +++ b/db_access.py @@ -1,6 +1,6 @@ # DB access fucntions -# Copyright (C) 2000 James Troup -# $Id: db_access.py,v 1.4 2000-12-18 07:11:25 troup Exp $ +# Copyright (C) 2000, 2001 James Troup +# $Id: db_access.py,v 1.9 2001-11-04 22:34:02 troup Exp $ # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by @@ -18,20 +18,28 @@ import pg, string -Cnf = None -projectB = None -suite_id_cache = {} -architecture_id_cache = {} -archive_id_cache = {} -component_id_cache = {} -location_id_cache = {} -maintainer_id_cache = {} -source_id_cache = {} -files_id_cache = {} +############################################################################################ + +Cnf = None; +projectB = None; +suite_id_cache = {}; +section_id_cache = {}; +priority_id_cache = {}; +override_type_id_cache = {}; +architecture_id_cache = {}; +archive_id_cache = {}; +component_id_cache = {}; +location_id_cache = {}; +maintainer_id_cache = {}; +source_id_cache = {}; +files_id_cache = {}; +maintainer_cache = {} + +############################################################################################ def init (config, sql): global Cnf, projectB - + Cnf = config; projectB = sql; @@ -45,25 +53,77 @@ def get_suite_id (suite): q = projectB.query("SELECT id FROM suite WHERE suite_name = '%s'" % (suite)) ql = q.getresult(); - if ql == []: - return -1; - - suite_id = q.getresult()[0][0] + if not ql: + return -1; + + suite_id = ql[0][0]; suite_id_cache[suite] = suite_id return suite_id +def get_section_id (section): + global section_id_cache + + if section_id_cache.has_key(section): + return section_id_cache[section] + + q = projectB.query("SELECT id FROM section WHERE section = '%s'" % (section)) + ql = q.getresult(); + if not ql: + return -1; + + section_id = ql[0][0]; + section_id_cache[section] = section_id + + return section_id + +def get_priority_id (priority): + global priority_id_cache + + if priority_id_cache.has_key(priority): + return priority_id_cache[priority] + + q = projectB.query("SELECT id FROM priority WHERE priority = '%s'" % (priority)) + ql = q.getresult(); + if not ql: + return -1; + + priority_id = ql[0][0]; + priority_id_cache[priority] = priority_id + + return priority_id + +def get_override_type_id (type): + global override_type_id_cache; + + if override_type_id_cache.has_key(type): + return override_type_id_cache[type]; + + q = projectB.query("SELECT id FROM override_type WHERE type = '%s'" % (type)); + ql = q.getresult(); + if not ql: + return -1; + + override_type_id = ql[0][0]; + override_type_id_cache[type] = override_type_id; + + return override_type_id; + def get_architecture_id (architecture): - global architecture_id_cache + global architecture_id_cache; if architecture_id_cache.has_key(architecture): - return architecture_id_cache[architecture] + return architecture_id_cache[architecture]; q = projectB.query("SELECT id FROM architecture WHERE arch_string = '%s'" % (architecture)) - architecture_id = q.getresult()[0][0] - architecture_id_cache[architecture] = architecture_id + ql = q.getresult(); + if not ql: + return -1; + + architecture_id = ql[0][0]; + architecture_id_cache[architecture] = architecture_id; - return architecture_id + return architecture_id; def get_archive_id (archive): global archive_id_cache @@ -72,7 +132,11 @@ def get_archive_id (archive): return archive_id_cache[archive] q = projectB.query("SELECT id FROM archive WHERE name = '%s'" % (archive)) - archive_id = q.getresult()[0][0] + ql = q.getresult(); + if not ql: + return -1; + + archive_id = ql[0][0] archive_id_cache[archive] = archive_id return archive_id @@ -85,10 +149,10 @@ def get_component_id (component): q = projectB.query("SELECT id FROM component WHERE lower(name) = '%s'" % (string.lower(component))) ql = q.getresult(); - if ql == []: + if not ql: return -1; - component_id = ql[0][0] + component_id = ql[0][0]; component_id_cache[component] = component_id return component_id @@ -107,7 +171,11 @@ def get_location_id (location, component, archive): q = projectB.query("SELECT id FROM location WHERE path = '%s' AND component = %d AND archive = %d" % (location, component_id, archive_id)) else: q = projectB.query("SELECT id FROM location WHERE path = '%s' AND archive = %d" % (location, archive_id)) - location_id = q.getresult()[0][0] + ql = q.getresult(); + if not ql: + return -1; + + location_id = ql[0][0] location_id_cache[cache_key] = location_id return location_id @@ -162,12 +230,12 @@ def get_files_id (filename, size, md5sum, location_id): if ql: if len(ql) != 1: return -1; - ql = ql[0] + ql = ql[0]; orig_size = int(ql[1]); orig_md5sum = ql[2]; if orig_size != size or orig_md5sum != md5sum: return -2; - files_id_cache[cache_key] = ql[0] + files_id_cache[cache_key] = ql[0]; return files_id_cache[cache_key] else: return None @@ -181,7 +249,7 @@ def set_files_id (filename, size, md5sum, location_id): cache_key = "%s~%d" % (filename, location_id); #print "INSERT INTO files (filename, size, md5sum, location) VALUES ('%s', %d, '%s', %d)" % (filename, long(size), md5sum, location_id); - projectB.query("INSERT INTO files (filename, size, md5sum, location) VALUES ('%s', %d, '%s', %d)" % (filename, long(size), md5sum, location_id)); + projectB.query("INSERT INTO files (filename, size, md5sum, location) VALUES ('%s', %d, '%s', %d)" % (filename, long(size), md5sum, location_id)); q = projectB.query("SELECT id FROM files WHERE id = currval('files_id_seq')"); ql = q.getresult()[0]; files_id_cache[cache_key] = ql[0] @@ -190,3 +258,13 @@ def set_files_id (filename, size, md5sum, location_id): ########################################################################################## +def get_maintainer (maintainer_id): + global maintainer_cache; + + if not maintainer_cache.has_key(maintainer_id): + q = projectB.query("SELECT name FROM maintainer WHERE id = %s" % (maintainer_id)); + maintainer_cache[maintainer_id] = q.getresult()[0][0]; + + return maintainer_cache[maintainer_id]; + +##########################################################################################