added income admin

1 files changed, 12 insertions, 3 deletions

diff --git a/app/income/admin.py b/app/income/admin.py
index c084eb0..6399463 100644
--- a/app/income/admin.py
+++ b/app/income/admin.py
@@ -6,9 +6,20 @@ from django.conf.urls import url
 from django.shortcuts import render
 
 
-from .models import Gig
+from .models import Gig, Invoice, InvoiceItem
 
 
+@admin.register(InvoiceItem)
+class InvoiceItemAdmin(admin.ModelAdmin):
+    list_display = ('time_start', 'time_end', 'work_done')
+
+
+@admin.register(Invoice)
+class InvoiceAdmin(admin.ModelAdmin):
+    list_display = ('title', 'date_start', 'date_end')
+
+
+@admin.register(Gig)
 class GigAdmin(OSMGeoAdmin):
     list_display = ('title', 'status', 'due_date', 'payment_status', 'payment', 'publisher', 'word_count')
     list_filter = ('publisher', 'status', 'payment_status')
@@ -75,5 +86,3 @@ class GigAdmin(OSMGeoAdmin):
         context['months'] = self.model.objects.dates('created', 'month')
         return render(request, 'admin/income_month.html', context)
 
-
-admin.site.register(Gig, GigAdmin)