54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from django.conf import settings
|
|
from rest_framework import serializers
|
|
|
|
from ..editor.hosters import get_hoster_for_package
|
|
from ..mapdata.models import Level, Package, Source
|
|
from .permissions import can_access_package
|
|
|
|
|
|
class BoundsMixin:
|
|
def to_representation(self, obj):
|
|
result = super().to_representation(obj)
|
|
if obj.bottom is not None:
|
|
result['bounds'] = ((obj.bottom, obj.left), (obj.top, obj.right))
|
|
return result
|
|
|
|
|
|
class LevelSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Level
|
|
fields = ('name', 'altitude', 'package')
|
|
|
|
|
|
class PackageSerializer(BoundsMixin, serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Package
|
|
fields = ('name', 'home_repo', 'depends')
|
|
|
|
def to_representation(self, obj):
|
|
result = super().to_representation(obj)
|
|
result['public'] = obj.name in settings.PUBLIC_PACKAGES
|
|
hoster = get_hoster_for_package(obj)
|
|
if 'request' in self.context:
|
|
result['access_granted'] = can_access_package(self.context['request'], obj)
|
|
if hoster is not None:
|
|
result['hoster'] = hoster.name
|
|
return result
|
|
|
|
|
|
class SourceSerializer(BoundsMixin, serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Source
|
|
fields = ('name', 'package')
|
|
|
|
|
|
class HosterSerializer(serializers.Serializer):
|
|
name = serializers.CharField()
|
|
base_url = serializers.CharField()
|
|
|
|
def to_representation(self, obj):
|
|
result = super().to_representation(obj)
|
|
result['packages'] = obj.get_packages().values_list('name', flat=True)
|
|
if 'request' in self.context:
|
|
result['signed_in'] = obj.is_access_granted(self.context['request'])
|
|
return result
|