Remove ansiprint dependency

Author: Photonios

psqlextra/management/commands/pgpartition.py

@@ -1,10 +1,6 @@
-import sys
 
 from typing import Optional
 
-import colorama
-
-from ansimarkup import ansiprint, ansistring
 from django.conf import settings
 from django.core.management.base import BaseCommand
 from django.utils.module_loading import import_string
@@ -70,10 +66,6 @@ def handle(
         *args,
         **kwargs,
     ):
-        # disable coloring if no terminal is attached
-        if not sys.stdout.isatty():
-            colorama.init(strip=True)
-
         partitioning_manager = self._partitioning_manager()
 
         plan = partitioning_manager.plan(
@@ -83,7 +75,7 @@ def handle(
         creations_count = len(plan.creations)
         deletions_count = len(plan.deletions)
         if creations_count == 0 and deletions_count == 0:
-            ansiprint("<b><white>Nothing to be done.</white></b>")
+            print("Nothing to be done.")
             return
 
         plan.print()
@@ -92,18 +84,14 @@ def handle(
             return
 
         if not yes:
-            sys.stdout.write(
-                ansistring(
-                    "<b><white>Do you want to proceed? (y/N) </white></b>"
-                )
-            )
+            print("Do you want to proceed? (y/N) ")
 
             if not self._ask_for_confirmation():
-                ansiprint("<b><white>Operation aborted.</white></b>")
+                print("Operation aborted.")
                 return
 
         plan.apply(using=using)
-        ansiprint("<b><white>Operations applied.</white></b>")
+        print("Operations applied.")
 
     @staticmethod
     def _ask_for_confirmation() -> bool:

psqlextra/partitioning/plan.py

@@ -1,7 +1,6 @@
 from dataclasses import dataclass, field
 from typing import List, Optional
 
-from ansimarkup import ansiprint
 from django.db import connections, transaction
 
 from .config import PostgresPartitioningConfig
@@ -49,17 +48,17 @@ def apply(self, using: Optional[str]) -> None:
     def print(self) -> None:
         """Prints this model plan to the terminal in a readable format."""
 
-        ansiprint(f"<b><white>{self.config.model.__name__}:</white></b>")
+        print(f"{self.config.model.__name__}:")
 
         for partition in self.deletions:
-            ansiprint("<b><red>  - %s</red></b>" % partition.name())
+            print("  - %s" % partition.name())
             for key, value in partition.deconstruct().items():
-                ansiprint(f"<white>     <b>{key}</b>: {value}</white>")
+                print(f"     {key}: {value}")
 
         for partition in self.creations:
-            ansiprint("<b><green>  + %s</green></b>" % partition.name())
+            print("  + %s" % partition.name())
             for key, value in partition.deconstruct().items():
-                ansiprint(f"<white>     <b>{key}</b>: {value}</white>")
+                print(f"     {key}: {value}")
 
 
 @dataclass
@@ -104,12 +103,8 @@ def print(self) -> None:
         create_count = len(self.creations)
         delete_count = len(self.deletions)
 
-        ansiprint(
-            f"<b><red>{delete_count} partitions will be deleted</red></b>"
-        )
-        ansiprint(
-            f"<b><green>{create_count} partitions will be created</green></b>"
-        )
+        print(f"{delete_count} partitions will be deleted")
+        print(f"{create_count} partitions will be created")
 
 
 __all__ = ["PostgresPartitioningPlan", "PostgresModelPartitioningPlan"]

setup.py

@@ -67,7 +67,6 @@ def run(self):
     install_requires=[
         "Django>=2.0,<5.0",
         "python-dateutil>=2.8.0,<=3.0.0",
-        "ansimarkup>=1.4.0,<=2.0.0",
     ],
     extras_require={
         ':python_version <= "3.6"': ["dataclasses"],