translate workspace

Author: n0rdlicht

django/gsmap/admin.py

@@ -1,10 +1,10 @@
 from django.contrib.gis import admin
+from django.contrib.admin import SimpleListFilter
 from django.contrib.postgres import fields
 from django.utils.translation import gettext as _
 from django_json_widget.widgets import JSONEditorWidget
 from django.utils.html import mark_safe
 from django.contrib import messages
-from django.forms import ModelForm
 from django.forms.widgets import Textarea, TextInput
 import requests
 from parler.admin import TranslatableAdmin
@@ -118,7 +118,7 @@ def save_model(self, request, obj, form, change):
             )
 
 
-class WorkspaceAdmin(admin.OSMGeoAdmin):
+class WorkspaceAdmin(TranslatableAdmin):
     readonly_fields = ('id', 'created', 'modified', 'get_absolute_link')
     fieldsets = (
         (_('Meta'), {
@@ -127,7 +127,7 @@ class WorkspaceAdmin(admin.OSMGeoAdmin):
             )
         }),
         (_('Main'), {
-            'fields': ('title', 'description', 'snapshots'),
+            'fields': ('group', 'title', 'description', 'snapshots'),
         }),
         (_('Annotations'), {
             'fields': (
@@ -138,18 +138,61 @@ class WorkspaceAdmin(admin.OSMGeoAdmin):
             )
         }),
     )
-    list_display = ('id', 'title', 'annotations_open', 'created', 'modified')
-    search_fields = ['title']
+    list_display = ('title', 'group', 'get_absolute_link', 'mode', 'annotations_contact_name', 'annotations_contact_email', 'findme_enabled', 'annotations_open', 'annotations_likes_enabled', 'polygon_open', 'polygon_likes_enabled', 'created', 'modified')
+    list_filter = ['mode']
+    search_fields = ['title', 'description']
 
     def formfield_for_manytomany(self, db_field, request=None, **kwargs):
         if db_field.name == 'snapshots' or db_field.name == 'categories' or db_field.name == 'usergroups':
             kwargs['widget'] = SortedFilteredSelectMultiple()
         return super().formfield_for_manytomany(db_field, request, **kwargs)
+    
+    def get_queryset(self, request):
+        qs = super().get_queryset(request)
+        if not request.user.is_superuser:
+            return qs.select_related('group').filter(group__in=request.user.groups.all())
+        return qs
 
 class AttachementInline(admin.TabularInline):
     model = Attachement
 
-class AnnotationAdmin(admin.OSMGeoAdmin):
+class AnnotationWorkspaceFilter(SimpleListFilter):
+    title = _('Workspace')
+    parameter_name = 'workspace'
+
+    def lookups(self, request, model_admin):
+        return [(x.id, x.title) for x in Workspace.objects.all() if request.user.is_superuser or x.group in request.user.groups.all()]
+    
+    def queryset(self, request, queryset):
+        if self.value():
+            return queryset.filter(workspace__id=self.value())
+        return queryset
+
+class CategoryGroupFilter(SimpleListFilter):
+    title = _('Category')
+    parameter_name = 'category'
+
+    def lookups(self, request, model_admin):
+        return [(x.id, f'{x.group.name}/{x.name}') for x in Category.objects.all() if request.user.is_superuser or x.group in request.user.groups.all()]
+    
+    def queryset(self, request, queryset):
+        if self.value():
+            return queryset.filter(category__id=self.value())
+        return queryset
+
+class UsergroupGroupFilter(SimpleListFilter):
+    title = _('Usergroup')
+    parameter_name = 'usergroup'
+
+    def lookups(self, request, model_admin):
+        return [(x.pk, f'{x.group.name}/{x.name}') for x in Usergroup.objects.all() if request.user.is_superuser or x.group in request.user.groups.all()]
+    
+    def queryset(self, request, queryset):
+        if self.value():
+            return queryset.filter(usergroup__pk=self.value())
+        return queryset
+
+class AnnotationAdmin(admin.ModelAdmin):
     readonly_fields = ('id','created', 'modified')
     fieldsets = (
         (_('Meta'), {
@@ -175,9 +218,15 @@ class AnnotationAdmin(admin.OSMGeoAdmin):
         'workspace',
     )
     inlines = [ AttachementInline, ]
-    list_filter = ('workspace', 'category', 'kind', 'usergroup')
+    list_filter = (AnnotationWorkspaceFilter, CategoryGroupFilter, 'kind', UsergroupGroupFilter,)
     search_fields = ('id', 'data')
 
+    def get_queryset(self, request): 
+        queryset = super().get_queryset(request)
+        if not request.user.is_superuser:
+            return queryset.select_related('workspace__group').filter(workspace__group__in=request.user.groups.all())
+        return queryset
+
 class CategoryAdminForm(TranslatableModelForm):
     class Meta:
         model = Category
@@ -197,22 +246,28 @@ class CategoryAdmin(TranslatableAdmin): # admin.OSMGeoAdmin,
     readonly_fields = ('id', 'created', 'modified')
     fieldsets = (
         (_('Meta'), {
-            'fields': ('deleted', 'hide_in_list', 'namespace'),
+            'fields': ('deleted', 'hide_in_list'),
         }),
         (_('Category'), {
-            'fields': ('name', 'icon', 'color'),
+            'fields': ('group', 'name', 'icon', 'color'),
         }),
     )
 
     list_display = (
         'name',
-        'namespace',
+        'group',
         'color',
         'hide_in_list'
     )
 
-    list_filter = ('namespace', 'hide_in_list', 'color')
-    search_fields = ('id', 'translations__name', 'namespace')
+    list_filter = (CategoryGroupFilter, 'hide_in_list', 'color')
+    search_fields = ('id', 'translations__name')
+
+    def get_queryset(self, request):
+        qs = super().get_queryset(request)
+        if not request.user.is_superuser:
+            return qs.select_related('group').filter(group__in=request.user.groups.all())
+        return qs
 
 class UsergroupAdmin(TranslatableAdmin): # admin.OSMGeoAdmin, 
     readonly_fields = ('created', 'modified')
@@ -221,17 +276,32 @@ class UsergroupAdmin(TranslatableAdmin): # admin.OSMGeoAdmin,
             'fields': ('deleted', 'created', 'modified'),
         }),
         (_('Category'), {
-            'fields': ('key', 'name'),
+            'fields': ('group', 'key', 'name'),
         }),
     )
 
     list_display = (
         'key',
+        'group',
         'name',
     )
 
     search_fields = ('id', 'name', 'key')
 
