feat: initial work on short link management

Author: arcstur

infographics/settings.py

@@ -55,6 +55,7 @@
     'web',
     'graphs',
     'video',
+    'shortlink',
 ]
 
 MIDDLEWARE = [

shortlink/__init__.py

@@ -0,0 +1,7 @@
+from django.contrib import admin
+
+from shortlink.models import ShortLink
+
+@admin.register(ShortLink)
+class ShortLinkAdmin(admin.ModelAdmin):
+    list_display = ["id", "query"]

shortlink/admin.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ShortlinkConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'shortlink'

shortlink/apps.py

@@ -0,0 +1,21 @@
+# Generated by Django 4.2.24 on 2025-10-22 14:30
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ShortLink',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('query', models.TextField()),
+            ],
+        ),
+    ]

shortlink/migrations/0001_initial.py

@@ -0,0 +1,27 @@
+from django.db import models
+from sqids import Sqids
+
+
+class ShortLinkManager(models.Manager):
+    def sqids(self):
+        return Sqids()
+
+    def encode(self, id):
+        return self.sqids().encode([id])
+
+    def decode(self, encoded):
+        return self.sqids().decode(encoded)[0]
+
+    def encoded_id_from_query(self, query):
+        short_link, created = self.get_or_create(query=query)
+        return self.encode(short_link.id)
+
+    def query_from_encoded_id(self, encoded):
+        id = self.decode(encoded)
+        return self.get(id=id).query
+
+
+class ShortLink(models.Model):
+    query = models.TextField()
+
+    objects = ShortLinkManager()

shortlink/migrations/__init__.py

@@ -0,0 +1,34 @@
+from django.test import TestCase
+
+from .models import ShortLink
+
+
+class ShortLinkTests(TestCase):
+    QUERY = """
+#title: population of Brazilian non-capital municipalities
+SELECT ?item ?itemLabel ?stateLabel ?population ?date WHERE {
+  ?item wdt:P31 wd:Q3184121.
+  ?item p:P1082 ?population_node.
+  ?population_node ps:P1082 ?population.
+  ?population_node pq:P585 ?date.
+  ?population_node pq:P459 wd:Q39825. # only census
+  MINUS { ?item wdt:P1376 []. } # remove capitals
+  ?item wdt:P131 ?state. # group by state
+  FILTER (?date >= "1970-01-01"^^xsd:dateTime) # after 1970
+  SERVICE wikibase:label { bd:serviceParam wikibase:language "pt-br". }
+}
+"""
+
+    def test_encode_and_decode(self):
+        encoded = ShortLink.objects.encoded_id_from_query(self.QUERY)
+        query = ShortLink.objects.query_from_encoded_id(encoded)
+        self.assertEqual(query, self.QUERY)
+
+    def test_existant_query(self):
+        sl = ShortLink.objects.create(query=self.QUERY)
+        encoded = ShortLink.objects.encoded_id_from_query(self.QUERY)
+        encoded2 = ShortLink.objects.encode(sl.id)
+        self.assertEqual(
+            ShortLink.objects.decode(encoded),
+            ShortLink.objects.decode(encoded2),
+        )

shortlink/models.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.