From f82589a4ad3d6501d4b07afe7b1f39467d355147 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Fri, 23 Feb 2018 10:21:53 +0100
Subject: Site labels: fix translation of customization

---
 archaeological_operations/wizards.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'archaeological_operations/wizards.py')

diff --git a/archaeological_operations/wizards.py b/archaeological_operations/wizards.py
index 18c4b20e8..dc96da7be 100644
--- a/archaeological_operations/wizards.py
+++ b/archaeological_operations/wizards.py
@@ -453,21 +453,23 @@ class OperationEditAdministrativeActWizard(OperationAdministrativeActWizard):
 
 
 class SiteLabel(object):
+    SITE_KEY = ""
+
     def get_label(self):
-        return unicode(_(u"Search {}")).format(
-            get_current_profile().get_site_label()
-        )
+        return get_current_profile().get_site_label(self.SITE_KEY)
 
 
 class SiteSearch(SiteLabel, SearchWizard):
-    pass
+    SITE_KEY = "search"
 
 
 class SiteWizard(SiteLabel, Wizard):
+    SITE_KEY = 'new'
     model = models.ArchaeologicalSite
 
 
 class SiteModificationWizard(SiteWizard):
+    SITE_KEY = 'modification'
     modification = True
 
 
-- 
cgit v1.2.3