fix flake model (#662)

codecov · Aug 29, 2024 · 3cdf975 · 3cdf975
1 parent 894f2b3
commit 3cdf975
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/database/base.py b/database/base.py
@@ -30,3 +30,15 @@ class MixinBaseClass(object):
     @property
     def id(self):
         return self.id_
+
+
+class MixinBaseClassNoExternalID(object):
+    id_ = Column("id", types.BigInteger, primary_key=True)
+    created_at = Column(types.DateTime(timezone=True), default=get_utc_now)
+    updated_at = Column(
+        types.DateTime(timezone=True), onupdate=get_utc_now, default=get_utc_now
+    )
+
+    @property
+    def id(self):
+        return self.id_
diff --git a/database/models/reports.py b/database/models/reports.py
@@ -9,7 +9,7 @@
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.orm import backref, relationship
 
-from database.base import CodecovBaseModel, MixinBaseClass
+from database.base import CodecovBaseModel, MixinBaseClass, MixinBaseClassNoExternalID
 from database.models.core import Commit, CompareCommit, Repository
 from database.utils import ArchiveField
 from helpers.clock import get_utc_now
@@ -332,7 +332,7 @@ class ReducedError(CodecovBaseModel, MixinBaseClass):
     message = Column(types.Text)
 
 
-class Flake(CodecovBaseModel, MixinBaseClass):
+class Flake(CodecovBaseModel, MixinBaseClassNoExternalID):
     __tablename__ = "reports_flake"
     repoid = Column(types.Integer, ForeignKey("repos.repoid"))
     repository = relationship("Repository", backref=backref("flakes"))