Make fields nullable again

CERT-Polska · Oct 27, 2024 · f399b1c · f399b1c
1 parent 2713b7f
commit f399b1c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/src/models/jobagent.py b/src/models/jobagent.py
@@ -13,9 +13,7 @@ class JobAgent(SQLModel, table=True):
     id: Union[int, None] = Field(default=None, primary_key=True)
     task_in_progress: int
 
-    job_id: int = Field(
-        sa_column=Column(ForeignKey("job.internal_id", ondelete="CASCADE"))
-    )
+    job_id: int = Field(foreign_key="job.internal_id", ondelete="CASCADE")
     job: "Job" = Relationship(back_populates="agents")
 
     agent_id: int = Field(foreign_key="agentgroup.id")

diff --git a/src/models/match.py b/src/models/match.py
@@ -16,7 +16,5 @@ class Match(SQLModel, table=True):
     # A list of yara rules matched to this file
     matches: List[str] = Field(sa_column=Column(ARRAY(String)))
 
-    job_id: int = Field(
-        sa_column=Column(ForeignKey("job.internal_id", ondelete="CASCADE"))
-    )
+    job_id: int = Field(foreign_key="job.internal_id", ondelete="CASCADE")
     job: Job = Relationship(back_populates="matches")