dnsbl: sources: Use the new db methods to fetch sources

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/dnsbl/sources.py b/src/dnsbl/sources.py
index 6577060e1d75c3540a666789d274172bc92b13df..f586ddc911678566dc3875c07d0df72bff39f902 100644 (file)
--- a/src/dnsbl/sources.py
+++ b/src/dnsbl/sources.py
@@ -42,28 +42,20 @@ class Sources(object):
                        )
                )
 
-               with self.backend.session() as session:
-                       result = session.execute(stmt)
-
-                       return result.scalar_one_or_none()
+               return self.backend.db.fetch_one(stmt)
 
        def create(self, list, name, url, created_by, license):
                """
                        Creates a new source
                """
-               # Create a new source
-               with self.backend.session() as session:
-                       source = Source(
-                               list       = list,
-                               name       = name,
-                               url        = url,
-                               created_by = created_by,
-                               license    = license,
-                       )
-                       session.add(source)
-                       session.commit()
-
-               return source
+               return self.backend.db.insert(
+                       Source,
+                       list       = list,
+                       name       = name,
+                       url        = url,
+                       created_by = created_by,
+                       license    = license,
+               )
 
 
 class Source(sqlmodel.SQLModel, database.BackendMixin, table=True):