Bump sqlalchemy version

database/base.py

@@ -2,8 +2,7 @@
 
 from sqlalchemy import Column, types
 from sqlalchemy.dialects.postgresql import UUID
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import Session
+from sqlalchemy.orm import Session, declarative_base
 
 from helpers.clock import get_utc_now
 

database/engine.py

@@ -54,7 +54,7 @@ def create_session(self):
             timeseries_engine = self.timeseries_engine
 
             class RoutingSession(Session):
-                def get_bind(self, mapper=None, clause=None):
+                def get_bind(self, mapper=None, clause=None, **kwargs):
                     if mapper is not None and issubclass(
                         mapper.class_, TimeseriesBaseModel
                     ):
@@ -78,12 +78,12 @@ def get_bind(self, mapper=None, clause=None):
     database_url=get_config(
         "services",
         "database_url",
-        default="postgres://postgres:@postgres:5432/postgres",
+        default="postgresql://postgres:@postgres:5432/postgres",
     ),
     timeseries_database_url=get_config(
         "services",
         "timeseries_database_url",
-        default="postgres://postgres:@timescale:5432/postgres",
+        default="postgresql://postgres:@timescale:5432/postgres",
     ),
 )
 

docker/test_codecov_config.yml

@@ -7,8 +7,8 @@ setup:
     enabled: true
 
 services:
-  database_url: postgres://postgres:password@postgres:5432/postgres
-  timeseries_database_url: postgres://postgres:password@timescale:5432/postgres
+  database_url: postgresql://postgres:password@postgres:5432/postgres
+  timeseries_database_url: postgresql://postgres:password@timescale:5432/postgres
   redis_url: redis://redis:6379
   minio:
     hash_key: testixik8qdauiab1yiffydimvi72ekq # never change this

pyproject.toml

@@ -39,8 +39,8 @@ dependencies = [
     "requests>=2.32.0",
     "sentry-sdk>=2.13.0",
     "shared",
-    "sqlalchemy==1.3.*",
-    "sqlparse==0.5.0",
+    "sqlalchemy<2",
+    "sqlparse>=0.5.0",
     "statsd>=3.3.0",
     "stripe>=11.4.1",
     "test-results-parser",

ta_storage/pg.py

@@ -37,14 +37,15 @@ def get_repo_flag_ids(db_session: Session, repoid: int, flags: list[str]) -> set
     if not flags:
         return set()
 
-    return set(
-        db_session.query(RepositoryFlag.id_)
+    return {
+        flag.id_
+        for flag in db_session.query(RepositoryFlag.id_)
         .filter(
             RepositoryFlag.repository_id == repoid,
             RepositoryFlag.flag_name.in_(flags),
         )
         .all()
-    )
+    }
 
 
 def modify_structures(

tasks/test_results_processor.py

@@ -44,14 +44,15 @@ def get_repo_flag_ids(db_session: Session, repoid: int, flags: list[str]) -> set
     if not flags:
         return set()
 
-    return set(
-        db_session.query(RepositoryFlag.id_)
+    return {
+        flag.id_
+        for flag in db_session.query(RepositoryFlag.id_)
         .filter(
             RepositoryFlag.repository_id == repoid,
             RepositoryFlag.flag_name.in_(flags),
         )
         .all()
-    )
+    }
 
 
 def create_daily_totals(