Minor updates

mt-krainski · Nov 28, 2024 · d3a4455 · d3a4455
1 parent d3a76c7
commit d3a4455
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/runs/run.sh b/runs/run.sh
@@ -1,7 +1,7 @@
 #!/usr/bin/env bash
 
 IFS=$'\n'
-for row in $(cat nov-europe.csv | sed 's/\r$//')
+for row in $(cat flights-to-scan.csv | sed 's/\r$//')
 do
   origin_airport_name=$(echo $row | cut -d "," -f 1)
   origin_airport_code=$(echo $row | cut -d "," -f 2)

diff --git a/yafs/pyproject.toml b/yafs/pyproject.toml
@@ -82,3 +82,9 @@ run-hooks = "pre-commit run --all-files"
 sequence = [
   { ref = "test" }
 ]
+
+[tool.poe.tasks.update]
+sequence = [
+  { shell = "poetry install" },
+  { shell = "playwright install" }
+]
diff --git a/yafs/yafs/flights.py b/yafs/yafs/flights.py
@@ -49,7 +49,7 @@ async def _input_location(page, label, location_name, location_code):
     location = page.get_by_label(label, exact=True)
     await location.fill(location_name)
     await page.wait_for_timeout(2000)
-    airport = page.get_by_text(location_code, exact=True)
+    airport = page.get_by_text(location_code, exact=True).first
     await airport.click()