Use proper path converter for lemonshort

This commit is contained in:
Danielle McLean 2022-03-12 15:04:05 +11:00
parent cfeb206154
commit 683adc1b46
5 changed files with 20 additions and 14 deletions

View file

@ -6,10 +6,11 @@ from string import ascii_lowercase, ascii_uppercase
chars = ascii_uppercase + ascii_lowercase chars = ascii_uppercase + ascii_lowercase
conv = BaseConverter(chars) conv = BaseConverter(chars)
class AbcIdConverter:
regex = '[a-zA-Z]+'
def abc_to_id(abc): def to_python(self, value: str) -> int:
return int(conv.decode(abc)) return int(conv.decode(value))
def to_url(self, value: int) -> str:
def id_to_abc(id): return conv.encode(value)
return conv.encode(id)

View file

@ -2,7 +2,7 @@ from django.apps import apps
from django.conf import settings from django.conf import settings
from typing import Any, Dict, Type from typing import Any, Dict, Type
from .convert import id_to_abc from .convert import AbcIdConverter
prefixes = {} # type: Dict[Type[Any], str] prefixes = {} # type: Dict[Type[Any], str]
@ -14,4 +14,4 @@ def short_url(entity):
base = '/' base = '/'
if hasattr(settings, 'SHORT_BASE_URL'): if hasattr(settings, 'SHORT_BASE_URL'):
base = settings.SHORT_BASE_URL base = settings.SHORT_BASE_URL
return base + prefixes[type(entity)] + id_to_abc(entity.id) return base + prefixes[type(entity)] + AbcIdConverter().to_url(entity.id)

View file

@ -1,7 +1,7 @@
from .. import convert from .. import convert
def test_abc_to_id(): def test_to_python():
samples = { samples = {
'A': 0, 'A': 0,
'B': 1, 'B': 1,
@ -12,8 +12,9 @@ def test_abc_to_id():
'BA': 52, 'BA': 52,
'BAB': 2705, 'BAB': 2705,
} }
converter = convert.AbcIdConverter()
for abc, id in samples.items(): for abc, id in samples.items():
assert convert.abc_to_id(abc) == id assert converter.to_python(abc) == id
def test_id_to_abc(): def test_id_to_abc():
@ -26,5 +27,6 @@ def test_id_to_abc():
104: 'CA', 104: 'CA',
130: 'Ca', 130: 'Ca',
} }
converter = convert.AbcIdConverter()
for id, abc in samples.items(): for id, abc in samples.items():
assert convert.id_to_abc(id) == abc assert converter.to_url(id) == abc

View file

@ -1,10 +1,13 @@
from django.conf import settings from django.conf import settings
from django.urls import path from django.urls import path, register_converter
from .convert import AbcIdConverter
from .views import unshort from .views import unshort
register_converter(AbcIdConverter, 'abc_id')
app_name = 'lemonshort' app_name = 'lemonshort'
urlpatterns = tuple( urlpatterns = tuple(
path('{0!s}<tiny>'.format(k), unshort, name=m, kwargs={'model': m}) path('{0!s}<abc_id:tiny>'.format(k), unshort, name=m, kwargs={'model': m})
for k, m in settings.SHORTEN_MODELS.items() for k, m in settings.SHORTEN_MODELS.items()
) )

View file

@ -1,9 +1,9 @@
from django.apps import apps from django.apps import apps
from django.shortcuts import get_object_or_404, redirect from django.shortcuts import get_object_or_404, redirect
from .convert import abc_to_id from .convert import AbcIdConverter
def unshort(request, model, tiny): def unshort(request, model, tiny):
entity = get_object_or_404(apps.get_model(model), pk=abc_to_id(tiny)) entity = get_object_or_404(apps.get_model(model), pk=tiny)
return redirect(entity, permanent=True) return redirect(entity, permanent=True)