Fix several issues

Co-authored-by: monouno <noreply@github.com>

Author: monouno

README.rst

@@ -17,13 +17,20 @@ django-pg-partitioning
 .. image:: https://codecov.io/gh/chaitin/django-pg-partitioning/branch/master/graph/badge.svg
    :target: https://codecov.io/gh/chaitin/django-pg-partitioning
 
+⚠️
+----
+
+目前已经有了更好用的 Django 插件（比如 django-postgres-extra）使得基于 Django 开发的项目能够方便地使用 PostgreSQL 数据库的高级功能，因此本项目不再维护。你依然可以 fork 本项目并进行二次开发，祝你生活愉快 :)
+
+There are already better Django plugins (such as django-postgres-extra) that make it easy for Django-based projects to use the advanced features of PostgreSQL databases, so this project is no longer maintained. You can still fork this project and do secondary development, have a nice life :)
+
+----
+
 一个支持 PostgreSQL 11 原生表分区的 Django 扩展，使您可以在 Django 中创建分区表并管理它们。目前它支持两种分区类型：
 
 - 时间范围分区（Time Range Partitioning）：将时序数据分开存储到不同的时间范围分区表中，支持创建连续且不重叠的时间范围分区并进行归档管理。
 - 列表分区（List Partitioning）：根据分区字段的确定值将数据分开存储到不同的分区表中。
 
-----
-
 A Django extension that supports PostgreSQL 11 native table partitioning, allowing you to create partitioned tables in Django
 and manage them. Currently it supports the following two partition types:
 

dev/postgres_init.sh

@@ -2,6 +2,7 @@
 from typing import Type
 
 from django.db import models
+
 from pg_partitioning.manager import ListPartitionManager, TimeRangePartitionManager
 
 logger = logging.getLogger(__name__)
@@ -75,8 +76,5 @@ class MyLog(models.Model):
 
     def __call__(self, model: Type[models.Model]):
         super().__call__(model)
-        support_field_types = [item.get_internal_type() for item in [models.TextField(), models.BooleanField(), models.IntegerField()]]
-        if model._meta.get_field(self.partition_key).get_internal_type() not in support_field_types:
-            raise NotImplementedError("The partition_key does not support this field type.")
         model.partitioning = ListPartitionManager(model, self.partition_key, self.options)
         return model

pg_partitioning/decorators.py

@@ -8,6 +8,7 @@
 from django.db import IntegrityError, models
 from django.db.models import Q
 from django.utils import timezone
