fix: support another slug pattern

browniebroke · Jan 18, 2021 · a4ce0cb · a4ce0cb
1 parent 202a6b3
commit a4ce0cb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/django_codemod/visitors/urls.py b/django_codemod/visitors/urls.py
@@ -19,6 +19,7 @@ class PatternNotSupported(RuntimeError):
     ".+": "path",
     "[-a-zA-Z0-9_]+": "slug",
     r"[\w-]+": "slug",
+    r"[-\w]+": "slug",
     "[^/]+": "str",
     "[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}": "uuid",
 }

diff --git a/tests/visitors/test_urls.py b/tests/visitors/test_urls.py
@@ -129,6 +129,7 @@ def test_grouped_path(self) -> None:
                 url(r'^page/(?P<number>\d+)/$', views.page, name='page'),
                 url(r'^post/(?P<slug>[-a-zA-Z0-9_]+)/$', views.post, name='post'),
                 url(r'^post/(?P<slug>[\w-]+)/$', views.post, name='post'),
+                url(r'^post/(?P<slug>[-\w]+)/$', views.post, name='post'),
                 url(r'^about/(?P<name>[^/]+)/$', views.about, name='about'),
                 url(r'^uuid/(?P<uuid>{uuid_re})/$', by_uuid),
                 url(r'^(?P<path>.+)/$', views.default, name='default'),
@@ -142,6 +143,7 @@ def test_grouped_path(self) -> None:
                 path('page/<int:number>/', views.page, name='page'),
                 path('post/<slug:slug>/', views.post, name='post'),
                 path('post/<slug:slug>/', views.post, name='post'),
+                path('post/<slug:slug>/', views.post, name='post'),
                 path('about/<str:name>/', views.about, name='about'),
                 path('uuid/<uuid:uuid>/', by_uuid),
                 path('<path:path>/', views.default, name='default'),