fix: handle new flake model (#958)

codecov · Dec 16, 2024 · 536c4e0 · 536c4e0
1 parent 0f8147b
commit 536c4e0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/database/models/reports.py b/database/models/reports.py
@@ -348,8 +348,9 @@ def id(self):
         return self.id_
 
 
-class Flake(CodecovBaseModel, MixinBaseClassNoExternalID):
+class Flake(CodecovBaseModel):
     __tablename__ = "reports_flake"
+    id_ = Column("id", types.BigInteger, primary_key=True)
     repoid = Column(types.Integer, ForeignKey("repos.repoid"))
     repository = relationship("Repository", backref=backref("flakes"))
 
@@ -367,6 +368,10 @@ class Flake(CodecovBaseModel, MixinBaseClassNoExternalID):
     start_date = Column(types.DateTime)
     end_date = Column(types.DateTime, nullable=True)
 
+    @property
+    def id(self):
+        return self.id_
+
 
 class DailyTestRollup(CodecovBaseModel, MixinBaseClassNoExternalID):
     __tablename__ = "reports_dailytestrollups"

diff --git a/requirements.in b/requirements.in
@@ -1,5 +1,5 @@
 https://github.com/codecov/test-results-parser/archive/c840502d1b4dd7d05b2efc2c1328affaf2acd27c.tar.gz#egg=test-results-parser
-https://github.com/codecov/shared/archive/e41a4a1d4dd4efe44a917b47032f1a0d73775426.tar.gz#egg=shared
+https://github.com/codecov/shared/archive/2674ae99811767e63151590906691aed4c5ce1f9.tar.gz#egg=shared
 https://github.com/codecov/timestring/archive/d37ceacc5954dff3b5bd2f887936a98a668dda42.tar.gz#egg=timestring
 asgiref>=3.7.2
 analytics-python==1.3.0b1

diff --git a/requirements.txt b/requirements.txt
@@ -328,7 +328,7 @@ sentry-sdk==2.13.0
     #   shared
 setuptools==75.6.0
     # via nodeenv
-shared @ https://github.com/codecov/shared/archive/e41a4a1d4dd4efe44a917b47032f1a0d73775426.tar.gz#egg=shared
+shared @ https://github.com/codecov/shared/archive/2674ae99811767e63151590906691aed4c5ce1f9.tar.gz#egg=shared
     # via -r requirements.in
 six==1.16.0
     # via