Fix lint

scidsg · Sep 20, 2024 · 002d0f9 · 002d0f9
1 parent af9c4a7
commit 002d0f9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/scripts/dev_data.py b/scripts/dev_data.py
@@ -58,9 +58,10 @@ def main() -> None:
         },
     ]
     for data in tiers:
-        tier = Tier.query.filter_by(name=data["name"]).first()
-        if not tier:
-            tier = Tier(name=data["name"], monthly_amount=data["monthly_amount"])
+        name = str(data["name"])  # type: ignore
+        monthly_amount = int(data["monthly_amount"])  # type: ignore
+        if not db.session.query(exists(Tier).where(Tier.name == data["name"])).scalar():
+            tier = Tier(name, monthly_amount)
             db.session.add(tier)
             db.session.commit()