+
 from pg_partitioning.shortcuts import double_quote, execute_sql, generate_set_indexes_tablespace_sql, single_quote
 
 from .constants import (
@@ -197,13 +198,15 @@ def create_partition(self, partition_name: str, value: Union[str, int, bool, Non
           tablespace(str): Partition tablespace name.
         """
 
-        sql_sequence = [
-            SQL_CREATE_LIST_PARTITION % {"parent": double_quote(self.model._meta.db_table), "child": double_quote(partition_name), "value": _db_value(value)}
-        ]
+        create_partition_sql = SQL_CREATE_LIST_PARTITION % {
+            "parent": double_quote(self.model._meta.db_table),
+            "child": double_quote(partition_name),
+            "value": _db_value(value),
+        }
         if tablespace:
-            sql_sequence[0] += SQL_APPEND_TABLESPACE % {"tablespace": tablespace}
-            sql_sequence.extend(generate_set_indexes_tablespace_sql(partition_name, tablespace))
-        execute_sql(sql_sequence)
+            create_partition_sql += SQL_APPEND_TABLESPACE % {"tablespace": tablespace}
+        execute_sql(create_partition_sql)
+        execute_sql(generate_set_indexes_tablespace_sql(partition_name, tablespace))
 
     def attach_partition(self, partition_name: str, value: Union[str, int, bool, None], tablespace: str = None) -> None:
         """Attach partitions.
@@ -219,8 +222,7 @@ def attach_partition(self, partition_name: str, value: Union[str, int, bool, Non
             sql_sequence.append(SQL_SET_TABLE_TABLESPACE % {"name": double_quote(partition_name), "tablespace": tablespace})
             sql_sequence.extend(generate_set_indexes_tablespace_sql(partition_name, tablespace))
         sql_sequence.append(
-            SQL_ATTACH_LIST_PARTITION
-            % {"parent": double_quote(self.model._meta.db_table), "child": double_quote(partition_name), "value": single_quote(_db_value(value))}
+            SQL_ATTACH_LIST_PARTITION % {"parent": double_quote(self.model._meta.db_table), "child": double_quote(partition_name), "value": _db_value(value)}
         )
         execute_sql(sql_sequence)
 

pg_partitioning/manager.py

@@ -1,5 +1,6 @@
 from django.apps import apps
 from django.db import models, transaction
+
 from pg_partitioning.signals import post_attach_partition, post_create_partition, post_detach_partition
 
 from .constants import (
@@ -75,23 +76,20 @@ def save(self, force_insert=False, force_update=False, using=None, update_fields
 
         model = apps.get_model(self.config.model_label)
         if self._state.adding:
-            sql_sequence = [
-                SQL_CREATE_TIME_RANGE_PARTITION
-                % {
-                    "parent": double_quote(model._meta.db_table),
-                    "child": double_quote(self.table_name),
-                    "date_start": single_quote(self.start.isoformat()),
-                    "date_end": single_quote(self.end.isoformat()),
-                }
-            ]
+            create_partition_sql = SQL_CREATE_TIME_RANGE_PARTITION % {
+                "parent": double_quote(model._meta.db_table),
+                "child": double_quote(self.table_name),
+                "date_start": single_quote(self.start.isoformat()),
+                "date_end": single_quote(self.end.isoformat()),
+            }
 
             if self.config.attach_tablespace:
-                sql_sequence[0] += SQL_APPEND_TABLESPACE % {"tablespace": self.config.attach_tablespace}
-                sql_sequence.extend(generate_set_indexes_tablespace_sql(self.table_name, self.config.attach_tablespace))
+                create_partition_sql += SQL_APPEND_TABLESPACE % {"tablespace": self.config.attach_tablespace}
 
             with transaction.atomic():
                 super().save(force_insert, force_update, using, update_fields)
-                execute_sql(sql_sequence)
+                execute_sql(create_partition_sql)
+                execute_sql(generate_set_indexes_tablespace_sql(self.table_name, self.config.attach_tablespace))
                 post_create_partition.send(sender=model, partition_log=self)
         else:
             with transaction.atomic():

pg_partitioning/models.py

@@ -3,6 +3,7 @@
 
 from django.apps.config import MODELS_MODULE_NAME
 from django.db.backends.postgresql.schema import DatabaseSchemaEditor
+
 from pg_partitioning.manager import _PartitionManagerBase
 
 logger = logging.getLogger(__name__)

pg_partitioning/patch/schema.py

@@ -2,6 +2,7 @@
 from typing import List, Optional, Tuple, Union
 
 from django.db import connection
+
 from pg_partitioning.constants import SQL_DROP_TABLE, SQL_GET_TABLE_INDEXES, SQL_SET_INDEX_TABLESPACE, SQL_SET_TABLE_TABLESPACE, SQL_TRUNCATE_TABLE
 
 logger = logging.getLogger(__name__)
@@ -23,8 +24,12 @@ def double_quote(name: str) -> str:
     return '"%s"' % name
 
 
-def execute_sql(sql_sequence: Union[str, List[str], Tuple[str]], fetch: bool = False) -> Optional[Tuple]:
+def execute_sql(sql_sequence: Union[str, List[str], Tuple[str]], fetch: bool = False) -> Optional[List]:
     """Execute SQL sequence and returning result."""
+    if not sql_sequence:
+        if fetch:
+            return []
+        return
 
     sql_str = ""
     for statement in sql_sequence if isinstance(sql_sequence, (list, tuple)) else [sql_sequence]:

pg_partitioning/shortcuts.py

@@ -1,6 +1,7 @@
 from django.contrib.postgres.indexes import BrinIndex
 from django.db import models
 from django.utils import timezone
+
 from pg_partitioning.constants import PeriodType
 from pg_partitioning.decorators import ListPartitioning, TimeRangePartitioning
 
@@ -29,6 +30,18 @@ class Meta:
 
 
 @ListPartitioning(partition_key="category")
-class ListTable(models.Model):
+class ListTableText(models.Model):
     category = models.TextField(default="A", null=True, blank=True)
     timestamp = models.DateTimeField(default=timezone.now)
+
+
+@ListPartitioning(partition_key="category")
+class ListTableInt(models.Model):
+    category = models.IntegerField(default=0, null=True)
+    timestamp = models.DateTimeField(default=timezone.now)
+
+
+@ListPartitioning(partition_key="category")
+class ListTableBool(models.Model):
+    category = models.NullBooleanField(default=False, null=True)
+    timestamp = models.DateTimeField(default=timezone.now)