lint

CERT-Polska · Feb 20, 2025 · 98b2092 · 98b2092
1 parent 5362cbd
commit 98b2092
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/app.py b/src/app.py
@@ -617,8 +617,7 @@ def add_files_to_queue(
 )
 def get_queue_status(ursadb_id: str):
     queue_status = db.get_queue_info(ursadb_id)
-    logging.error(queue_status)
-    logging.error(type(queue_status))
+
     return QueueStatusSchema(
         ursadb_id=ursadb_id,
         size=queue_status.size,

diff --git a/src/db.py b/src/db.py
@@ -510,14 +510,16 @@ def get_queue_info(self, ursadb_id: str) -> QueueStatusDatabaseSchema:
                 func.min(QueuedFile.created_at),
                 func.max(QueuedFile.created_at),
             ).where(QueuedFile.ursadb_id == ursadb_id)
-            queue_info = session.exec(query).one()
+            queue_info = session.exec(query).first()
 
-        return QueueStatusDatabaseSchema(
+        queue_status = QueueStatusDatabaseSchema(
             size=queue_info[0],
             oldest_file=queue_info[1],
             newest_file=queue_info[2],
         )
 
+        return queue_status
+
     def delete_queued_files(self, ursadb_id: str) -> None:
         with self.session() as session:
             session.query(QueuedFile).filter_by(ursadb_id=ursadb_id).delete()