mirror of
https://github.com/ae-utbm/sith.git
synced 2025-11-10 05:53:06 +00:00
add hmac_key to ApiClient
This commit is contained in:
@@ -17,6 +17,15 @@ class ApiClientAdmin(admin.ModelAdmin):
|
|||||||
"owner__nick_name",
|
"owner__nick_name",
|
||||||
)
|
)
|
||||||
autocomplete_fields = ("owner", "groups", "client_permissions")
|
autocomplete_fields = ("owner", "groups", "client_permissions")
|
||||||
|
readonly_fields = ("hmac_key",)
|
||||||
|
actions = ("reset_hmac_key",)
|
||||||
|
|
||||||
|
@admin.action(permissions=["change"], description=_("Reset HMAC key"))
|
||||||
|
def reset_hmac_key(self, _request: HttpRequest, queryset: QuerySet[ApiClient]):
|
||||||
|
objs = list(queryset)
|
||||||
|
for obj in objs:
|
||||||
|
obj.reset_hmac(commit=False)
|
||||||
|
ApiClient.objects.bulk_update(objs, fields=["hmac_key"])
|
||||||
|
|
||||||
|
|
||||||
@admin.register(ApiKey)
|
@admin.register(ApiKey)
|
||||||
|
|||||||
19
api/migrations/0002_apiclient_hmac_key.py
Normal file
19
api/migrations/0002_apiclient_hmac_key.py
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
# Generated by Django 5.2.3 on 2025-10-26 10:15
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
import api.models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
dependencies = [("api", "0001_initial")]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name="apiclient",
|
||||||
|
name="hmac_key",
|
||||||
|
field=models.CharField(
|
||||||
|
default=api.models.get_hmac_key, max_length=128, verbose_name="HMAC Key"
|
||||||
|
),
|
||||||
|
),
|
||||||
|
]
|
||||||
@@ -1,3 +1,4 @@
|
|||||||
|
import secrets
|
||||||
from typing import Iterable
|
from typing import Iterable
|
||||||
|
|
||||||
from django.contrib.auth.models import Permission
|
from django.contrib.auth.models import Permission
|
||||||
@@ -10,6 +11,10 @@ from django.utils.translation import pgettext_lazy
|
|||||||
from core.models import Group, User
|
from core.models import Group, User
|
||||||
|
|
||||||
|
|
||||||
|
def get_hmac_key():
|
||||||
|
return secrets.token_hex(64)
|
||||||
|
|
||||||
|
|
||||||
class ApiClient(models.Model):
|
class ApiClient(models.Model):
|
||||||
name = models.CharField(_("name"), max_length=64)
|
name = models.CharField(_("name"), max_length=64)
|
||||||
owner = models.ForeignKey(
|
owner = models.ForeignKey(
|
||||||
@@ -28,6 +33,7 @@ class ApiClient(models.Model):
|
|||||||
help_text=_("Specific permissions for this api client."),
|
help_text=_("Specific permissions for this api client."),
|
||||||
related_name="clients",
|
related_name="clients",
|
||||||
)
|
)
|
||||||
|
hmac_key = models.CharField(_("HMAC Key"), max_length=128, default=get_hmac_key)
|
||||||
created_at = models.DateTimeField(auto_now_add=True)
|
created_at = models.DateTimeField(auto_now_add=True)
|
||||||
updated_at = models.DateTimeField(auto_now=True)
|
updated_at = models.DateTimeField(auto_now=True)
|
||||||
|
|
||||||
@@ -59,6 +65,13 @@ class ApiClient(models.Model):
|
|||||||
raise ValueError("perm_list must be an iterable of permissions.")
|
raise ValueError("perm_list must be an iterable of permissions.")
|
||||||
return all(self.has_perm(perm) for perm in perm_list)
|
return all(self.has_perm(perm) for perm in perm_list)
|
||||||
|
|
||||||
|
def reset_hmac(self, *, commit: bool = True) -> str:
|
||||||
|
"""Reset and return the HMAC key for this client."""
|
||||||
|
self.hmac_key = get_hmac_key()
|
||||||
|
if commit:
|
||||||
|
self.save()
|
||||||
|
return self.hmac_key
|
||||||
|
|
||||||
|
|
||||||
class ApiKey(models.Model):
|
class ApiKey(models.Model):
|
||||||
PREFIX_LENGTH = 5
|
PREFIX_LENGTH = 5
|
||||||
|
|||||||
Reference in New Issue
Block a user