From ad3966b2aa261751c23dc9b4307cac36b22996a9 Mon Sep 17 00:00:00 2001 From: Dmitriy Kuzmenko Date: Sat, 16 Nov 2019 22:56:32 +0300 Subject: [PATCH] add method for label --- apps/tag/management/commands/add_tags.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/apps/tag/management/commands/add_tags.py b/apps/tag/management/commands/add_tags.py index b0c54e5c..3c43a14d 100644 --- a/apps/tag/management/commands/add_tags.py +++ b/apps/tag/management/commands/add_tags.py @@ -43,11 +43,11 @@ def metadata_tags_sql(): cursor.execute( """ SELECT - value, - `key` as category, - establishment_id -FROM metadata -WHERE establishment_id is not null""" + value, + `key` as category, + establishment_id + FROM metadata + WHERE establishment_id is not null""" ) return namedtuplefetchall(cursor) @@ -64,6 +64,11 @@ class Command(BaseCommand): meta_type = 'bool' return meta_type + def get_label(self, text): + sp = text.key.split('_') + label = ' '.join([sp[0].capitalize()] + sp[1:]) + return label + def handle(self, *args, **kwargs): existing_establishment = Establishment.objects.filter( old_id__isnull=False @@ -86,7 +91,7 @@ class Command(BaseCommand): defaults={ "public": True if meta.public == 1 else False, "value_type": self.get_type(meta), - "label": {"en-GB": meta.key} + "label": {"en-GB": self.get_label(meta.key)} } ) @@ -102,7 +107,7 @@ class Command(BaseCommand): category=TagCategory.objects.get(index_name=meta_tag.category), value=meta_tag.value, defaults={ - "label": {"en-GB": meta_tag.value} + "label": {"en-GB": self.get_label(meta_tag.value)} } ) establishment = existing_establishment.filter(old_id=meta_tag.establishment_id).first()