23 lines
616 B
Python
23 lines
616 B
Python
from rest_framework import viewsets
|
|
|
|
from orchestra.api import router, SetPasswordApiMixin
|
|
from orchestra.contrib.accounts.api import AccountApiMixin
|
|
|
|
from .models import Address, Mailbox
|
|
from .serializers import AddressSerializer, MailboxSerializer
|
|
|
|
|
|
class AddressViewSet(AccountApiMixin, viewsets.ModelViewSet):
|
|
model = Address
|
|
serializer_class = AddressSerializer
|
|
|
|
|
|
|
|
class MailboxViewSet(SetPasswordApiMixin, AccountApiMixin, viewsets.ModelViewSet):
|
|
model = Mailbox
|
|
serializer_class = MailboxSerializer
|
|
|
|
|
|
router.register(r'mailboxes', MailboxViewSet)
|
|
router.register(r'addresses', AddressViewSet)
|