Make fields nullable again

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

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -22,7 +22,7 @@ redis==5.0.8
 requests==2.32.2
 rq==1.16.2
 sniffio==1.3.1
-sqlmodel==0.0.11
+sqlmodel==0.0.22
 starlette==0.38.6
 typed-config==2.0.3
 types-requests==2.32.0.20240914

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")