From 57f3aaf575d0feeb0847af8c095221f0d13658ff Mon Sep 17 00:00:00 2001
From: Yariv Menachem <yariv220895@gmail.com>
Date: Tue, 31 Dec 2024 17:38:18 +0200
Subject: [PATCH] fixed location in db with migration

---
 src/jobspy/jobs/__init__.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/jobspy/jobs/__init__.py b/src/jobspy/jobs/__init__.py
index 970493d..21eda50 100644
--- a/src/jobspy/jobs/__init__.py
+++ b/src/jobspy/jobs/__init__.py
@@ -182,10 +182,17 @@ class Country(Enum):
 
 
 class Location(BaseModel):
-    country: Country | str | None = None
+    country: Optional[str] = None
     city: Optional[str] = None
     state: Optional[str] = None
-    text: str = None
+    text: str = ""
+
+    def dict(self, *args, **kwargs):
+        """
+        Override the dict() method to exclude fields with None values.
+        """
+        data = super().model_dump(*args, **kwargs)
+        return {k: v for k, v in data.items() if v is not None}
 
     def display_location(self) -> str:
         location_parts = []