Make use of db.cursor() in with blocks

2022-10-26 15:49:14 +02:00
parent 217bbcadb5
commit 4dc7e972a2
3 changed files with 103 additions and 95 deletions
--- a/lib/exporter.py
+++ b/lib/exporter.py
@@ -12,13 +12,13 @@ class Exporter:

    def run(self):
        db = DB()
-        cur = db.cursor()
-        cur.execute(
-            "SELECT * from revisions where package=%s ORDER BY project,rev",
-            (self.package,),
-        )
-        data = {"revisions": []}
-        for row in cur.fetchall():
-            data["revisions"].append(DBRevision(row).as_dict(db))
+        with db.cursor() as cur:
+            cur.execute(
+                "SELECT * from revisions where package=%s ORDER BY project,rev",
+                (self.package,),
+            )
+            data = {"revisions": []}
+            for row in cur.fetchall():
+                data["revisions"].append(DBRevision(row).as_dict(db))

        yaml.dump(data, sys.stdout, default_flow_style=False)