forked from CGM_Public/pretix_original
138 lines
5.2 KiB
Python
138 lines
5.2 KiB
Python
#
|
|
# This file is part of pretix (Community Edition).
|
|
#
|
|
# Copyright (C) 2014-2020 Raphael Michel and contributors
|
|
# Copyright (C) 2020-today pretix GmbH and contributors
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General
|
|
# Public License as published by the Free Software Foundation in version 3 of the License.
|
|
#
|
|
# ADDITIONAL TERMS APPLY: Pursuant to Section 7 of the GNU Affero General Public License, additional terms are
|
|
# applicable granting you additional permissions and placing additional restrictions on your usage of this software.
|
|
# Please refer to the pretix LICENSE file to obtain the full terms applicable to this work. If you did not receive
|
|
# this file, see <https://pretix.eu/about/en/license>.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
|
|
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License along with this program. If not, see
|
|
# <https://www.gnu.org/licenses/>.
|
|
#
|
|
import json
|
|
|
|
from django.db.models import prefetch_related_objects
|
|
from rest_framework import serializers
|
|
from rest_framework.exceptions import ValidationError
|
|
|
|
|
|
class AsymmetricField(serializers.Field):
|
|
def __init__(self, read, write, **kwargs):
|
|
self.read = read
|
|
self.write = write
|
|
super().__init__(
|
|
required=self.write.required,
|
|
default=self.write.default,
|
|
initial=self.write.initial,
|
|
source=self.write.source if self.write.source != self.write.field_name else None,
|
|
label=self.write.label,
|
|
allow_null=self.write.allow_null,
|
|
error_messages=self.write.error_messages,
|
|
validators=self.write.validators,
|
|
**kwargs
|
|
)
|
|
|
|
def to_internal_value(self, data):
|
|
return self.write.to_internal_value(data)
|
|
|
|
def to_representation(self, value):
|
|
return self.read.to_representation(value)
|
|
|
|
def run_validation(self, data=serializers.empty):
|
|
return self.write.run_validation(data)
|
|
|
|
|
|
class CompatibleJSONField(serializers.JSONField):
|
|
def to_internal_value(self, data):
|
|
try:
|
|
return json.dumps(data)
|
|
except (TypeError, ValueError):
|
|
self.fail('invalid')
|
|
|
|
def to_representation(self, value):
|
|
if value:
|
|
return json.loads(value)
|
|
return value
|
|
|
|
|
|
class SalesChannelMigrationMixin:
|
|
"""
|
|
Translates between the old field "sales_channels" and the new field combo "all_sales_channels"/"limit_sales_channels".
|
|
"""
|
|
|
|
@property
|
|
def organizer(self):
|
|
if "organizer" in self.context:
|
|
return self.context["organizer"]
|
|
elif "event" in self.context:
|
|
return self.context["event"].organizer
|
|
else:
|
|
raise ValueError("organizer not in context")
|
|
|
|
def to_internal_value(self, data):
|
|
if "sales_channels" in data:
|
|
if data["sales_channels"] is None:
|
|
raise ValidationError({
|
|
"sales_channels": [
|
|
"The legacy attribute 'sales_channels' cannot be set to None, it must be a list."
|
|
]
|
|
})
|
|
|
|
prefetch_related_objects([self.organizer], "sales_channels")
|
|
all_channels = {
|
|
s.identifier for s in
|
|
self.organizer.sales_channels.all()
|
|
}
|
|
|
|
if data.get("all_sales_channels") and set(data["sales_channels"]) != all_channels:
|
|
raise ValidationError({
|
|
"all_sales_channels": [
|
|
"If 'all_sales_channels' is set, the legacy attribute 'sales_channels' must not be set or set to "
|
|
"the list of all sales channels."
|
|
]
|
|
})
|
|
|
|
if data.get("limit_sales_channels") and set(data["sales_channels"]) != set(data["limit_sales_channels"]):
|
|
raise ValidationError({
|
|
"limit_sales_channels": [
|
|
"If 'limit_sales_channels' is set, the legacy attribute 'sales_channels' must not be set or set to "
|
|
"the same list."
|
|
]
|
|
})
|
|
|
|
if set(data["sales_channels"]) == all_channels:
|
|
data["all_sales_channels"] = True
|
|
data["limit_sales_channels"] = []
|
|
else:
|
|
data["all_sales_channels"] = False
|
|
data["limit_sales_channels"] = data["sales_channels"]
|
|
|
|
del data["sales_channels"]
|
|
|
|
if data.get("all_sales_channels"):
|
|
data["limit_sales_channels"] = []
|
|
|
|
return super().to_internal_value(data)
|
|
|
|
def to_representation(self, value):
|
|
value = super().to_representation(value)
|
|
if value.get("all_sales_channels"):
|
|
prefetch_related_objects([self.organizer], "sales_channels")
|
|
value["sales_channels"] = sorted([
|
|
s.identifier for s in
|
|
self.organizer.sales_channels.all()
|
|
])
|
|
else:
|
|
value["sales_channels"] = value["limit_sales_channels"]
|
|
return value
|