Remove enum34, convert to Pytest-style, add more supported targets

.python-version

@@ -0,0 +1,3 @@
+3.6
+3.8
+3.9

CHANGELOG.md

@@ -1,3 +1,23 @@
+# 3.1.0
+
+- No functional changes: added support targets and refactored tests and dependencies.
+
+## Major
+
+- Support Django 3.2 and 4.2.
+- Add support for Python 3.8 and 3.9.
+- Cope with psycopg2 bug: only 2.8 is supported through Django 2.2, but 2.9 is supported for higher Django versions.
+- Parameterize test suite on USE_TZ to confirm Capone works for both True and False.
+- Remove unneeded dependency `enum34`.
+- Refactor to use `django.utils.timezone`.
+
+## Minor
+
+- Convert tests to Pytest style.
+- Default in tests to USE_TZ == True.
+- Remove remaining Python 2 compatibility shims.
+- Unpin lots of test dependencies.
+
 # 3.0.0
 
 - Drop Django < 1.11 support.

Makefile

@@ -71,7 +71,6 @@ clean:
 	find . -name __pycache__ -delete
 	rm -f MANIFEST
 	rm -f test.db
-	rm -f xunit.xml
 	find . -type f -name '*.pyc' -delete
 	rm -rf coverage .coverage*
 	dropdb --if-exists capone_test_db -U postgres

capone/models.py

@@ -10,8 +10,7 @@
 from django.contrib.contenttypes.models import ContentType
 from django.db import models
 from django.db.models import Q
-from django.utils.encoding import python_2_unicode_compatible
-from django.utils.translation import ugettext_lazy as _
+from django.utils.translation import gettext_lazy as _
 
 from capone.exceptions import TransactionBalanceException
 
@@ -20,7 +19,6 @@
 NEGATIVE_DEBITS_HELP_TEXT = "Amount for this entry.  Debits are negative, and credits are positive."  # noqa: E501
 
 
-@python_2_unicode_compatible
 class TransactionRelatedObject(models.Model):
     """
     A piece of evidence for a particular Transaction.
@@ -164,7 +162,6 @@ def filter_by_related_objects(
             raise ValueError("Invalid match_type.")
 
 
-@python_2_unicode_compatible
 class TransactionType(models.Model):
     """
     A user-defined "type" to group `Transactions`.
@@ -202,7 +199,6 @@ def get_or_create_manual_transaction_type_id():
     return get_or_create_manual_transaction_type().id
 
 
-@python_2_unicode_compatible
 class Transaction(models.Model):
     """
     The main model for representing a financial event in `capone`.
@@ -267,10 +263,11 @@ def validate(self):
         Instead, the only check that makes sense is that the entries for the
         transaction still balance.
         """
-        total = sum([entry.amount for entry in self.entries.all()])
-        if total != Decimal(0):
-            raise TransactionBalanceException(
-                "Credits do not equal debits. Mis-match of %s." % total)
+        if self.pk:
+            total = sum([entry.amount for entry in self.entries.all()])
+            if total != Decimal(0):
+                raise TransactionBalanceException(
+                    "Credits do not equal debits. Mis-match of %s." % total)
         return True
 
     def save(self, **kwargs):
@@ -292,7 +289,6 @@ def summary(self):
         }
 
 
-@python_2_unicode_compatible
 class Ledger(models.Model):
     """
     A group of `LedgerEntries` all debiting or crediting the same resource.
@@ -368,7 +364,6 @@ def __str__(self):
             amount=self.amount, ledger=self.ledger.name)
 
 
-@python_2_unicode_compatible
 class LedgerBalance(models.Model):
     """
     A Denormalized balance for a related object in a ledger.

capone/tests/factories.py

@@ -13,7 +13,7 @@
 from capone.tests.models import Order
 
 
-class UserFactory(factory.DjangoModelFactory):
+class UserFactory(factory.django.DjangoModelFactory):
     """
     Factory for django.contrib.auth.get_user_model()
 
@@ -27,7 +27,7 @@ class Meta:
     email = username = factory.Sequence(lambda n: "TransactionUser #%s" % n)
 
 
-class LedgerFactory(factory.DjangoModelFactory):
+class LedgerFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = Ledger
 
@@ -36,22 +36,22 @@ class Meta:
     number = factory.Sequence(lambda n: n)
 
 
-class OrderFactory(factory.DjangoModelFactory):
+class OrderFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = Order
 
     patient_name = factory.Sequence(lambda n: "Patient %s" % n)
     barcode = factory.Sequence(lambda n: str(n))
 
 
-class CreditCardTransactionFactory(factory.DjangoModelFactory):
+class CreditCardTransactionFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = CreditCardTransaction
 
     cardholder_name = factory.Sequence(lambda n: "Cardholder %s" % n)
 
 
-class TransactionTypeFactory(factory.DjangoModelFactory):
+class TransactionTypeFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = TransactionType
 

capone/tests/settings.py

@@ -44,3 +44,5 @@
     'django.contrib.messages.middleware.MessageMiddleware',
 )
 SESSION_ENGINE = 'django.contrib.sessions.backends.db'
+
+USE_TZ = True