1 files changed, 29 insertions, 0 deletions
diff --git a/mediagoblin/db/migrations.py b/mediagoblin/db/migrations.py
index 36ad736a..2df06fc0 100644
--- a/mediagoblin/db/migrations.py
+++ b/mediagoblin/db/migrations.py
@@ -1693,6 +1693,15 @@ def federation_collection_schema(db):
     # Add the fields onto the Collection model, we need to set these as
     # not null to avoid DB integreity errors. We will add the not null
     # constraint later.
+    public_id_column = Column(
+        "public_id",
+        Unicode,
+        unique=True
+    )
+    public_id_column.create(
+        collection_table,
+        unique_name="collection_public_id")
+
     updated_column = Column(
         "updated",
         DateTime,
@@ -1846,3 +1855,23 @@ def federation_graveyard(db):
 
     # Commit changes to the db
     db.commit()
+
+@RegisterMigration(40, MIGRATIONS)
+def add_public_id(db):
+    metadata = MetaData(bind=db.bind)
+
+    # Get the table
+    activity_table = inspect_table(metadata, "core__activities")
+    activity_public_id = Column(
+        "public_id",
+        Unicode,
+        unique=True,
+        nullable=True
+    )
+    activity_public_id.create(
+        activity_table,
+        unique_name="activity_public_id"
+    )
+
+    # Commit this.
+    db.commit()