48 lines
2.1 KiB
Python
48 lines
2.1 KiB
Python
"""回填 lunbo / gonggao / popup_page / tupianpeizhi 的 club_id 与 lunbo.page_key。"""
|
||
from django.core.management.base import BaseCommand
|
||
from django.db import transaction
|
||
|
||
from jituan.constants import CLUB_ID_DEFAULT
|
||
from config.models import Gonggao, Lunbo, PopupPage, Tupianpeizhi
|
||
|
||
|
||
class Command(BaseCommand):
|
||
help = '回填展示配置表的 club_id，并为 lunbo 补 page_key'
|
||
|
||
def handle(self, *args, **options):
|
||
updated = {'lunbo_club': 0, 'lunbo_page': 0, 'gonggao': 0, 'popup': 0, 'tupian': 0}
|
||
|
||
with transaction.atomic():
|
||
for row in Lunbo.query.all().only('id', 'club_id', 'page_key', 'ImageType'):
|
||
fields = []
|
||
if not row.club_id:
|
||
row.club_id = CLUB_ID_DEFAULT
|
||
fields.append('club_id')
|
||
updated['lunbo_club'] += 1
|
||
if not row.page_key:
|
||
if row.ImageType == 2:
|
||
row.page_key = 'dashou_center'
|
||
else:
|
||
row.page_key = 'order_pool'
|
||
fields.append('page_key')
|
||
updated['lunbo_page'] += 1
|
||
if fields:
|
||
row.save(update_fields=fields)
|
||
|
||
for row in Gonggao.query.filter(club_id='').only('id', 'club_id'):
|
||
Gonggao.query.filter(id=row.id).update(club_id=CLUB_ID_DEFAULT)
|
||
updated['gonggao'] += 1
|
||
|
||
for row in PopupPage.query.filter(club_id='').only('id', 'club_id'):
|
||
PopupPage.query.filter(id=row.id).update(club_id=CLUB_ID_DEFAULT)
|
||
updated['popup'] += 1
|
||
|
||
for row in Tupianpeizhi.query.filter(club_id='').only('id', 'club_id'):
|
||
Tupianpeizhi.query.filter(id=row.id).update(club_id=CLUB_ID_DEFAULT)
|
||
updated['tupian'] += 1
|
||
|
||
self.stdout.write(self.style.SUCCESS(
|
||
f'回填完成: lunbo_club={updated["lunbo_club"]}, lunbo_page={updated["lunbo_page"]}, '
|
||
f'gonggao={updated["gonggao"]}, popup={updated["popup"]}, tupian={updated["tupian"]}'
|
||
))
|