From 1f01548f81733a6a1313f5b4d5faa8cf49606572 Mon Sep 17 00:00:00 2001
From: Sharad Heft <stephenswat@gmail.com>
Date: Sun, 5 Nov 2017 23:50:44 +0100
Subject: [PATCH] Add admin interface for OAuth2 related models.

---
 test_oauth/admin.py                   | 41 +++++++++++++++++++++++++++
 test_oauth/migrations/0001_initial.py |  4 +++
 test_oauth/models.py                  |  8 ++++++
 3 files changed, 53 insertions(+)
 create mode 100644 test_oauth/admin.py

diff --git a/test_oauth/admin.py b/test_oauth/admin.py
new file mode 100644
index 0000000..40a2d16
--- /dev/null
+++ b/test_oauth/admin.py
@@ -0,0 +1,41 @@
+from django.contrib import admin
+
+from test_oauth.models import Character, TESTOAuth2Data
+
+
+class CharacterAdmin(admin.ModelAdmin):
+    fieldsets = (
+        ('Character Information', {'fields': ('id', 'name')}),
+        ('Allegiance', {'fields': ('alliance_id', 'corporation_id')}),
+    )
+
+    list_display = ('id', 'name', 'alliance_id', 'corporation_id')
+    readonly_fields = ('id', 'name', 'alliance_id', 'corporation_id')
+    search_fields = ('name', 'alliance_id', 'corporation_id')
+    ordering = ('id',)
+
+    class Meta:
+        verbose_name = 'Character'
+        verbose_name_plural = 'Characters'
+
+
+class TESTOAuth2DataAdmin(admin.ModelAdmin):
+    fieldsets = (
+        ('User', {'fields': ('user',)}),
+        ('OAuth2', {'fields': ('access_token', 'refresh_token')}),
+        ('Data', {'fields': ('primary_character',)}),
+    )
+
+    list_display = ('user', 'primary_character')
+    readonly_fields = ('user',)
+    search_fields = ('user',)
+    ordering = ('user',)
+
+
+    class Meta:
+        verbose_name = 'OAuth2 Token'
+        verbose_name_plural = 'OAuth2 Tokens'
+
+
+admin.site.register(Character, CharacterAdmin)
+admin.site.register(TESTOAuth2Data, TESTOAuth2DataAdmin)
diff --git a/test_oauth/migrations/0001_initial.py b/test_oauth/migrations/0001_initial.py
index 246a7fe..9ff6b55 100644
--- a/test_oauth/migrations/0001_initial.py
+++ b/test_oauth/migrations/0001_initial.py
@@ -21,6 +21,8 @@ class Migration(migrations.Migration):
                 ('corporation_id', models.BigIntegerField(null=True)),
             ],
             options={
+                'verbose_name': 'Character',
+                'verbose_name_plural': 'Characters',
             },
             bases=(models.Model,),
         ),
@@ -33,6 +35,8 @@ class Migration(migrations.Migration):
                 ('primary_character', models.ForeignKey(default=None, to='test_oauth.Character', null=True)),
             ],
             options={
+                'verbose_name': 'OAuth2 Token',
+                'verbose_name_plural': 'OAuth2 Tokens',
             },
             bases=(models.Model,),
         ),
diff --git a/test_oauth/models.py b/test_oauth/models.py
index c36b5a5..e860460 100644
--- a/test_oauth/models.py
+++ b/test_oauth/models.py
@@ -15,6 +15,10 @@ class Character(models.Model):
     alliance_id = models.BigIntegerField(null=True)
     corporation_id = models.BigIntegerField(null=True)
 
+    class Meta:
+        verbose_name = 'Character'
+        verbose_name_plural = 'Characters'
+
 
 class TESTOAuth2Data(models.Model):
     user = models.OneToOneField(
@@ -27,3 +31,7 @@ class TESTOAuth2Data(models.Model):
     refresh_token = models.CharField(max_length=128)
 
     primary_character = models.ForeignKey(Character, null=True, default=None)
+
+    class Meta:
+        verbose_name = 'OAuth2 Token'
+        verbose_name_plural = 'OAuth2 Tokens'
-- 
GitLab