Make fields nullable again

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

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/models/jobagent.py b/src/models/jobagent.py
@@ -1,5 +1,5 @@
-from sqlalchemy import Column, ForeignKey
-from sqlmodel import SQLModel, Field, Relationship
+from sqlalchemy import ForeignKey
+from sqlmodel import SQLModel, Field, Relationship, Column
 from typing import Union, TYPE_CHECKING
 
 if TYPE_CHECKING:
@@ -14,7 +14,9 @@ class JobAgent(SQLModel, table=True):
     task_in_progress: int
 
     job_id: int = Field(
-        sa_column=Column(ForeignKey("job.internal_id", ondelete="CASCADE"))
+        sa_column=Column(
+            ForeignKey("job.internal_id", ondelete="CASCADE"), nullable=False
+        ),
     )
     job: "Job" = Relationship(back_populates="agents")
 

diff --git a/src/models/match.py b/src/models/match.py
@@ -17,6 +17,8 @@ class Match(SQLModel, table=True):
     matches: List[str] = Field(sa_column=Column(ARRAY(String)))
 
     job_id: int = Field(
-        sa_column=Column(ForeignKey("job.internal_id", ondelete="CASCADE"))
+        sa_column=Column(
+            ForeignKey("job.internal_id", ondelete="CASCADE"), nullable=False
+        )
     )
     job: Job = Relationship(back_populates="matches")