]> git.decadent.org.uk Git - dak.git/blobdiff - daklib/dbconn.py
export some exceptions
[dak.git] / daklib / dbconn.py
index f6b35d949d3b41b61b3adc4235b74b8d87ab0a30..42ef7e9f78eb5da11fd116ed88231fba25791fd5 100755 (executable)
@@ -47,7 +47,7 @@ from singleton import Singleton
 
 ################################################################################
 
-__all__ = []
+__all__ = ['IntegrityError', 'SQLAlchemyError']
 
 ################################################################################
 
@@ -109,6 +109,8 @@ def get_architecture_suites(architecture, session=None):
 
 __all__.append('get_architecture_suites')
 
+################################################################################
+
 class Archive(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -143,6 +145,8 @@ def get_archive(archive, session=None):
 
 __all__.append('get_archive')
 
+################################################################################
+
 class BinAssociation(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -152,18 +156,20 @@ class BinAssociation(object):
 
 __all__.append('BinAssociation')
 
-class Binary(object):
+################################################################################
+
+class DBBinary(object):
     def __init__(self, *args, **kwargs):
         pass
 
     def __repr__(self):
-        return '<Binary %s (%s, %s)>' % (self.package, self.version, self.architecture)
+        return '<DBBinary %s (%s, %s)>' % (self.package, self.version, self.architecture)
 
-__all__.append('Binary')
+__all__.append('DBBinary')
 
 def get_binary_from_id(id, session=None):
     """
-    Returns Binary object for given C{id}
+    Returns DBBinary object for given C{id}
 
     @type id: int
     @param id: Id of the required binary
@@ -172,12 +178,12 @@ def get_binary_from_id(id, session=None):
     @param session: Optional SQLA session object (a temporary one will be
     generated if not supplied)
 
-    @rtype: Binary
-    @return: Binary object for the given binary (None if not present)
+    @rtype: DBBinary
+    @return: DBBinary object for the given binary (None if not present)
     """
     if session is None:
         session = DBConn().session()
-    q = session.query(Binary).filter_by(binary_id=id)
+    q = session.query(DBBinary).filter_by(binary_id=id)
     if q.count() == 0:
         return None
     return q.one()
@@ -186,24 +192,26 @@ __all__.append('get_binary_from_id')
 
 def get_binaries_from_name(package, session=None):
     """
-    Returns list of Binary objects for given C{package} name
+    Returns list of DBBinary objects for given C{package} name
 
     @type package: str
-    @param package: Binary package name to search for
+    @param package: DBBinary package name to search for
 
     @type session: Session
     @param session: Optional SQL session object (a temporary one will be
     generated if not supplied)
 
     @rtype: list
-    @return: list of Binary objects for the given name (may be empty)
+    @return: list of DBBinary objects for the given name (may be empty)
     """
     if session is None:
         session = DBConn().session()
-    return session.query(Binary).filter_by(package=package).all()
+    return session.query(DBBinary).filter_by(package=package).all()
 
 __all__.append('get_binaries_from_name')
 
+################################################################################
+
 class Component(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -235,6 +243,8 @@ def get_component(component, session=None):
 
 __all__.append('get_component')
 
+################################################################################
+
 class DBConfig(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -244,6 +254,8 @@ class DBConfig(object):
 
 __all__.append('DBConfig')
 
+################################################################################
+
 class ContentFilename(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -287,6 +299,8 @@ def get_or_set_contents_file_id(filename, session=None):
 
 __all__.append('get_or_set_contents_file_id')
 
+################################################################################
+
 class ContentFilepath(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -330,6 +344,8 @@ def get_or_set_contents_path_id(filepath, session):
 
 __all__.append('get_or_set_contents_path_id')
 
+################################################################################
+
 class ContentAssociation(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -389,6 +405,8 @@ def insert_content_paths(binary_id, fullpaths, session=None):
 
 __all__.append('insert_content_paths')
 
+################################################################################
+
 class DSCFile(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -398,6 +416,8 @@ class DSCFile(object):
 
 __all__.append('DSCFile')
 
+################################################################################
+
 class PoolFile(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -434,6 +454,8 @@ def get_poolfile_by_name(filename, location_id=None, session=None):
 
 __all__.append('get_poolfile_by_name')
 
+################################################################################
+
 class Fingerprint(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -443,6 +465,8 @@ class Fingerprint(object):
 
 __all__.append('Fingerprint')
 
+################################################################################
+
 class Keyring(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -452,6 +476,8 @@ class Keyring(object):
 
 __all__.append('Keyring')
 
+################################################################################
+
 class Location(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -497,6 +523,8 @@ def get_location(location, component=None, archive=None, session=None):
 
 __all__.append('get_location')
 
+################################################################################
+
 class Maintainer(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -506,6 +534,8 @@ class Maintainer(object):
 
 __all__.append('Maintainer')
 
+################################################################################
+
 class Override(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -515,6 +545,8 @@ class Override(object):
 
 __all__.append('Override')
 
+################################################################################
+
 class OverrideType(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -548,6 +580,8 @@ def get_override_type(override_type, session=None):
 
 __all__.append('get_override_type')
 
+################################################################################
+
 class PendingContentAssociation(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -623,6 +657,8 @@ def insert_pending_content_paths(package, fullpaths, session=None):
 
 __all__.append('insert_pending_content_paths')
 
+################################################################################
+
 class Priority(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -656,6 +692,8 @@ def get_priority(priority, session=None):
 
 __all__.append('get_priority')
 
+################################################################################
+
 class Queue(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -665,6 +703,8 @@ class Queue(object):
 
 __all__.append('Queue')
 
+################################################################################
+
 class QueueBuild(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -674,6 +714,8 @@ class QueueBuild(object):
 
 __all__.append('QueueBuild')
 
+################################################################################
+
 class Section(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -707,38 +749,40 @@ def get_section(section, session=None):
 
 __all__.append('get_section')
 
-class Source(object):
+################################################################################
+
+class DBSource(object):
     def __init__(self, *args, **kwargs):
         pass
 
     def __repr__(self):
-        return '<Source %s (%s)>' % (self.source, self.version)
+        return '<DBSource %s (%s)>' % (self.source, self.version)
 
-__all__.append('Source')
+__all__.append('DBSource')
 
 def get_sources_from_name(source, session=None):
     """
-    Returns list of Source objects for given C{source} name
+    Returns list of DBSource objects for given C{source} name
 
     @type source: str
-    @param source: Source package name to search for
+    @param source: DBSource package name to search for
 
     @type session: Session
     @param session: Optional SQL session object (a temporary one will be
     generated if not supplied)
 
     @rtype: list
-    @return: list of Source objects for the given name (may be empty)
+    @return: list of DBSource objects for the given name (may be empty)
     """
     if session is None:
         session = DBConn().session()
-    return session.query(Source).filter_by(source=source).all()
+    return session.query(DBSource).filter_by(source=source).all()
 
 __all__.append('get_sources_from_name')
 
 def get_source_in_suite(source, suite, session=None):
     """
-    Returns list of Source objects for a combination of C{source} and C{suite}.
+    Returns list of DBSource objects for a combination of C{source} and C{suite}.
 
       - B{source} - source package name, eg. I{mailfilter}, I{bbdb}, I{glibc}
       - B{suite} - a suite name, eg. I{unstable}
@@ -765,6 +809,8 @@ def get_source_in_suite(source, suite, session=None):
 
 __all__.append('get_source_in_suite')
 
+################################################################################
+
 class SrcAssociation(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -774,6 +820,8 @@ class SrcAssociation(object):
 
 __all__.append('SrcAssociation')
 
+################################################################################
+
 class SrcUploader(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -783,6 +831,8 @@ class SrcUploader(object):
 
 __all__.append('SrcUploader')
 
+################################################################################
+
 class Suite(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -847,6 +897,8 @@ def get_suite(suite, session=None):
 
 __all__.append('get_suite')
 
+################################################################################
+
 class SuiteArchitecture(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -881,6 +933,8 @@ def get_suite_architectures(suite, session=None):
 
 __all__.append('get_suite_architectures')
 
+################################################################################
+
 class Uid(object):
     def __init__(self, *args, **kwargs):
         pass
@@ -948,16 +1002,16 @@ class DBConn(Singleton):
                                  suite_id = self.tbl_bin_associations.c.suite,
                                  suite = relation(Suite),
                                  binary_id = self.tbl_bin_associations.c.bin,
-                                 binary = relation(Binary)))
+                                 binary = relation(DBBinary)))
 
-        mapper(Binary, self.tbl_binaries,
+        mapper(DBBinary, self.tbl_binaries,
                properties = dict(binary_id = self.tbl_binaries.c.id,
                                  package = self.tbl_binaries.c.package,
                                  version = self.tbl_binaries.c.version,
                                  maintainer_id = self.tbl_binaries.c.maintainer,
                                  maintainer = relation(Maintainer),
                                  source_id = self.tbl_binaries.c.source,
-                                 source = relation(Source),
+                                 source = relation(DBSource),
                                  arch_id = self.tbl_binaries.c.architecture,
                                  architecture = relation(Architecture),
                                  poolfile_id = self.tbl_binaries.c.file,
@@ -983,7 +1037,7 @@ class DBConn(Singleton):
                                  filepath_id = self.tbl_content_associations.c.filepath,
                                  filepath    = relation(ContentFilepath),
                                  binary_id   = self.tbl_content_associations.c.binary_pkg,
-                                 binary      = relation(Binary)))
+                                 binary      = relation(DBBinary)))
 
 
         mapper(ContentFilename, self.tbl_content_file_names,
@@ -997,7 +1051,7 @@ class DBConn(Singleton):
         mapper(DSCFile, self.tbl_dsc_files,
                properties = dict(dscfile_id = self.tbl_dsc_files.c.id,
                                  source_id = self.tbl_dsc_files.c.source,
-                                 source = relation(Source),
+                                 source = relation(DBSource),
                                  poolfile_id = self.tbl_dsc_files.c.file,
                                  poolfile = relation(PoolFile)))
 
@@ -1066,7 +1120,7 @@ class DBConn(Singleton):
         mapper(Section, self.tbl_section,
                properties = dict(section_id = self.tbl_section.c.id))
 
-        mapper(Source, self.tbl_source,
+        mapper(DBSource, self.tbl_source,
                properties = dict(source_id = self.tbl_source.c.id,
                                  version = self.tbl_source.c.version,
                                  maintainer_id = self.tbl_source.c.maintainer,
@@ -1089,12 +1143,12 @@ class DBConn(Singleton):
                                  suite_id = self.tbl_src_associations.c.suite,
                                  suite = relation(Suite),
                                  source_id = self.tbl_src_associations.c.source,
-                                 source = relation(Source)))
+                                 source = relation(DBSource)))
 
         mapper(SrcUploader, self.tbl_src_uploaders,
                properties = dict(uploader_id = self.tbl_src_uploaders.c.id,
                                  source_id = self.tbl_src_uploaders.c.source,
-                                 source = relation(Source,
+                                 source = relation(DBSource,
                                                    primaryjoin=(self.tbl_src_uploaders.c.source==self.tbl_source.c.id)),
                                  maintainer_id = self.tbl_src_uploaders.c.maintainer,
                                  maintainer = relation(Maintainer,
@@ -1105,7 +1159,7 @@ class DBConn(Singleton):
 
         mapper(SuiteArchitecture, self.tbl_suite_architectures,
                properties = dict(suite_id = self.tbl_suite_architectures.c.suite,
-                                 suite = relation(Suite),
+                                 suite = relation(Suite, backref='suitearchitectures'),
                                  arch_id = self.tbl_suite_architectures.c.architecture,
                                  architecture = relation(Architecture)))