diff options
-rw-r--r-- | archaeological_operations/wizards.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/archaeological_operations/wizards.py b/archaeological_operations/wizards.py index 1e4de70b7..b31e7b7f6 100644 --- a/archaeological_operations/wizards.py +++ b/archaeological_operations/wizards.py @@ -109,9 +109,18 @@ class OperationWizard(Wizard): def get_available_parcels(self, file): parcels = [] + current_parcels = [] + operation = self.get_current_object() + for parcel in operation.parcels.all(): + current_parcels.append((parcel.town, parcel.section, + parcel.parcel_number)) + parcels.append((parcel.pk, parcel.short_label)) try: - parcels = [(parcel.pk, parcel.short_label) - for parcel in file.parcels.all()] + for parcel in file.parcels.all(): + if (parcel.town, parcel.section, parcel.parcel_number) \ + in current_parcels: + continue + parcels.append((parcel.pk, parcel.short_label)) except (ValueError, ObjectDoesNotExist): pass return sorted(parcels, key=lambda x:x[1]) |