Bump versions and move to GitHub actions (#1)

* Remove testing with Django<2.2 and Python<3.7

- Newer versions of mypy also (incorrectly) don't accept overriding enum
  attributes when subclassing. Mitigated by temporarily removing
  attributes from Enum base class

* Move to GitHub actions

Author: Petter Friberg

.github/workflows/tests.yml

@@ -0,0 +1,43 @@
+name: Test
+
+on:
+  push:
+    branches:
+      - master
+  pull_request:
+
+jobs:
+  tests:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        include:
+          - python: "3.7"
+            tox_env: py37-django22
+          - python: "3.7"
+            tox_env: py37-django30
+          - python: "3.7"
+            tox_env: py37-django31
+          - python: "3.8"
+            tox_env: py38-django22
+          - python: "3.8"
+            tox_env: py38-django30
+          - python: "3.8"
+            tox_env: py38-django31
+          - python: "3.9"
+            tox_env: py39-django22
+          - python: "3.9"
+            tox_env: py39-django30
+          - python: "3.9"
+            tox_env: py39-django31
+          - python: "3.7"
+            tox_env: checks
+    name: ${{ matrix.tox_env }}
+    steps:
+      - uses: actions/checkout@v2
+      - uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python }}
+      - run: pip install tox
+      - name: Run ${{ matrix.tox_env }} job
+        run: tox -e ${{ matrix.tox_env }}

.travis.yml

@@ -1,6 +1,12 @@
 # Changelog
 
 
+## [unreleased]
+
+- Move CI to GitHub Actions
+- Dropped support for Python < 3.7
+- Dropped support for Django < 2.2
+
 ## [2.0.2]
 
 - Added Django 3.1 support. (Pull #63)

CHANGELOG.md

@@ -23,7 +23,7 @@ black-check:
 	black --check django_enumfield run_tests.py setup.py
 
 .PHONY: checks
-checks: flake8 mypy black-check
+checks: mypy flake8 black-check
 
 .PHONY: format
 format: black isort

Makefile

@@ -1,4 +1,3 @@
-import six
 from django.utils.translation import ugettext_lazy as _
 from rest_framework import serializers
 
@@ -18,7 +17,7 @@ def get_choice_value(self, enum_value):
         return enum_value.value
 
     def to_internal_value(self, data):
-        if isinstance(data, six.string_types) and data.isdigit():
+        if isinstance(data, str) and data.isdigit():
             data = int(data)
 
         try:

django_enumfield/contrib/drf.py

@@ -2,7 +2,6 @@
 from functools import partial
 from typing import Any, Callable  # noqa: F401
 
-import six
 from django import forms
 from django.db import models
 from django.utils.encoding import force_text
@@ -19,7 +18,6 @@
     def partialishmethod(method):
         return _partialmethod(method)
 
-
 except ImportError:  # pragma: no cover
     from django.utils.functional import curry
 
@@ -85,7 +83,7 @@ def from_db_value(self, value, *_):
 
     def to_python(self, value):
         if value is not None:
-            if isinstance(value, six.text_type) and value.isdigit():
+            if isinstance(value, str) and value.isdigit():
                 value = int(value)
             return self.enum.get(value)
 
@@ -120,10 +118,8 @@ def set_enum(self, new_value):
                 except ValueError:
                     raise InvalidStatusOperationError(
                         ugettext(
-                            six.text_type(
-                                "{value!r} is not one of the available choices "
-                                "for enum {enum}."
-                            )
+                            "{value!r} is not one of the available choices "
+                            "for enum {enum}."
                         ).format(value=new_value, enum=enum)
                     )
             setattr(self, private_att_name, new_value)

django_enumfield/db/fields.py

@@ -1,10 +1,9 @@
 from __future__ import absolute_import
 
 import logging
-from enum import Enum as NativeEnum, IntEnum as NativeIntEnum
-from typing import Any, Dict, List, Optional, Sequence, Tuple, TypeVar, Union, cast
-
-import six
+import enum
+from typing import Any, List, Optional, Sequence, Tuple, TypeVar, Union, cast, Mapping
+from django.utils.encoding import force_str
 
 try:
     from django.utils.functional import classproperty  # type: ignore
@@ -20,7 +19,7 @@
 RAISE = object()
 
 
-class BlankEnum(NativeEnum):
+class BlankEnum(enum.Enum):
     BLANK = ""
 
     @property
@@ -45,13 +44,12 @@ def __get__(self, instance, cls=None):
 T = TypeVar("T", bound="Enum")
 
 
-@six.python_2_unicode_compatible
-class Enum(NativeIntEnum):
-    """ A container for holding and restoring enum values """
+class Enum(enum.IntEnum):
+    """A container for holding and restoring enum values"""
 
-    __labels__ = {}  # type: Dict[int, six.text_type]
+    __labels__ = {}  # type: Mapping[int, str]
     __default__ = None  # type: Optional[int]
-    __transitions__ = {}  # type: Dict[int, Sequence[int]]
+    __transitions__ = {}  # type: Mapping[int, Sequence[int]]
 
     def __str__(self):
         return self.label
@@ -68,19 +66,20 @@ def label(self):
         :return: label for value
         :rtype: str
         """
-        label = cast(str, self.__class__.__labels__.get(self.value, self.name))
-        return six.text_type(label)
+        labels = self.__class__.__labels__
+        return force_str(labels.get(self.value, self.name))
 
-    @classproperty
+    @classproperty  # type: ignore[arg-type]
     def do_not_call_in_templates(cls):
         # type: () -> bool
         # Fix for Django templates so that any lookups of enums won't fail
         # More info: https://stackoverflow.com/questions/35953132/how-to-access-enum-types-in-django-templates  # noqa: E501
         return True
 
-    @classproperty
-    def values(cls):  # type: ignore
-        return {member.value: member for member in cls}
+    @classproperty  # type: ignore[arg-type]
+    def values(cls):
+        # type: () -> Mapping[int, Enum]
+        return {member.value: member for member in cls}  # type: ignore[attr-defined]
 
     def deconstruct(self):
         """
@@ -103,13 +102,13 @@ def items(cls):
 
     @classmethod
     def choices(cls, blank=False):
-        # type: (bool) -> List[Tuple[Union[int, str], NativeEnum]]
+        # type: (bool) -> List[Tuple[Union[int, str], enum.Enum]]
         """Choices for Enum
         :return: List of tuples (<value>, <member>)
         """
         choices = sorted(
             [(member.value, member) for member in cls], key=lambda x: x[0]
-        )  # type: List[Tuple[Union[str, int], NativeEnum]]
+        )  # type: List[Tuple[Union[str, int], enum.Enum]]
         if blank:
             choices.insert(0, (BlankEnum.BLANK.value, BlankEnum.BLANK))
         return choices
@@ -164,7 +163,7 @@ def get(
                 return cls(name_or_numeric)
             except ValueError:
                 pass
-        elif isinstance(name_or_numeric, six.string_types):
+        elif isinstance(name_or_numeric, str):
             try:
                 return cls[name_or_numeric]
             except KeyError:
@@ -225,4 +224,5 @@ def transition_origins(cls, to_value):
         """
         if isinstance(to_value, cls):
             to_value = to_value.value
-        return cast(Sequence[int], cls.__transitions__.get(to_value, []))
+
+        return cls.__transitions__.get(to_value, [])