Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
65 changes: 58 additions & 7 deletions web-app/django/VIM/apps/instruments/admin.py
Original file line number Diff line number Diff line change
@@ -1,32 +1,83 @@
from django.contrib import admin
from VIM.apps.instruments.models import Instrument, InstrumentName, Language, AVResource
from VIM.apps.instruments.models import (
Instrument,
InstrumentName,
InstrumentNameSource,
Source,
Language,
AVResource,
)

admin.site.register(Instrument)
admin.site.register(InstrumentNameSource)
admin.site.register(Language)
admin.site.register(AVResource)


class InstrumentNameSourceInline(admin.TabularInline):
model = InstrumentNameSource
extra = 1
fields = (
"source",
"contributor",
)
readonly_fields = ()

def get_readonly_fields(self, request, obj=None):
"""
Make fields read-only for users in the 'reviewer' group.
"""
if request.user.groups.filter(name="reviewer").exists():
return ("source", "contributor")
return super().get_readonly_fields(request, obj)


@admin.register(InstrumentName)
class InstrumentNameAdmin(admin.ModelAdmin):
list_filter = ("verification_status", "on_wikidata") # Filter by status
list_filter = ("verification_status", "on_wikidata")
search_fields = (
"name",
"source_name",
"sources__name",
"instrument__wikidata_id",
) # Search by name, source name, and instrument wikidata ID
)
inlines = [InstrumentNameSourceInline]
list_editable = ("verification_status",) # Allow editing of verification_status
list_display = (
"name",
"instrument",
"language",
"umil_label",
"on_wikidata",
"verification_status",
"all_sources",
) # Show all sources in list view

def get_readonly_fields(self, request, obj=None):
"""
Make all fields except 'verification_status' read-only for users in the 'reviewer' group.
Make fields read-only for users in the 'reviewer' group.
"""
if request.user.groups.filter(name="reviewer").exists():
return (
"instrument",
"language",
"name",
"source_name",
"umil_label",
"contributor",
"on_wikidata",
)
return super().get_readonly_fields(request, obj)

def all_sources(self, obj):
"""
Display all associated source names for this instrument name
"""
return ", ".join(obj.sources.all.iterator())

all_sources.short_description = "Sources"


@admin.register(Source)
class SourceAdmin(admin.ModelAdmin):
list_display = ("name", "is_visible")
list_filter = ("is_visible",)
search_fields = ("name",)
list_editable = ("is_visible",) # Allow editing of is_visible
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,14 @@
from django.contrib.auth import get_user_model
from django.core.management.base import BaseCommand
from django.db import transaction
from VIM.apps.instruments.models import Instrument, InstrumentName, Language, AVResource
from VIM.apps.instruments.models import (
Instrument,
InstrumentName,
InstrumentNameSource,
Source,
Language,
AVResource,
)


class Command(BaseCommand):
Expand All @@ -32,6 +39,12 @@ def __init__(self):
f"Default contributor {settings.DDMAL_USERNAME} not found in the database."
)

# Set Wikidata as a source to associate with all instrument names imported from Wikidata dumps
self.wikidata_source, _ = Source.objects.get_or_create(
name="Wikidata",
defaults={"is_visible": True},
)

