models: add dedicated CUTPartner model (#73170)

This commit is contained in:
Paul Marillonnet 2023-01-10 09:50:41 +01:00
parent 25e7948951
commit 3b405b5eae
2 changed files with 83 additions and 0 deletions

View File

@ -0,0 +1,53 @@
# Generated by Django 2.2.26 on 2023-01-10 10:42
import django.contrib.postgres.fields
import django.db.models.deletion
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('authentic2_idp_oidc', '0017_oidcaccesstoken_profile'),
('authentic2_cut', '0006_delete_journal'),
]
operations = [
migrations.CreateModel(
name='CUTPartner',
fields=[
(
'id',
models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID'),
),
(
'domains',
django.contrib.postgres.fields.ArrayField(
base_field=models.CharField(blank=True, max_length=128, null=True),
size=None,
verbose_name='Domains',
null=True,
),
),
('url', models.URLField(verbose_name='Partner URL', null=True)),
(
'stat_emails',
django.contrib.postgres.fields.ArrayField(
base_field=models.EmailField(blank=True, max_length=254, null=True),
size=None,
verbose_name='Statistics email recipients',
null=True,
),
),
(
'oidc_client',
models.OneToOneField(
related_name='cut_partner',
on_delete=django.db.models.deletion.CASCADE,
to='authentic2_idp_oidc.OIDCClient',
verbose_name='OIDC Client',
),
),
],
),
]

View File

@ -1,9 +1,11 @@
import os.path
from datetime import timedelta
from authentic2_idp_oidc.models import OIDCClient
from django.conf import settings
from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.contrib.postgres.fields import ArrayField
from django.core.files.storage import default_storage
from django.db import models
from django.db.models.query import Q
@ -11,6 +13,7 @@ from django.db.models.signals import post_delete
from django.dispatch import receiver
from django.urls import reverse
from django.utils.timezone import now
from django.utils.translation import gettext_lazy as _
from sorl.thumbnail import delete, get_thumbnail
@ -226,3 +229,30 @@ def auto_delete_image_files(sender, instance, **kwargs):
pass
if os.path.isfile(instance.image.path):
instance.image.delete(save=False)
class CUTPartner(models.Model):
oidc_client = models.OneToOneField(
to='authentic2_idp_oidc.OIDCClient',
verbose_name=_('OIDC Client'),
on_delete=models.CASCADE,
related_name='cut_partner',
)
domains = ArrayField(
base_field=models.CharField(blank=True, null=True, max_length=128),
verbose_name=_('Domains'),
null=True,
)
url = models.URLField(verbose_name=_('Partner URL'), null=True)
stat_emails = ArrayField(
base_field=models.EmailField(blank=True, null=True),
verbose_name=_('Statistics email recipients'),
null=True,
)
def as_dict():
return {
'domains': self.domains,
'url': self.urls,
'stat_emails': self.stat_emails,
}