X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=daklib%2Fdbconn.py;h=aeffdcb328562582f4bea9c22b2788f436256e61;hb=d1fee7b2e83c715efbaa7cc006a43d79bf0614c5;hp=98ebcaa1ca43aaaa065cf8f5ce350776caf8a1eb;hpb=0b8842a27c568841dce63ad16084d33567751505;p=dak.git

diff --git a/daklib/dbconn.py b/daklib/dbconn.py
index 98ebcaa1..aeffdcb3 100755
--- a/daklib/dbconn.py
+++ b/daklib/dbconn.py
@@ -1992,6 +1992,7 @@ class SuiteSrcFormat(object):
 
 __all__.append('SuiteSrcFormat')
 
+@session_wrapper
 def get_suite_src_formats(suite, session=None):
     """
     Returns list of allowed SrcFormat for C{suite}.
@@ -2007,22 +2008,12 @@ def get_suite_src_formats(suite, session=None):
     @return: the list of allowed source formats for I{suite}
     """
 
-    privatetrans = False
-    if session is None:
-        session = DBConn().session()
-        privatetrans = True
-
     q = session.query(SrcFormat)
     q = q.join(SuiteSrcFormat)
     q = q.join(Suite).filter_by(suite_name=suite)
     q = q.order_by('format_name')
 
-    ret = q.all()
-
-    if privatetrans:
-        session.close()
-
-    return ret
+    return q.all()
 
 __all__.append('get_suite_src_formats')