From: Michael Tremer <michael.tremer@ipfire.org>
Date: Fri, 5 Dec 2025 17:27:04 +0000 (+0000)
Subject: dnsbl: sources: Use the new db methods to fetch sources
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=7354d3fbe4d092a6ce62e1660b7fb224af3a77a8;p=dnsbl.git

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 6577060..f586ddc 100644
--- 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):