gitrusprusandCursor
5155b3a37a
Support per-map contact membership with scoped graph views, relation editing on edges, layout caching, and automatic detection of full JSON backups so contacts and relations import together. Co-authored-by: Cursor <cursoragent@cursor.com>
120 lines
5.3 KiB
Python
120 lines
5.3 KiB
Python
import django.core.validators
|
||
from django.db import migrations, models
|
||
import django.db.models.deletion
|
||
|
||
|
||
def migrate_contact_map_data(apps, schema_editor):
|
||
Contact = apps.get_model('contacts', 'Contact')
|
||
NetworkMap = apps.get_model('contacts', 'NetworkMap')
|
||
NetworkMapMembership = apps.get_model('contacts', 'NetworkMapMembership')
|
||
|
||
default_map, _ = NetworkMap.objects.get_or_create(
|
||
name='Основная карта',
|
||
defaults={'description': 'Мигрировано из прежней единой карты сети'},
|
||
)
|
||
|
||
for contact in Contact.objects.filter(include_on_network_map=True):
|
||
NetworkMapMembership.objects.create(
|
||
map=default_map,
|
||
contact=contact,
|
||
life_sphere=contact.life_sphere,
|
||
network_circle=contact.network_circle,
|
||
importance=contact.importance,
|
||
map_angle=contact.map_angle,
|
||
map_radius_ratio=contact.map_radius_ratio,
|
||
)
|
||
|
||
|
||
class Migration(migrations.Migration):
|
||
|
||
dependencies = [
|
||
('contacts', '0005_contact_map_angle_contact_map_radius_ratio'),
|
||
]
|
||
|
||
operations = [
|
||
migrations.CreateModel(
|
||
name='NetworkMap',
|
||
fields=[
|
||
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||
('name', models.CharField(max_length=255, verbose_name='Название')),
|
||
('description', models.TextField(blank=True, verbose_name='Описание')),
|
||
('created_at', models.DateTimeField(auto_now_add=True)),
|
||
('updated_at', models.DateTimeField(auto_now=True)),
|
||
],
|
||
options={
|
||
'verbose_name': 'Карта сети',
|
||
'verbose_name_plural': 'Карты сети',
|
||
'ordering': ['name'],
|
||
},
|
||
),
|
||
migrations.CreateModel(
|
||
name='NetworkMapMembership',
|
||
fields=[
|
||
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||
('life_sphere', models.CharField(
|
||
choices=[
|
||
('work', 'Работа'), ('study', 'Учёба'), ('hobby', 'Хобби'),
|
||
('family', 'Семья'), ('health', 'Здоровье'), ('other', 'Другое'),
|
||
],
|
||
default='other',
|
||
max_length=32,
|
||
verbose_name='Сфера жизни',
|
||
)),
|
||
('network_circle', models.CharField(
|
||
choices=[
|
||
('support', 'Круг поддержки'),
|
||
('productivity', 'Круг продуктивности'),
|
||
('development', 'Круг развития'),
|
||
],
|
||
default='productivity',
|
||
max_length=32,
|
||
verbose_name='Круг сети',
|
||
)),
|
||
('importance', models.PositiveSmallIntegerField(
|
||
default=3,
|
||
validators=[
|
||
django.core.validators.MinValueValidator(1),
|
||
django.core.validators.MaxValueValidator(5),
|
||
],
|
||
verbose_name='Важность (1–5)',
|
||
)),
|
||
('map_angle', models.FloatField(blank=True, null=True, verbose_name='Угол позиции на карте')),
|
||
('map_radius_ratio', models.FloatField(
|
||
blank=True,
|
||
null=True,
|
||
validators=[
|
||
django.core.validators.MinValueValidator(0),
|
||
django.core.validators.MaxValueValidator(1),
|
||
],
|
||
verbose_name='Радиус позиции на карте (доля)',
|
||
)),
|
||
('created_at', models.DateTimeField(auto_now_add=True)),
|
||
('updated_at', models.DateTimeField(auto_now=True)),
|
||
('contact', models.ForeignKey(
|
||
on_delete=django.db.models.deletion.CASCADE,
|
||
related_name='map_memberships',
|
||
to='contacts.contact',
|
||
verbose_name='Контакт',
|
||
)),
|
||
('map', models.ForeignKey(
|
||
on_delete=django.db.models.deletion.CASCADE,
|
||
related_name='memberships',
|
||
to='contacts.networkmap',
|
||
verbose_name='Карта',
|
||
)),
|
||
],
|
||
options={
|
||
'verbose_name': 'Участие на карте',
|
||
'verbose_name_plural': 'Участия на картах',
|
||
'unique_together': {('map', 'contact')},
|
||
},
|
||
),
|
||
migrations.RunPython(migrate_contact_map_data, migrations.RunPython.noop),
|
||
migrations.RemoveField(model_name='contact', name='include_on_network_map'),
|
||
migrations.RemoveField(model_name='contact', name='life_sphere'),
|
||
migrations.RemoveField(model_name='contact', name='network_circle'),
|
||
migrations.RemoveField(model_name='contact', name='importance'),
|
||
migrations.RemoveField(model_name='contact', name='map_angle'),
|
||
migrations.RemoveField(model_name='contact', name='map_radius_ratio'),
|
||
]
|