def parse_instrument_data(
self, instrument_id: str, instrument_data: dict
) -> dict[str, str | dict[str, str]]:
Expand Down Expand Up @@ -142,16 +155,22 @@ def create_database_objects(
)
)
continue
InstrumentName.objects.update_or_create(
name_obj, _ = InstrumentName.objects.update_or_create(
instrument=instrument,
language=self.language_map[lang],
umil_label=True,
defaults={
"name": name,
"source_name": "Wikidata",
"contributor": self.default_contributor,
"verification_status": "verified",
"on_wikidata": True,
"verification_status": "verified",
},
)

InstrumentNameSource.objects.update_or_create(
instrument_name=name_obj,
source=self.wikidata_source,
defaults={
"contributor": self.default_contributor,
},
)

Expand All @@ -166,16 +185,22 @@ def create_database_objects(
)
continue
for alias in aliases:
InstrumentName.objects.update_or_create(
alias_obj, _ = InstrumentName.objects.update_or_create(
instrument=instrument,
language=self.language_map[lang],
name=alias,
umil_label=False,
defaults={
"source_name": "Wikidata",
"contributor": self.default_contributor,
"verification_status": "verified",
"name": alias,
"on_wikidata": True,
"verification_status": "verified",
},
)

InstrumentNameSource.objects.update_or_create(
instrument_name=name_obj,
source=self.wikidata_source,
defaults={
"contributor": self.default_contributor,
},
)

Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,99 @@
# Generated by Django 4.2.5 on 2025-12-04 02:52

from django.conf import settings
from django.db import migrations, models
import django.db.models.deletion


class Migration(migrations.Migration):
dependencies = [
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
("instruments", "0011_language_html_direction"),
]

operations = [
migrations.CreateModel(
name="Source",
fields=[
(
"id",
models.BigAutoField(
auto_created=True,
primary_key=True,
serialize=False,
verbose_name="ID",
),
),
(
"name",
models.CharField(
help_text="Who or what called the instrument this?",
max_length=50,
),
),
(
"is_visible",
models.BooleanField(
default=True, help_text="Should this source be visible?"
),
),
],
),
migrations.RemoveField(
model_name="instrumentname",
name="contributor",
),
migrations.RemoveField(
model_name="instrumentname",
name="source_name",
),
migrations.CreateModel(
name="InstrumentNameSource",
fields=[
(
"id",
models.BigAutoField(
auto_created=True,
primary_key=True,
serialize=False,
verbose_name="ID",
),
),
(
"contributor",
models.ForeignKey(
blank=True,
help_text="Users who contributed this source to this name",
null=True,
on_delete=django.db.models.deletion.SET_NULL,
related_name="instrument_name_sources",
to=settings.AUTH_USER_MODEL,
),
),
(
"instrument_name",
models.ForeignKey(
on_delete=django.db.models.deletion.CASCADE,
related_name="source_links",
to="instruments.instrumentname",
),
),
(
"source",
models.ForeignKey(
on_delete=django.db.models.deletion.CASCADE,
to="instruments.source",
),
),
],
),
migrations.AddField(
model_name="instrumentname",
name="sources",
field=models.ManyToManyField(
related_name="instrument_names",
through="instruments.InstrumentNameSource",
to="instruments.source",
),
),
]
2 changes: 2 additions & 0 deletions web-app/django/VIM/apps/instruments/models/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
from VIM.apps.instruments.models.instrument import Instrument
from VIM.apps.instruments.models.instrument_name import InstrumentName
from VIM.apps.instruments.models.instrument_name_source import InstrumentNameSource
from VIM.apps.instruments.models.source import Source
from VIM.apps.instruments.models.language import Language
from VIM.apps.instruments.models.avresource import AVResource
32 changes: 15 additions & 17 deletions web-app/django/VIM/apps/instruments/models/instrument_name.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,21 @@ class InstrumentName(models.Model):
instrument = models.ForeignKey("Instrument", on_delete=models.CASCADE)
language = models.ForeignKey("Language", on_delete=models.PROTECT)
name = models.CharField(max_length=100, blank=False)
source_name = models.CharField(
max_length=50, blank=False, help_text="Who or what called the instrument this?"
) # Stand-in for source data; format TBD

# An instrument name may be supported by multiple sources, each with its own contributor
sources = models.ManyToManyField(
"Source", through="InstrumentNameSource", related_name="instrument_names"
)
umil_label = models.BooleanField(
default=False,
help_text="Is this the label for the instrument? If true, it will be used as the main name.",
)

on_wikidata = models.BooleanField(
default=False,
help_text="Is this name already on Wikidata?",
)

verification_status = models.CharField(
max_length=50,
choices=[
Expand All @@ -20,20 +32,6 @@ class InstrumentName(models.Model):
default="unverified",
help_text="Status of the name entry",
)
umil_label = models.BooleanField(
default=False,
help_text="Is this the label for the instrument? If true, it will be used as the main name.",
)
contributor = models.ForeignKey(
"auth.User",
null=False,
on_delete=models.PROTECT,
help_text="User who contributed this name",
)
on_wikidata = models.BooleanField(
default=False,
help_text="Is this name already on Wikidata?",
)

# Custom validation to ensure at most one UMIL label per instrument language
class Meta:
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
from django.db import models


class InstrumentNameSource(models.Model):
instrument_name = models.ForeignKey(
"InstrumentName", on_delete=models.CASCADE, related_name="source_links"
)
source = models.ForeignKey("Source", on_delete=models.CASCADE)

contributor = models.ForeignKey(
"auth.User",
on_delete=models.SET_NULL,
help_text="Users who contributed this source to this name",
related_name="instrument_name_sources",
blank=True,
null=True,
)

def __str__(self):
return f"{self.source.name} for {self.instrument_name.name}"
14 changes: 14 additions & 0 deletions web-app/django/VIM/apps/instruments/models/source.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
from django.db import models


class Source(models.Model):
name = models.CharField(
max_length=50, blank=False, help_text="Who or what called the instrument this?"
)

is_visible = models.BooleanField(
default=True, help_text="Should this source be visible?"
)

def __str__(self):
return f"{self.name}"
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,10 @@ def get_context_data(self, **kwargs):

# Query the instrument names in all languages
instrument_names = (
context["instrument"].instrumentname_set.all().select_related("language")
context["instrument"]
.instrumentname_set.all()
.select_related("language")
.prefetch_related("sources")
)
if self.request.user.is_authenticated:
# Show all names for authenticated users
Expand Down
Loading