22 lines
617 B
Python
22 lines
617 B
Python
|
from rest_framework import viewsets
|
||
|
|
|
||
|
|
from orchestra.api import router
|
||
|
|
from orchestra.apps.accounts.api import AccountApiMixin
|
||
|
|
|
||
|
|
from .models import Contact, InvoiceContact
|
||
|
|
from .serializers import ContactSerializer, InvoiceContactSerializer
|
||
|
|
|
||
|
|
|
||
|
|
class ContactViewSet(AccountApiMixin, viewsets.ModelViewSet):
|
||
|
|
model = Contact
|
||
|
|
serializer_class = ContactSerializer
|
||
|
|
|
||
|
|
|
||
|
|
class InvoiceContactViewSet(AccountApiMixin, viewsets.ModelViewSet):
|
||
|
|
model = InvoiceContact
|
||
|
|
serializer_class = InvoiceContactSerializer
|
||
|
|
|
||
|
|
|
||
|
|
router.register(r'contacts', ContactViewSet)
|
||
|
|
router.register(r'invoicecontacts', InvoiceContactViewSet)
|