using class property for delete

Author: dantownsend

piccolo/main.py

@@ -192,7 +192,7 @@ def backwards(migration_name: str):
                 MIGRATION_MODULES[s].backwards()  # type: ignore
             )
 
-            Migration.delete().where(
+            Migration.delete.where(
                 Migration.name == s
             ).run_sync()
     else:

piccolo/table.py

@@ -39,6 +39,7 @@ def __new__(cls, name, bases, namespace, **kwds):
                 columns.append(value)
                 value.name = key
 
+        # In case super classes also have columns.
         if bases:
             for base in bases:
                 if hasattr(base, 'Meta'):
@@ -49,6 +50,17 @@ def __new__(cls, name, bases, namespace, **kwds):
         table.Meta.columns = columns
         return table
 
+    ###########################################################################
+
+    @property
+    def delete(cls) -> Delete:
+        """
+        await Band.delete.where(Band.name == 'CSharps').run()
+        """
+        return Delete(
+            table=cls
+        )
+
 
 class Table(metaclass=TableMeta):
 
@@ -114,7 +126,7 @@ def remove(self):
 
         self.id = None
 
-        return self.__class__.delete().where(
+        return self.__class__.delete.where(
             self.__class__.id == _id
         )
 
@@ -232,15 +244,6 @@ def update(cls, **columns) -> Update:
             base=f'UPDATE {cls.Meta.tablename} SET {columns_str}'
         )
 
-    @classmethod
-    def delete(cls) -> Delete:
-        """
-        await Band.delete().where(Band.name == 'CSharps').run()
-        """
-        return Delete(
-            table=cls
-        )
-
     @classmethod
     def create(cls) -> Create:
         """

tests/table/test_delete.py

@@ -7,7 +7,7 @@ class TestDelete(DBTestCase):
     def test_delete(self):
         self.insert_rows()
 
-        Band.delete().where(
+        Band.delete.where(
             Band.name == 'CSharps'
         ).run_sync()