use six instead of relying on python version

1 files changed, 2 insertions, 6 deletions

diff --git a/mastodon/Mastodon.py b/mastodon/Mastodon.py
index 88f4906..9e6d3ce 100644
--- a/mastodon/Mastodon.py
+++ b/mastodon/Mastodon.py
@@ -17,6 +17,7 @@ import re
 import copy
 import threading
 import sys
+import six
 
 try:
     from urllib.parse import urlparse
@@ -1083,13 +1084,8 @@ class Mastodon:
         """
         Converts json string IDs to native python bignums.
         """
-        if sys.version_info.major >= 3:
-            str_type = str
-        else:
-            str_type = unicode
-
         if ('id' in json_object and
-                isinstance(json_object['id'], str_type)):
+                isinstance(json_object['id'], six.text_type)):
             try:
                 json_object['id'] = int(json_object['id'])
             except ValueError: