49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
|
from django.db import migrations, models
|
||
|
|
|
||
|
|
|
||
|
|
def seed_billboard_applets(apps, schema_editor):
|
||
|
|
Applet = apps.get_model("applets", "Applet")
|
||
|
|
for slug, name, cols, rows in [
|
||
|
|
("billboard-my-scrolls", "My Scrolls", 4, 3),
|
||
|
|
("billboard-my-contacts", "Contacts", 4, 3),
|
||
|
|
("billboard-most-recent", "Most Recent", 8, 6),
|
||
|
|
]:
|
||
|
|
Applet.objects.get_or_create(
|
||
|
|
slug=slug,
|
||
|
|
defaults={"name": name, "grid_cols": cols, "grid_rows": rows, "context": "billboard"},
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def remove_billboard_applets(apps, schema_editor):
|
||
|
|
Applet = apps.get_model("applets", "Applet")
|
||
|
|
Applet.objects.filter(slug__in=[
|
||
|
|
"billboard-my-scrolls",
|
||
|
|
"billboard-my-contacts",
|
||
|
|
"billboard-most-recent",
|
||
|
|
]).delete()
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
|
||
|
|
dependencies = [
|
||
|
|
("applets", "0005_gameboard_applet_heights"),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.AlterField(
|
||
|
|
model_name="applet",
|
||
|
|
name="context",
|
||
|
|
field=models.CharField(
|
||
|
|
choices=[
|
||
|
|
("dashboard", "Dashboard"),
|
||
|
|
("gameboard", "Gameboard"),
|
||
|
|
("wallet", "Wallet"),
|
||
|
|
("billboard", "Billboard"),
|
||
|
|
],
|
||
|
|
default="dashboard",
|
||
|
|
max_length=20,
|
||
|
|
),
|
||
|
|
),
|
||
|
|
migrations.RunPython(seed_billboard_applets, remove_billboard_applets),
|
||
|
|
]
|