diff --git a/task_center/urls.py b/task_center/urls.py index a9ed947..b2bd45a 100644 --- a/task_center/urls.py +++ b/task_center/urls.py @@ -14,15 +14,15 @@ Including another URLconf 1. Import the include() function: from django.urls import include, path 2. Add a URL to urlpatterns: path('blog/', include('blog.urls')) """ -from django.contrib import admin -from django.urls import path, include -from django.conf import settings -from django.conf.urls.static import static -from tasks.views_frontend import index - -urlpatterns = [ - path('houtai/', admin.site.urls), - path('api/', include('tasks.urls')), - # Root URL points to home page - path('', index, name='index'), +from django.contrib import admin +from django.urls import path, include +from django.conf import settings +from django.conf.urls.static import static +from tasks.views_frontend import index + +urlpatterns = [ + path('houtai/', admin.site.urls), + path('api/', include('tasks.urls')), + # Root URL points to home page + path('', index, name='index'), ] + static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT) diff --git a/tasks/admin.py b/tasks/admin.py index a264d99..9718eb2 100644 --- a/tasks/admin.py +++ b/tasks/admin.py @@ -53,4 +53,4 @@ class TaskResultAdmin(admin.ModelAdmin): 'classes': ('collapse',) }), ) - + diff --git a/tasks/models.py b/tasks/models.py index 71eeb1e..01eb9d0 100644 --- a/tasks/models.py +++ b/tasks/models.py @@ -1,5 +1,6 @@ from django.db import models from django.utils import timezone +import secrets # Status choices for tasks STATUS_CHOICES = [ @@ -20,6 +21,13 @@ class Client(models.Model): def __str__(self): return self.name + + def save(self, *args, **kwargs): + # Generate a unique token if this is a new client + if not self.pk and not self.token: + # Use secrets module to generate a secure random token + self.token = secrets.token_urlsafe(64) # 64 bytes -> ~86 characters + super().save(*args, **kwargs) class Meta: verbose_name = '客户端'