blob: 591033ce1244af6054ec1a5011fcbbc5282d4fde (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
"""List subscription policy
Revision ID: 16c2b25c7b
Revises: 46e92facee7
Create Date: 2015-03-21 11:00:44.634883
"""
import sqlalchemy as sa
from alembic import op
from mailman.database.helpers import exists_in_db, is_sqlite
from mailman.database.types import Enum
from mailman.interfaces.mailinglist import SubscriptionPolicy
# Revision identifiers, used by Alembic.
revision = '16c2b25c7b'
down_revision = '46e92facee7'
def upgrade():
# Update the schema.
if not exists_in_db(op.get_bind(), 'mailinglist', 'subscription_policy'):
# SQLite may not have removed it when downgrading.
op.add_column('mailinglist', sa.Column(
'subscription_policy', Enum(SubscriptionPolicy), nullable=True))
# Now migrate the data. Don't import the table definition from the
# models, it may break this migration when the model is updated in the
# future (see the Alembic doc).
mlist = sa.sql.table(
'mailinglist',
sa.sql.column('subscription_policy', Enum(SubscriptionPolicy))
)
# There were no enforced subscription policy before, so all lists are
# considered open.
op.execute(mlist.update().values(
{'subscription_policy': op.inline_literal(SubscriptionPolicy.open)}))
def downgrade():
if not is_sqlite(op.get_bind()):
# SQLite does not support dropping columns.
op.drop_column('mailinglist', 'subscription_policy')
|
