# Patchwork - automated patch tracking system

# Copyright (C) 2016 Linaro Corporation

#

# This file is part of the Patchwork package.

#

# Patchwork is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# Patchwork 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 General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with Patchwork; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

from django.shortcuts import get_object_or_404

from rest_framework.generics import ListAPIView

from rest_framework.generics import RetrieveUpdateAPIView

from rest_framework.serializers import CharField

from rest_framework.serializers import HyperlinkedModelSerializer

from patchwork.api.base import PatchworkPermission

from patchwork.api.embedded import UserProfileSerializer

from patchwork.models import Project

class ProjectSerializer(HyperlinkedModelSerializer):

link_name = CharField(max_length=255, source='linkname')

list_id = CharField(max_length=255, source='listid')

list_email = CharField(max_length=200, source='listemail')

maintainers = UserProfileSerializer(many=True, read_only=True,

source='maintainer_project')

class Meta:

model = Project

fields = ('id', 'url', 'name', 'link_name', 'list_id', 'list_email',

'web_url', 'scm_url', 'webscm_url', 'maintainers')

read_only_fields = ('name', 'maintainers')

extra_kwargs = {

'url': {'view_name': 'api-project-detail'},

}

class ProjectMixin(object):

permission_classes = (PatchworkPermission,)

serializer_class = ProjectSerializer

def get_object(self):

queryset = self.filter_queryset(self.get_queryset())

assert 'pk' in self.kwargs

try:

obj = queryset.get(id=int(self.kwargs['pk']))

except (ValueError, Project.DoesNotExist):

obj = get_object_or_404(queryset, linkname=self.kwargs['pk'])

# NOTE(stephenfin): We must do this to make sure the 'url'

# field is populated correctly

self.kwargs['pk'] = obj.id

self.check_object_permissions(self.request, obj)

return obj

def get_queryset(self):

return Project.objects.all().prefetch_related('maintainer_project')

class ProjectList(ProjectMixin, ListAPIView):

"""List projects."""

search_fields = ('link_name', 'list_id', 'list_email', 'web_url',

'scm_url', 'webscm_url')

ordering_fields = ('id', 'name', 'link_name', 'list_id')

class ProjectDetail(ProjectMixin, RetrieveUpdateAPIView):

"""Show a project."""

pass