From 01f5435f6388a8161553e8abad52f47499a4be2c Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@digitalocean.com>
Date: Tue, 21 Mar 2017 12:59:44 -0400
Subject: [PATCH] Tweak how we set the API version

---
 netbox/netbox/settings.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/netbox/netbox/settings.py b/netbox/netbox/settings.py
index c79175f45..a3c2526e2 100644
--- a/netbox/netbox/settings.py
+++ b/netbox/netbox/settings.py
@@ -184,7 +184,7 @@ LOGIN_URL = '/{}login/'.format(BASE_PATH)
 SECRETS_MIN_PUBKEY_SIZE = 2048
 
 # Django REST framework (API)
-REST_FRAMEWORK_VERSION = VERSION.rsplit('.', 1)[0],  # Use major.minor as API version
+REST_FRAMEWORK_VERSION = VERSION[0:3]  # Use major.minor as API version
 REST_FRAMEWORK = {
     'DEFAULT_AUTHENTICATION_CLASSES': (
         'rest_framework.authentication.SessionAuthentication',
@@ -198,7 +198,7 @@ REST_FRAMEWORK = {
         'utilities.api.TokenPermissions',
     ),
     'DEFAULT_VERSION': REST_FRAMEWORK_VERSION,
-    'ALLOWED_VERSIONS': REST_FRAMEWORK_VERSION,
+    'ALLOWED_VERSIONS': [REST_FRAMEWORK_VERSION],
     'DEFAULT_VERSIONING_CLASS': 'rest_framework.versioning.AcceptHeaderVersioning',
     'PAGE_SIZE': PAGINATE_COUNT,
 }