Fix get sql timestamp

Author: davidgamez

functions-python/helpers/locations.py

@@ -35,6 +35,7 @@ def get_country_code(country_name: str) -> Optional[str]:
         Optional[str]: Two-letter ISO country code or None if not found
     """
     import pycountry
+
     # Return None for empty or whitespace-only strings
     if not country_name or not country_name.strip():
         logging.error("Could not find country code for: empty string")

functions-python/tasks_executor/src/tasks/geojson/update_geojson_files_precision.py

@@ -20,6 +20,8 @@
 import re
 from typing import Any, Dict, List
 
+from sqlalchemy import select, func
+
 from shared.database.database import with_db_session
 from shared.database_gen.sqlacodegen_models import Gtfsfeed
 from shared.helpers.locations import round_geojson_coords
@@ -151,7 +153,7 @@ def update_geojson_files_precision_handler(
 
     feeds: [Gtfsfeed] = query_unprocessed_feeds(limit, db_session)
     logging.info("Found %s feeds", len(feeds))
-    timestamp = db_session.execute("SELECT CURRENT_TIMESTAMP").scalar()
+    timestamp = db_session.execute(select(func.current_timestamp())).scalar()
     for feed in feeds:
         try:
             if processed % 100 == 0:

functions-python/tasks_executor/tests/conftest.py

@@ -14,15 +14,6 @@
 #  limitations under the License.
 #
 
-from pathlib import Path
-import sys
-
-# Ensure project's src is on sys.path so imports like 'shared' resolve when tests start
-_repo_root = Path(__file__).resolve().parents[1]
-_src_dir = str(_repo_root / "src")
-if _src_dir not in sys.path:
-    sys.path.insert(0, _src_dir)
-
 from datetime import datetime, UTC, timedelta
 
 from sqlalchemy.orm import Session

functions-python/tasks_executor/tests/tasks/geojson/test_update_geojson_files_precision.py

@@ -81,7 +81,10 @@ def test_process_geojson_round_and_remove_osm_keys(self):
             "features": [
                 {
                     "type": "Feature",
-                    "geometry": {"type": "Point", "coordinates": [12.3456789, -98.7654321]},
+                    "geometry": {
+                        "type": "Point",
+                        "coordinates": [12.3456789, -98.7654321],
+                    },
                     "properties": {
                         "name": "A",
                         "osm_id": 123,
@@ -113,13 +116,18 @@ def test_process_geojson_single_feature_and_list_variants(self):
         }
         out1 = process_geojson(feat, precision=4)
         self.assertIsInstance(out1, dict)
-        self.assertEqual(out1["geometry"]["coordinates"], [round(1.23456789, 4), round(2.3456789, 4)])
+        self.assertEqual(
+            out1["geometry"]["coordinates"], [round(1.23456789, 4), round(2.3456789, 4)]
+        )
         self.assertNotIn("osm", out1.get("properties", {}))
 
         lst = [feat]
         out2 = process_geojson(lst, precision=3)
         self.assertIsInstance(out2, list)
-        self.assertEqual(out2[0]["geometry"]["coordinates"], [round(1.23456789, 3), round(2.3456789, 3)])
+        self.assertEqual(
+            out2[0]["geometry"]["coordinates"],
+            [round(1.23456789, 3), round(2.3456789, 3)],
+        )
 
     @patch("tasks.geojson.update_geojson_files_precision.query_unprocessed_feeds")
     def test_handler_uploads_and_updates_feed_info(self, mock_query):
@@ -128,7 +136,10 @@ def test_handler_uploads_and_updates_feed_info(self, mock_query):
             "features": [
                 {
                     "type": "Feature",
-                    "geometry": {"type": "Point", "coordinates": [100.1234567, 0.9876543]},
+                    "geometry": {
+                        "type": "Point",
+                        "coordinates": [100.1234567, 0.9876543],
+                    },
                     "properties": {"id": "node/1", "keep": "x"},
                 }
             ],
@@ -149,7 +160,11 @@ def test_handler_uploads_and_updates_feed_info(self, mock_query):
         fake_feed = types.SimpleNamespace(
             stable_id=feed_stable_id,
             geolocation_file_created_date=None,
-            gtfsdatasets=[types.SimpleNamespace(bounding_box=types.SimpleNamespace(id="bbid"), downloaded_date=None)],
+            gtfsdatasets=[
+                types.SimpleNamespace(
+                    bounding_box=types.SimpleNamespace(id="bbid"), downloaded_date=None
+                )
+            ],
         )
 
         mock_query.return_value = [fake_feed]
@@ -168,7 +183,12 @@ def commit(self):
 
         fake_db = FakeDBSession()
 
-        payload = {"bucket_name": "any-bucket", "dry_run": False, "precision": 5, "limit": 1}
+        payload = {
+            "bucket_name": "any-bucket",
+            "dry_run": False,
+            "precision": 5,
+            "limit": 1,
+        }
 
         # Inject modules into sys.modules for the duration of the handler call
         with patch.dict(sys.modules, {"google.cloud": cloud_mod, "google": google_mod}):