diff --git a/apps/product/management/commands/add_public_mark.py b/apps/product/management/commands/add_public_mark.py new file mode 100644 index 00000000..b6bcc296 --- /dev/null +++ b/apps/product/management/commands/add_public_mark.py @@ -0,0 +1,25 @@ +from django.core.management.base import BaseCommand +from tqdm import tqdm + +from product.models import Product + + +class Command(BaseCommand): + help = """Add public_mark to product from reviews.""" + + def handle(self, *args, **kwarg): + update_products = [] + products = Product.objects.filter( + public_mark__isnull=True, + reviews__isnull=False).distinct() + + for product in tqdm(products): + review = product.reviews.published().filter( + mark__isnull=False).order_by('-published_at').first() + if review: + product.public_mark = review.mark + update_products.append(product) + + Product.objects.bulk_update(update_products, ['public_mark', ]) + self.stdout.write( + self.style.WARNING(f'Updated products: {len(update_products)}')) \ No newline at end of file diff --git a/make_data_migration.sh b/make_data_migration.sh index bed1afb7..70e27ccb 100755 --- a/make_data_migration.sh +++ b/make_data_migration.sh @@ -2,6 +2,7 @@ ./manage.py transfer -a ./manage.py transfer -d ./manage.py transfer -e +./manage.py upd_transportation ./manage.py transfer --fill_city_gallery ./manage.py transfer -l ./manage.py transfer --product