From 371525e8821849635fc836aeabc209e380662a15 Mon Sep 17 00:00:00 2001
From: Sharad Heft <stephenswat@gmail.com>
Date: Wed, 13 Dec 2017 18:17:12 +0100
Subject: [PATCH] Update the reusable app for Django 2.0.

---
 setup.py                               |  2 +-
 test_oauth/migrations/0002_django20.py | 21 +++++++++++++++++++++
 test_oauth/models.py                   |  8 +++++++-
 3 files changed, 29 insertions(+), 2 deletions(-)
 create mode 100644 test_oauth/migrations/0002_django20.py

diff --git a/setup.py b/setup.py
index bae2afe..6aa6e6a 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import setup
 
 setup(
     name='test_oauth2_django',
-    version='0.1',
+    version='0.2',
     description='A basic library for Django apps to use TEST Alliance OAuth2.',
     url='https://git.pleaseignore.com/sharad/test_oauth2_django',
     author='Sharad Heft',
diff --git a/test_oauth/migrations/0002_django20.py b/test_oauth/migrations/0002_django20.py
new file mode 100644
index 0000000..80f5f3a
--- /dev/null
+++ b/test_oauth/migrations/0002_django20.py
@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import models, migrations
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('test_oauth', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='testoauth2data',
+            name='primary_character',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.SET_NULL, default=None, to='test_oauth.Character', null=True),
+            preserve_default=True,
+        ),
+    ]
diff --git a/test_oauth/models.py b/test_oauth/models.py
index dce6e49..99bc4bb 100644
--- a/test_oauth/models.py
+++ b/test_oauth/models.py
@@ -26,6 +26,7 @@ class Character(models.Model):
 class TESTOAuth2Data(models.Model):
     user = models.OneToOneField(
         settings.AUTH_USER_MODEL,
+        on_delete=models.CASCADE,
         primary_key=True,
         related_name='auth'
     )
@@ -33,7 +34,12 @@ class TESTOAuth2Data(models.Model):
     access_token = models.CharField(max_length=128)
     refresh_token = models.CharField(max_length=128)
 
-    primary_character = models.ForeignKey(Character, null=True, default=None)
+    primary_character = models.ForeignKey(
+        Character,
+        on_delete=models.SET_NULL,
+        null=True, 
+        default=None
+    )
 
     class Meta:
         verbose_name = 'OAuth2 Token'
-- 
GitLab