173 lines
11 KiB
Python
173 lines
11 KiB
Python
from django.core.management.base import BaseCommand
|
|
from establishment.models import Establishment
|
|
from location.models import Country, Language
|
|
from transfer.models import Collections
|
|
from collection.models import Collection
|
|
from django.conf import settings
|
|
from news.models import News
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Import collection'
|
|
|
|
def handle(self, *args, **kwargs):
|
|
raw_qs = Collections.objects.raw('''
|
|
select
|
|
distinct
|
|
a.id,
|
|
a.collection_id,
|
|
-- a.establishment_id,
|
|
a.title,
|
|
a.tag_name,
|
|
a.slug,
|
|
-- a.attachment_file_name,
|
|
-- a.attachment_content_type,
|
|
-- a.attachment_file_size,
|
|
a.attachment_suffix_url,
|
|
-- active as is_publish,
|
|
a.country_code,
|
|
-- a.geometries,
|
|
a.description,
|
|
min(a.start) AS start
|
|
from
|
|
(
|
|
select distinct
|
|
c.id,
|
|
c.id as collection_id,
|
|
m.establishment_id,
|
|
c.title, c.tag_name,
|
|
c.slug, c.attachment_file_name,
|
|
c.attachment_content_type, c.attachment_file_size,
|
|
c.attachment_suffix_url,
|
|
active,
|
|
s.country_code_2 as country_code,
|
|
c.geometries,
|
|
c.title as description,
|
|
m.created_at as start
|
|
from collections as c
|
|
join metadata m on m.value = c.tag_name
|
|
join establishments e on e.id = m.establishment_id
|
|
join sites s on s.id = c.site_id
|
|
where m.`key` = 'collection'
|
|
|
|
union
|
|
|
|
select distinct
|
|
c.id,
|
|
c.id as collection_id,
|
|
m.establishment_id,
|
|
c.title, c.tag_name,
|
|
c.slug, c.attachment_file_name,
|
|
c.attachment_content_type, c.attachment_file_size,
|
|
c.attachment_suffix_url,
|
|
active,
|
|
s.country_code_2 as country_code,
|
|
c.geometries,
|
|
c.title as description,
|
|
m.created_at as start
|
|
from collections as c
|
|
join metadata m on m.value = c.slug
|
|
join establishments e on e.id = m.establishment_id
|
|
join sites s on s.id = c.site_id
|
|
where m.`key` = 'collection'
|
|
) a
|
|
group by
|
|
a.id,
|
|
a.collection_id,
|
|
-- a.establishment_id,
|
|
a.title,
|
|
a.tag_name,
|
|
a.slug,
|
|
-- a.attachment_file_name,
|
|
-- a.attachment_content_type,
|
|
-- a.attachment_file_size,
|
|
a.attachment_suffix_url,
|
|
-- active as is_publish,
|
|
a.country_code,
|
|
a.description
|
|
''')
|
|
objects = []
|
|
queryset = [vars(query) for query in raw_qs]
|
|
for obj in queryset:
|
|
# establishment = Establishment.objects.filter(old_id=obj['establishment_id']).first()
|
|
# lang = Language.objects.filter(locale=obj['country_code'])
|
|
country = Country.objects.filter(code=obj['country_code']).first()
|
|
if country:
|
|
objects.append(
|
|
Collection(name={settings.FALLBACK_LOCALE: obj['title']},
|
|
collection_type=Collection.POP if obj['description'].lower().find('pop') != -1
|
|
else Collection.ORDINARY,
|
|
country=country,
|
|
description={settings.FALLBACK_LOCALE: obj['description']},
|
|
slug=obj['slug'], old_id=obj['collection_id'],
|
|
start=obj['start'],
|
|
image_url='https://s3.eu-central-1.amazonaws.com/gm-test.com/media/'+obj['attachment_suffix_url']
|
|
)
|
|
)
|
|
Collection.objects.bulk_create(objects)
|
|
|
|
raw_qs = Collections.objects.raw('''
|
|
select
|
|
distinct
|
|
a.id,
|
|
a.collection_id,
|
|
a.establishment_id,
|
|
a.active
|
|
from
|
|
(
|
|
select distinct
|
|
c.id,
|
|
c.id as collection_id,
|
|
m.establishment_id,
|
|
c.title, c.tag_name,
|
|
c.slug, c.attachment_file_name,
|
|
c.attachment_content_type, c.attachment_file_size,
|
|
c.attachment_suffix_url,
|
|
active,
|
|
s.country_code_2 as country_code,
|
|
c.geometries,
|
|
c.title as description,
|
|
m.created_at as start
|
|
from collections as c
|
|
join metadata m on m.value = c.tag_name
|
|
join establishments e on e.id = m.establishment_id
|
|
join sites s on s.id = c.site_id
|
|
where m.`key` = 'collection'
|
|
|
|
union
|
|
|
|
select distinct
|
|
c.id,
|
|
c.id as collection_id,
|
|
m.establishment_id,
|
|
c.title, c.tag_name,
|
|
c.slug, c.attachment_file_name,
|
|
c.attachment_content_type, c.attachment_file_size,
|
|
c.attachment_suffix_url,
|
|
active,
|
|
s.country_code_2 as country_code,
|
|
c.geometries,
|
|
c.title as description,
|
|
m.created_at as start
|
|
from collections as c
|
|
join metadata m on m.value = c.slug
|
|
join establishments e on e.id = m.establishment_id
|
|
join sites s on s.id = c.site_id
|
|
where m.`key` = 'collection'
|
|
) a
|
|
''')
|
|
|
|
queryset = [vars(query) for query in raw_qs]
|
|
for obj in queryset:
|
|
print('COLLECT_ID: {}'.format(obj['collection_id']))
|
|
est = Establishment.objects.filter(old_id=obj['establishment_id'])
|
|
if est.exists():
|
|
inst = est.first()
|
|
collect = Collection.objects.filter(old_id=obj['collection_id'])
|
|
collect.update(is_publish=obj['active'])
|
|
print(f'COLLECT COUNT {collect.count()}')
|
|
inst.collections.add(*list(collect))
|
|
# for c in collect:
|
|
# inst.collections.add(c)
|
|
inst.save()
|