import sys
|
import logging
|
|
from django.db import migrations
|
|
logger = logging.getLogger(__name__)
|
|
|
def forwards(apps, schema_editor):
|
from tasks.functions import calculate_stats_all_orgs
|
from django.conf import settings
|
|
if settings.VERSION_EDITION == 'Community':
|
run_command = 'label-studio calculate_stats_all_orgs'
|
else:
|
run_command = 'cd /label-studio-enterprise/label_studio_enterprise && ' \
|
'python3 manage.py calculate_stats_all_orgs'
|
|
if '--skip-long-migrations' in sys.argv:
|
logger.error(
|
f"You used --skip-long-migrations, so you should run the migration manually as a separate process "
|
f"to recalculate task counters, please use Django command `{run_command}`"
|
)
|
return
|
|
logger.debug('=> Starting calculate_stats_all_orgs for task counters again')
|
calculate_stats_all_orgs(from_scratch=True, redis=True, migration_name='data_manager-0008_manual_counters_update')
|
|
|
def backwards(apps, schema_editor):
|
pass
|
|
|
class Migration(migrations.Migration):
|
atomic = False
|
|
dependencies = [
|
('data_manager', '0007_auto_20220708_0832'),
|
('tasks', '0023_auto_20220620_1007'),
|
('core', '0001_initial'),
|
('projects', '0017_project_pinned_at'),
|
]
|
|
operations = [
|
migrations.RunPython(forwards, backwards),
|
]
|
