make both category and protocol a list strings

carbonplan_offsets_db/models.py

@@ -1,8 +1,8 @@
 import datetime
 
 import pydantic
-from sqlalchemy import BigInteger, Column
-from sqlmodel import Field, Relationship, SQLModel
+from sqlalchemy.dialects import postgresql
+from sqlmodel import BigInteger, Column, Field, Relationship, SQLModel, String
 
 from .schemas import FileCategory, FileStatus, Pagination
 
@@ -32,8 +32,12 @@ class ProjectBase(SQLModel):
     name: str | None = Field(description='Name of the project')
     registry: str = Field(description='Name of the registry')
     proponent: str | None
-    protocol: str | None
-    category: str | None
+    protocol: list[str] | None = Field(
+        description='List of protocols', default=None, sa_column=Column(postgresql.ARRAY(String()))
+    )
+    category: list[str] | None = Field(
+        description='List of categories', default=None, sa_column=Column(postgresql.ARRAY(String()))
+    )
     developer: str | None
     voluntary_status: str | None
     country: str | None

migrations/versions/d2bc99dedcf6_make_both_category_and_protocol_a_list_.py

@@ -0,0 +1,26 @@
+"""make both category and protocol a list strings
+
+Revision ID: d2bc99dedcf6
+Revises: e0c9ea45fc91
+Create Date: 2023-09-18 06:52:14.701020
+
+"""
+
+
+# revision identifiers, used by Alembic.
+revision = 'd2bc99dedcf6'
+down_revision = 'e0c9ea45fc91'
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###