[dak/master] allow querying on dm_upload_allowed
Signed-off-by: Mark Hymers <mhy@debian.org>
---
daklib/dbconn.py | 13 +++++++++++--
1 files changed, 11 insertions(+), 2 deletions(-)
diff --git a/daklib/dbconn.py b/daklib/dbconn.py
index 537f822..b3811a1 100755
--- a/daklib/dbconn.py
+++ b/daklib/dbconn.py
@@ -777,13 +777,17 @@ class DBSource(object):
__all__.append('DBSource')
-def get_sources_from_name(source, session=None):
+def get_sources_from_name(source, dm_upload_allowed=None, session=None):
"""
Returns list of DBSource objects for given C{source} name
@type source: str
@param source: DBSource package name to search for
+ @type dm_upload_allowed: bool
+ @param dm_upload_allowed: If None, no effect. If True or False, only
+ return packages with that dm_upload_allowed setting
+
@type session: Session
@param session: Optional SQL session object (a temporary one will be
generated if not supplied)
@@ -793,7 +797,12 @@ def get_sources_from_name(source, session=None):
"""
if session is None:
session = DBConn().session()
- return session.query(DBSource).filter_by(source=source).all()
+
+ q = session.query(DBSource).filter_by(source=source)
+ if dm_upload_allowed is not None:
+ q = q.filter_by(dm_upload_allowed=dm_upload_allowed)
+
+ return q.all()
__all__.append('get_sources_from_name')
--
1.5.6.5
Reply to: