From 52f30e498be00f4b563d7e35fa070b3a7409e0c2 Mon Sep 17 00:00:00 2001
From: Daniel Lindsley <dlindsley@mediaphormedia.com>
Date: Sat, 18 Oct 2008 21:40:22 -0500
Subject: [PATCH] Customized admins.

---
 budget/admin.py              | 38 ++++++++++++++++++++++++++++++++++--
 budget/categories/admin.py   | 21 +++++++++++++++++++-
 budget/transactions/admin.py | 19 +++++++++++++++++-
 3 files changed, 74 insertions(+), 4 deletions(-)

diff --git a/budget/admin.py b/budget/admin.py
index 49b3d03..0b11f5d 100644
--- a/budget/admin.py
+++ b/budget/admin.py
@@ -1,5 +1,39 @@
 from django.contrib import admin
 from budget.models import Budget, BudgetEstimate
 
-admin.site.register(Budget)
-admin.site.register(BudgetEstimate)
+
+class BudgetAdmin(admin.ModelAdmin):
+    date_hierarchy = 'start_date'
+    fieldsets = (
+        (None, {
+            'fields': ('name', 'slug', 'start_date'),
+        }),
+        ('Metadata', {
+            'classes': ('collapse',),
+            'fields': ('created', 'updated', 'is_deleted')
+        })
+    )
+    list_display = ('name', 'start_date', 'is_deleted')
+    list_filter = ('is_deleted',)
+    prepopulated_fields = {
+        'slug': ('name',),
+    }
+    search_fields = ('name',)
+
+
+class BudgetEstimateAdmin(admin.ModelAdmin):
+    fieldsets = (
+        (None, {
+            'fields': ('budget', 'category', 'amount'),
+        }),
+        ('Metadata', {
+            'classes': ('collapse',),
+            'fields': ('created', 'updated', 'is_deleted')
+        })
+    )
+    list_display = ('category', 'budget', 'amount', 'is_deleted')
+    list_filter = ('is_deleted',)
+
+
+admin.site.register(Budget, BudgetAdmin)
+admin.site.register(BudgetEstimate, BudgetEstimateAdmin)
diff --git a/budget/categories/admin.py b/budget/categories/admin.py
index 564ace1..1539fc5 100644
--- a/budget/categories/admin.py
+++ b/budget/categories/admin.py
@@ -1,4 +1,23 @@
 from django.contrib import admin
 from budget.categories.models import Category
 
-admin.site.register(Category)
+
+class CategoryAdmin(admin.ModelAdmin):
+    fieldsets = (
+        (None, {
+            'fields': ('name', 'slug'),
+        }),
+        ('Metadata', {
+            'classes': ('collapse',),
+            'fields': ('created', 'updated', 'is_deleted')
+        })
+    )
+    list_display = ('name', 'is_deleted')
+    list_filter = ('is_deleted',)
+    prepopulated_fields = {
+        'slug': ('name',),
+    }
+    search_fields = ('name',)
+
+
+admin.site.register(Category, CategoryAdmin)
diff --git a/budget/transactions/admin.py b/budget/transactions/admin.py
index 85b895b..f244e17 100644
--- a/budget/transactions/admin.py
+++ b/budget/transactions/admin.py
@@ -1,4 +1,21 @@
 from django.contrib import admin
 from budget.transactions.models import Transaction
 
-admin.site.register(Transaction)
+
+class TransactionAdmin(admin.ModelAdmin):
+    date_hierarchy = 'date'
+    fieldsets = (
+        (None, {
+            'fields': ('transaction_type', 'notes', 'category', 'amount', 'date'),
+        }),
+        ('Metadata', {
+            'classes': ('collapse',),
+            'fields': ('created', 'updated', 'is_deleted')
+        })
+    )
+    list_display = ('notes', 'transaction_type', 'amount', 'date', 'is_deleted')
+    list_filter = ('is_deleted',)
+    search_fields = ('notes',)
+
+
+admin.site.register(Transaction, TransactionAdmin)