rucki
141 lines
8.7 KiB
Python
141 lines
8.7 KiB
Python
# Generated by Pi coding agent
|
|
|
|
from django.conf import settings
|
|
from django.db import migrations, models
|
|
import django.db.models.deletion
|
|
|
|
|
|
def forwards(apps, schema_editor):
|
|
User = apps.get_model(*settings.AUTH_USER_MODEL.split('.'))
|
|
Workspace = apps.get_model('core', 'Workspace')
|
|
WorkspaceMembership = apps.get_model('core', 'WorkspaceMembership')
|
|
Tag = apps.get_model('core', 'Tag')
|
|
Team = apps.get_model('core', 'Team')
|
|
Item = apps.get_model('core', 'Item')
|
|
Collection = apps.get_model('core', 'Collection')
|
|
Kanban = apps.get_model('core', 'Kanban')
|
|
ApiKey = apps.get_model('core', 'ApiKey')
|
|
|
|
personal = {}
|
|
for user in User.objects.all():
|
|
full_name = f'{getattr(user, "first_name", "")} {getattr(user, "last_name", "")}'.strip()
|
|
name = full_name or getattr(user, 'username', '') or f'User {user.pk}'
|
|
workspace, _ = Workspace.objects.get_or_create(slug=f'user-{user.pk}', defaults={'name': name, 'owner_id': user.pk})
|
|
WorkspaceMembership.objects.get_or_create(workspace=workspace, user=user, defaults={'role': 'owner'})
|
|
personal[user.pk] = workspace
|
|
|
|
def workspace_for_user(user_id):
|
|
if user_id not in personal:
|
|
user = User.objects.get(pk=user_id)
|
|
workspace, _ = Workspace.objects.get_or_create(slug=f'user-{user.pk}', defaults={'name': getattr(user, 'username', '') or f'User {user.pk}', 'owner_id': user.pk})
|
|
WorkspaceMembership.objects.get_or_create(workspace=workspace, user=user, defaults={'role': 'owner'})
|
|
personal[user_id] = workspace
|
|
return personal[user_id]
|
|
|
|
for team in Team.objects.select_related('owner'):
|
|
team.workspace = workspace_for_user(team.owner_id)
|
|
team.save(update_fields=['workspace'])
|
|
|
|
for item in Item.objects.select_related('owner', 'team'):
|
|
item.workspace = item.team.workspace if item.team_id and item.team.workspace_id else workspace_for_user(item.owner_id)
|
|
item.save(update_fields=['workspace'])
|
|
|
|
for collection in Collection.objects.select_related('owner', 'team'):
|
|
collection.workspace = collection.team.workspace if collection.team_id and collection.team.workspace_id else workspace_for_user(collection.owner_id)
|
|
collection.save(update_fields=['workspace'])
|
|
|
|
for kanban in Kanban.objects.select_related('owner'):
|
|
kanban.workspace = workspace_for_user(kanban.owner_id)
|
|
kanban.save(update_fields=['workspace'])
|
|
|
|
for api_key in ApiKey.objects.select_related('owner'):
|
|
api_key.workspace = workspace_for_user(api_key.owner_id)
|
|
api_key.save(update_fields=['workspace'])
|
|
|
|
fallback_workspace = next(iter(personal.values()), None)
|
|
for tag in list(Tag.objects.all()):
|
|
workspace_ids = set(tag.items.exclude(workspace__isnull=True).values_list('workspace_id', flat=True))
|
|
workspace_ids.update(tag.tag_collections.exclude(workspace__isnull=True).values_list('workspace_id', flat=True))
|
|
workspace_ids.update(tag.kanbans.exclude(workspace__isnull=True).values_list('workspace_id', flat=True))
|
|
workspace_ids.update(tag.api_keys.exclude(workspace__isnull=True).values_list('workspace_id', flat=True))
|
|
workspace_ids.update(Team.objects.filter(slug=tag.name).exclude(workspace__isnull=True).values_list('workspace_id', flat=True))
|
|
if not workspace_ids and fallback_workspace:
|
|
workspace_ids.add(fallback_workspace.pk)
|
|
if not workspace_ids:
|
|
tag.delete()
|
|
continue
|
|
|
|
first = True
|
|
for workspace_id in sorted(workspace_ids):
|
|
if first:
|
|
scoped_tag = tag
|
|
scoped_tag.workspace_id = workspace_id
|
|
scoped_tag.save(update_fields=['workspace'])
|
|
first = False
|
|
else:
|
|
scoped_tag, _ = Tag.objects.get_or_create(workspace_id=workspace_id, name=tag.name)
|
|
|
|
for item in list(tag.items.filter(workspace_id=workspace_id)):
|
|
item.tags.remove(tag)
|
|
item.tags.add(scoped_tag)
|
|
for collection in list(tag.tag_collections.filter(workspace_id=workspace_id)):
|
|
collection.filter_tags.remove(tag)
|
|
collection.filter_tags.add(scoped_tag)
|
|
for api_key in list(tag.api_keys.filter(workspace_id=workspace_id)):
|
|
api_key.tags.remove(tag)
|
|
api_key.tags.add(scoped_tag)
|
|
Kanban.objects.filter(workspace_id=workspace_id, tag=tag).update(tag=scoped_tag)
|
|
|
|
for team in Team.objects.exclude(workspace__isnull=True):
|
|
Tag.objects.get_or_create(workspace=team.workspace, name=team.slug)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
|
|
('core', '0021_collection_type_filters'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.CreateModel(
|
|
name='Workspace',
|
|
fields=[
|
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
|
('name', models.CharField(max_length=120)),
|
|
('slug', models.SlugField(allow_unicode=True, max_length=80, unique=True)),
|
|
('created_at', models.DateTimeField(auto_now_add=True)),
|
|
('owner', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='owned_workspaces', to=settings.AUTH_USER_MODEL)),
|
|
],
|
|
options={'ordering': ['name']},
|
|
),
|
|
migrations.CreateModel(
|
|
name='WorkspaceMembership',
|
|
fields=[
|
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
|
('role', models.CharField(choices=[('owner', 'Owner'), ('admin', 'Admin'), ('member', 'Member')], default='member', max_length=10)),
|
|
('created_at', models.DateTimeField(auto_now_add=True)),
|
|
('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='workspace_memberships', to=settings.AUTH_USER_MODEL)),
|
|
('workspace', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='memberships', to='core.workspace')),
|
|
],
|
|
options={'unique_together': {('workspace', 'user')}},
|
|
),
|
|
migrations.AddField('tag', 'workspace', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, related_name='tags', to='core.workspace')),
|
|
migrations.AlterField('tag', 'name', models.SlugField(allow_unicode=True, max_length=80)),
|
|
migrations.AddField('team', 'workspace', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, related_name='teams', to='core.workspace')),
|
|
migrations.AlterField('team', 'slug', models.SlugField(allow_unicode=True, help_text='Team-Tag ohne #, z.B. my2dos', max_length=80)),
|
|
migrations.AddField('item', 'workspace', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, related_name='items', to='core.workspace')),
|
|
migrations.AddField('collection', 'workspace', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, related_name='collections', to='core.workspace')),
|
|
migrations.AddField('kanban', 'workspace', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, related_name='kanbans', to='core.workspace')),
|
|
migrations.AddField('apikey', 'workspace', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, related_name='api_keys', to='core.workspace')),
|
|
migrations.RunPython(forwards, migrations.RunPython.noop),
|
|
migrations.AlterField('tag', 'workspace', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='tags', to='core.workspace')),
|
|
migrations.AlterField('team', 'workspace', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='teams', to='core.workspace')),
|
|
migrations.AlterField('item', 'workspace', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='items', to='core.workspace')),
|
|
migrations.AlterField('collection', 'workspace', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='collections', to='core.workspace')),
|
|
migrations.AlterField('kanban', 'workspace', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='kanbans', to='core.workspace')),
|
|
migrations.AlterField('apikey', 'workspace', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='api_keys', to='core.workspace')),
|
|
migrations.AddConstraint('tag', models.UniqueConstraint(fields=('workspace', 'name'), name='unique_workspace_tag_name')),
|
|
migrations.AddConstraint('team', models.UniqueConstraint(fields=('workspace', 'slug'), name='unique_workspace_team_slug')),
|
|
]
|