+    def get_queryset(self, request):
+        qs = super().get_queryset(request)
+        if not request.user.is_superuser:
+            return qs.select_related('group').filter(group__in=request.user.groups.all())
+        return qs
+
+def _has_change_permission(needed, group, user_groups):
+    for g in user_groups:
+        if g == group:
+            for p in g.permissions.all():
+                if p.codename == needed:
+                    return True
+    return False
+
 admin.site.register(Municipality, MunicipalityAdmin)
 admin.site.register(Snapshot, SnapshotAdmin)
 admin.site.register(Workspace, WorkspaceAdmin)

django/gsmap/migrations/0013_auto_20200405_1555.py

@@ -3,6 +3,7 @@
 from django.db import migrations, models
 import gsmap.models
 import sortedm2m.fields
+import parler.models
 
 
 class Migration(migrations.Migration):
@@ -30,5 +31,6 @@ class Migration(migrations.Migration):
             options={
                 'ordering': ['-created'],
             },
+            bases=(parler.models.TranslatableModelMixin, models.Model),
         ),
     ]

django/gsmap/migrations/0037_add_translation_model.py

@@ -0,0 +1,46 @@
+# Generated by Django 3.1.13 on 2022-02-23 13:23
+
+from django.db import migrations, models
+import django.db.models.deletion
+import parler.fields
+import parler.models
+import sortedm2m.fields
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('gsmap', '0036_auto_20220223_1048'),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name='workspace',
+            old_name='title',
+            new_name='_title',
+        ),
+        migrations.RenameField(
+            model_name='workspace',
+            old_name='description',
+            new_name='_description',
+        ),
+        migrations.CreateModel(
+            name='WorkspaceTranslation',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('language_code', models.CharField(db_index=True, max_length=15, verbose_name='Language')),
+                ('title', models.CharField(default='', max_length=150)),
+                ('description', models.TextField(default='')),
+                ('master', parler.fields.TranslationsForeignKey(editable=False, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='translations', to='gsmap.workspace')),
+            ],
+            options={
+                'verbose_name': 'workspace Translation',
+                'db_table': 'gsmap_workspace_translation',
+                'db_tablespace': '',
+                'managed': True,
+                'default_permissions': (),
+                'unique_together': {('language_code', 'master')},
+            },
+            bases=(parler.models.TranslatedFieldsModelMixin, models.Model),
+        ),
+    ]

django/gsmap/migrations/0038_migrate_translatable_fields.py

