Update ml id column sites

pvsite_datamodel/sqlmodels.py

@@ -122,6 +122,9 @@ class SiteSQL(Base, CreatedMixin):
     """
 
     __tablename__ = "sites"
+    __table_args__ = (
+        UniqueConstraint("client_site_name", "ml_id", name="uniq_cons_client_name_ml_id"),
+    )
 
     site_uuid = sa.Column(UUID(as_uuid=True), default=uuid.uuid4, primary_key=True)
     client_site_id = sa.Column(
@@ -165,7 +168,6 @@ class SiteSQL(Base, CreatedMixin):
         sa.Integer,
         autoincrement=True,
         nullable=False,
-        unique=True,
         comment="Auto-incrementing integer ID of the site for use in ML training",
     )
 

pvsite_datamodel/write/user_and_site.py

@@ -74,6 +74,7 @@ def create_site(
     tilt: Optional[float] = None,
     inverter_capacity_kw: Optional[float] = None,
     module_capacity_kw: Optional[float] = None,
+    ml_id: Optional[int] = None,
 ) -> [SiteSQL, str]:
     """
     Create a site and adds it to the database.
@@ -93,6 +94,7 @@ def create_site(
     :param tilt: tilt of site, default is 35
     :param inverter_capacity_kw: inverter capacity of site in kw
     :param module_capacity_kw: module capacity of site in kw
+    :param ml_id: internal ML modelling id
 
     """
     max_ml_id = session.query(func.max(SiteSQL.ml_id)).scalar()
@@ -133,7 +135,7 @@ def create_site(
         dno = json.dumps(dno)
 
     site = SiteSQL(
-        ml_id=max_ml_id + 1,
+        ml_id=ml_id if ml_id else max_ml_id + 1,
         client_site_id=client_site_id,
         client_site_name=client_site_name,
         latitude=latitude,

tests/test_write.py

@@ -203,6 +203,34 @@ def test_create_new_site_twice(db_session):
     assert site_2.ml_id == 2
 
 
+# test for create_site to check ml_id duplicates are allowed for separate clients
+def test_ml_id_duplicate_for_unique_clients(db_session):
+    """Test create sites with duplicate ml_id for different clients"""
+
+    site_1, _ = create_site(
+        session=db_session,
+        client_site_id=6932,
+        client_site_name="test_site_name_1",
+        latitude=1.0,
+        longitude=1.0,
+        capacity_kw=1.0,
+        ml_id=1,
+    )
+
+    site_2, _ = create_site(
+        session=db_session,
+        client_site_id=6933,
+        client_site_name="test_site_name_2",
+        latitude=1.0,
+        longitude=1.0,
+        capacity_kw=1.0,
+        ml_id=1,
+    )
+
+    assert site_1.ml_id == 1
+    assert site_2.ml_id == 1
+
+
 def test_create_user(db_session):
     "Test to create a new user."