diff --git a/apps/establishment/management/commands/add_employee.py b/apps/establishment/management/commands/add_employee.py index 49eaa056..6fbf562f 100644 --- a/apps/establishment/management/commands/add_employee.py +++ b/apps/establishment/management/commands/add_employee.py @@ -2,7 +2,7 @@ from django.core.management.base import BaseCommand from django.db import connections from establishment.management.commands.add_position import namedtuplefetchall from establishment.models import Employee - +from django.db.models import Q class Command(BaseCommand): @@ -30,8 +30,7 @@ class Command(BaseCommand): def handle(self, *args, **options): objects = [] for e in self.employees_sql(): - count = Employee.objects.filter(old_id=e.profile_id).count() - count += Employee.objects.filter(name=e.name).count() + count = Employee.objects.filter(Q(old_id=e.profile_id) | Q(name=e.name)).count() if count == 0: objects.append(Employee(name=e.name, old_id=e.profile_id)) print(e.name)