adds m2m column to the playground (#1234)

* adds m2m column to the playground

* adds @dantownsend sugestions

* add a reason

---------

Co-authored-by: Daniel Townsend <[email protected]>

Author: sinisaos

piccolo/apps/playground/commands/run.py

@@ -12,13 +12,15 @@
 
 from piccolo.columns import (
     JSON,
+    M2M,
     UUID,
     Array,
     Boolean,
     Date,
     ForeignKey,
     Integer,
     Interval,
+    LazyTableReference,
     Numeric,
     Serial,
     Text,
@@ -49,6 +51,7 @@ class Band(Table):
     name = Varchar(length=50)
     manager = ForeignKey(references=Manager, null=True)
     popularity = Integer()
+    genres = M2M(LazyTableReference("GenreToBand", module_path=__name__))
 
     @classmethod
     def get_readable(cls) -> Readable:
@@ -161,6 +164,26 @@ def get_readable(cls) -> Readable:
         )
 
 
+class Genre(Table):
+    id: Serial
+    name = Varchar()
+    bands = M2M(LazyTableReference("GenreToBand", module_path=__name__))
+
+    @classmethod
+    def get_readable(cls) -> Readable:
+        return Readable(
+            template="%s",
+            columns=[cls.name],
+        )
+
+
+class GenreToBand(Table):
+    id: Serial
+    band = ForeignKey(Band)
+    genre = ForeignKey(Genre)
+    reason = Text(null=True, default=None)
+
+
 TABLES = (
     Manager,
     Band,
@@ -171,6 +194,8 @@ def get_readable(cls) -> Readable:
     DiscountCode,
     RecordingStudio,
     Album,
+    Genre,
+    GenreToBand,
 )
 
 
@@ -282,6 +307,24 @@ def populate():
         ),
     ).run_sync()
 
+    genres = Genre.insert(
+        Genre(name="Rock"),
+        Genre(name="Classical"),
+        Genre(name="Folk"),
+    ).run_sync()
+
+    GenreToBand.insert(
+        GenreToBand(
+            band=pythonistas.id,
+            genre=genres[0]["id"],
+            reason="Because they rock.",
+        ),
+        GenreToBand(band=pythonistas.id, genre=genres[2]["id"]),
+        GenreToBand(band=rustaceans.id, genre=genres[2]["id"]),
+        GenreToBand(band=c_sharps.id, genre=genres[0]["id"]),
+        GenreToBand(band=c_sharps.id, genre=genres[1]["id"]),
+    ).run_sync()
+
 
 def run(
     engine: str = "sqlite",

tests/columns/m2m/base.py

@@ -11,7 +11,7 @@
 from piccolo.engine.finder import engine_finder
 from piccolo.schema import SchemaManager
 from piccolo.table import Table, create_db_tables_sync, drop_db_tables_sync
-from tests.base import engine_is, engines_skip
+from tests.base import engines_skip
 
 engine = engine_finder()
 
@@ -51,54 +51,25 @@ def _setUp(self, schema: Optional[str] = None):
 
         create_db_tables_sync(*self.all_tables, if_not_exists=True)
 
-        if engine_is("cockroach"):
-            bands = (
-                Band.insert(
-                    Band(name="Pythonistas"),
-                    Band(name="Rustaceans"),
-                    Band(name="C-Sharps"),
-                )
-                .returning(Band.id)
-                .run_sync()
-            )
-
-            genres = (
-                Genre.insert(
-                    Genre(name="Rock"),
-                    Genre(name="Folk"),
-                    Genre(name="Classical"),
-                )
-                .returning(Genre.id)
-                .run_sync()
-            )
-
-            GenreToBand.insert(
-                GenreToBand(band=bands[0]["id"], genre=genres[0]["id"]),
-                GenreToBand(band=bands[0]["id"], genre=genres[1]["id"]),
-                GenreToBand(band=bands[1]["id"], genre=genres[1]["id"]),
-                GenreToBand(band=bands[2]["id"], genre=genres[0]["id"]),
-                GenreToBand(band=bands[2]["id"], genre=genres[2]["id"]),
-            ).run_sync()
-        else:
-            Band.insert(
-                Band(name="Pythonistas"),
-                Band(name="Rustaceans"),
-                Band(name="C-Sharps"),
-            ).run_sync()
+        bands = Band.insert(
+            Band(name="Pythonistas"),
+            Band(name="Rustaceans"),
+            Band(name="C-Sharps"),
+        ).run_sync()
 
-            Genre.insert(
-                Genre(name="Rock"),
-                Genre(name="Folk"),
-                Genre(name="Classical"),
-            ).run_sync()
+        genres = Genre.insert(
+            Genre(name="Rock"),
+            Genre(name="Folk"),
+            Genre(name="Classical"),
+        ).run_sync()
 
-            GenreToBand.insert(
-                GenreToBand(band=1, genre=1),
-                GenreToBand(band=1, genre=2),
-                GenreToBand(band=2, genre=2),
-                GenreToBand(band=3, genre=1),
-                GenreToBand(band=3, genre=3),
-            ).run_sync()
+        GenreToBand.insert(
+            GenreToBand(band=bands[0]["id"], genre=genres[0]["id"]),
+            GenreToBand(band=bands[0]["id"], genre=genres[1]["id"]),
+            GenreToBand(band=bands[1]["id"], genre=genres[1]["id"]),
+            GenreToBand(band=bands[2]["id"], genre=genres[0]["id"]),
+            GenreToBand(band=bands[2]["id"], genre=genres[2]["id"]),
+        ).run_sync()
 
     def tearDown(self):
         drop_db_tables_sync(*self.all_tables)