@@ -0,0 +1,52 @@
+# Generated by Django 3.1.13 on 2022-02-23 13:24
+
+from django.db import migrations
+from django.conf import settings
+from django.core.exceptions import ObjectDoesNotExist
+
+
+def forwards_func(apps, schema_editor):
+    Workspace = apps.get_model('gsmap', 'Workspace')
+    WorkspaceTranslation = apps.get_model('gsmap', 'WorkspaceTranslation')
+
+    for object in Workspace.objects.all():
+        WorkspaceTranslation.objects.create(
+            master_id=object.pk,
+            language_code=settings.PARLER_DEFAULT_LANGUAGE_CODE,
+            title=object._title,
+            description=object._description
+        )
+
+def backwards_func(apps, schema_editor):
+    Workspace = apps.get_model('gsmap', 'Workspace')
+    WorkspaceTranslation = apps.get_model('gsmap', 'WorkspaceTranslation')
+
+    for object in Workspace.objects.all():
+        translation = _get_translation(object, WorkspaceTranslation)
+        object._title = translation.title
+        object._description = translation.description
+        object.save()   # Note this only calls Model.save()
+
+def _get_translation(object, WorkspaceTranslation):
+    translations = WorkspaceTranslation.objects.filter(master_id=object.pk)
+    try:
+        # Try default translation
+        return translations.get(language_code=settings.PARLER_DEFAULT_LANGUAGE_CODE)
+    except ObjectDoesNotExist:
+        try:
+            # Try default language
+            return translations.get(language_code=settings.LANGUAGE_CODE)
+        except ObjectDoesNotExist:
+            # Maybe the object was translated only in a specific language?
+            # Hope there is a single translation
+            return translations.get()
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('gsmap', '0037_add_translation_model'),
+    ]
+
+    operations = [
+        migrations.RunPython(forwards_func, backwards_func),
+    ]

django/gsmap/migrations/0039_auto_20220223_1600.py

@@ -0,0 +1,21 @@
+# Generated by Django 3.1.13 on 2022-02-23 16:00
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('gsmap', '0038_migrate_translatable_fields'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='workspace',
+            name='_description',
+        ),
+        migrations.RemoveField(
+            model_name='workspace',
+            name='_title',
+        ),
+    ]

django/gsmap/models.py

@@ -31,7 +31,7 @@
 
 from parler.models import TranslatableModel, TranslatedFields
 
-
+from django.contrib.auth.models import Group
 from gsuser.models import User
 from main.utils import get_website
 
@@ -376,9 +376,6 @@ class Meta:
     created = models.DateTimeField(auto_now_add=True)
     modified = models.DateTimeField(auto_now=True)
 
-    title = models.CharField(max_length=150, default='')
-    description = models.TextField(default='')
-
     snapshots = SortedManyToManyField(Snapshot)
 
     categories = SortedManyToManyField(Category)
@@ -402,6 +399,11 @@ class Meta:
 
     findme_enabled = models.BooleanField(default=False, help_text="Enable 'Find Me'", verbose_name="Find me enabled")
 
+    translations = TranslatedFields(
+        title = models.CharField(max_length=150, default=''),
+        description = models.TextField(default=''),
+    )
+
     def get_absolute_link(self):
         website = get_website(Site.objects.get_current())
         return format_html(
@@ -430,7 +432,7 @@ def test_exists(pk):
         super().save(*args, **kwargs)
 
     def __str__(self):
-        return f'{self.id} {self.title}'
+        return f'{self.id}'
 
 class Annotation(models.Model):
     class Meta:
@@ -457,7 +459,7 @@ class Meta:
     )
     author_email = models.EmailField(max_length=254)
     rating = models.DecimalField(default=0, decimal_places=2, max_digits=6)
-    workspace =  models.ForeignKey(Workspace, on_delete=models.CASCADE)
+    workspace = models.ForeignKey(Workspace, on_delete=models.CASCADE)
 
     def save(self, *args, **kwargs):
         if self.data['properties']['description']:

django/gsmap/schema.py

@@ -182,6 +182,13 @@ class Meta:
         interfaces = [graphene.relay.Node]
 
     pk = graphene.String(source='id')
+    title = graphene.String(
+        language_code=graphene.Argument(Q_LANGUAGE, default_value=Q_LANGUAGE[settings.PARLER_DEFAULT_LANGUAGE_CODE]),
+    )
+
+    description = graphene.String(
+        language_code=graphene.Argument(Q_LANGUAGE, default_value=Q_LANGUAGE[settings.PARLER_DEFAULT_LANGUAGE_CODE]),
+    )
     snapshots = graphene.List(SnapshotNode)
 
     annotations = graphene.List(AnnotationNode)
@@ -192,6 +199,14 @@ class Meta:
     )
 
     usergroups = graphene.List(UsergroupNode)
+    
+    def resolve_title(self, info, language_code=None):
+        lang = Q_LANGUAGE.get(language_code).name
+        return self.safe_translation_getter("title", language_code=lang)
+
+    def resolve_description(self, info, language_code=None):
+        lang = Q_LANGUAGE.get(language_code).name
+        return self.safe_translation_getter("description", language_code=lang)
 
     def resolve_snapshots(self, info):
         return gql_optimizer.query(self.snapshots.all(), info)