From 0b537ed1c177990db9cf6e4ad09a550561a120bb Mon Sep 17 00:00:00 2001
From: Jakub K <dzejbobix@gmail.com>
Date: Sun, 17 Mar 2024 17:38:47 +0100
Subject: [PATCH] =?UTF-8?q?wyb=C3=B3r=20p=C3=B3l=20w=20wfh,=20employment?=
 =?UTF-8?q?=5Ftypes?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 jobposting/models.py | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/jobposting/models.py b/jobposting/models.py
index 426dfa0..474c57f 100644
--- a/jobposting/models.py
+++ b/jobposting/models.py
@@ -52,6 +52,20 @@ class JobOffer(models.Model):
         ('G', 'Inne'),
     ]
 
+    work_from_home = [
+        ('wfh', 'Praca zdalna'),
+        ('hyb', 'Hybrydowa'),
+        ('off', 'Stacjonarna'),    
+    ]
+    
+    employment_types = [
+        ('B2B', 'Kontrakt B2B'),
+        ('FT', 'Umowa o pracę'),
+        ('MC', 'Umowa zlecenie'),
+        ('CW', 'Umowa o dzieło'),
+        ('INT', 'Staż'),
+    ]
+
     posting_option = models.CharField(max_length=1, choices=posting_options)
     company_name = models.CharField(max_length=255)
     name = models.CharField(max_length=255)
@@ -71,8 +85,8 @@ class JobOffer(models.Model):
     expiration_date = models.DateTimeField(null=True, blank=True)
     image = models.CharField(max_length=255, null=True, blank=True)
     experience_level = models.CharField(max_length=255, choices=experience_levels, default='G')
-    employment_type = models.CharField(max_length=255, null=True, blank=True)
-    work_from_home = models.CharField(max_length=255)
+    employment_type = models.CharField(max_length=3, choices=employment_types, default='FT')
+    work_from_home = models.CharField(max_length=3, choices=work_from_home, default='off')
     updated_at = models.DateTimeField(auto_now=True)
     skill_levels = models.JSONField(null=True, blank=True, default=list)
     anonymous_user_data = models.ForeignKey(AnonymousUserData,
@@ -95,6 +109,7 @@ class JobOffer(models.Model):
     def __str__(self):
         return f"{self.name} at {self.company_name}"
 
+
 class Skill(models.Model):
     skill_name = models.CharField(max_length=255)
     
@@ -107,4 +122,4 @@ class CompanyLogo(models.Model):
     company_name = models.CharField(max_length=255)
 
     def __str__(self):
-        return f"{self.company_name}"
\ No newline at end of file
+        return f"{self.company_name}"