diff --git a/django_migration_linter/migration_linter.py b/django_migration_linter/migration_linter.py index 660e8df..73156dc 100644 --- a/django_migration_linter/migration_linter.py +++ b/django_migration_linter/migration_linter.py @@ -5,6 +5,7 @@ import inspect import logging import os import re +import sys from enum import Enum, unique from subprocess import PIPE, Popen @@ -12,7 +13,6 @@ from django.conf import settings from django.core.management import call_command from django.db import DEFAULT_DB_ALIAS, ProgrammingError, connections from django.db.migrations import RunPython, RunSQL -from six import PY2 from .cache import Cache from .constants import ( @@ -26,6 +26,8 @@ from .utils import clean_bytes_to_str, get_migration_abspath, split_migration_pa logger = logging.getLogger("django_migration_linter") +PY2 = sys.version_info[0] == 2 + @unique class MessageType(Enum): diff --git a/setup.py b/setup.py index c81f874..7044479 100644 --- a/setup.py +++ b/setup.py @@ -35,7 +35,6 @@ setup( "django>=1.11", "appdirs>=1.4.3", 'enum34>=1.1.6;python_version<"3.4"', - "six>=1.14.0", "toml>=0.10.2", ], extras_require={