__all__.append('get_priority')
+def get_priorities(session=None):
+ """
+ Returns dictionary of priority names -> id mappings
+
+ @type session: Session
+ @param session: Optional SQL session object (a temporary one will be
+ generated if not supplied)
+
+ @rtype: dictionary
+ @return: dictionary of priority names -> id mappings
+ """
+ if session is None:
+ session = DBConn().session()
+
+ ret = {}
+ q = session.query(Priority)
+ for x in q.all():
+ ret[x.priority] = x.priority_id
+
+ return ret
+
+__all__.append('get_priorities')
+
################################################################################
class Queue(object):
__all__.append('QueueBuild')
-def get_queue_build(filename, suite_id, session=None):
+def get_queue_build(filename, suite, session=None):
"""
- Returns QueueBuild object for given C{filename} and C{suite id}.
+ Returns QueueBuild object for given C{filename} and C{suite}.
@type filename: string
@param filename: The name of the file
- @type suiteid: int
- @param suiteid: Suite ID
+ @type suiteid: int or str
+ @param suiteid: Suite name or ID
@type session: Session
@param session: Optional SQLA session object (a temporary one will be
"""
if session is None:
session = DBConn().session()
- q = session.query(QueueBuild).filter_by(filename=filename).filter_by(suite_id=suite_id)
+ if isinstance(suite, int):
+ q = session.query(QueueBuild).filter_by(filename=filename).filter_by(suite_id=suite)
+ else:
+ q = session.query(QueueBuild).filter_by(filename=filename)
+ q = q.join(Suite).filter_by(suite_name=suite)
+
if q.count() == 0:
return None
return q.one()
__all__.append('get_section')
+def get_sections(session=None):
+ """
+ Returns dictionary of section names -> id mappings
+
+ @type session: Session
+ @param session: Optional SQL session object (a temporary one will be
+ generated if not supplied)
+
+ @rtype: dictionary
+ @return: dictionary of section names -> id mappings
+ """
+ if session is None:
+ session = DBConn().session()
+
+ ret = {}
+ q = session.query(Section)
+ for x in q.all():
+ ret[x.section] = x.section_id
+
+ return ret
+
+__all__.append('get_sections')
+
################################################################################
class DBSource(object):