diff --git a/golem/database/database.py b/golem/database/database.py index 7d41ee5c75..224e14153a 100644 --- a/golem/database/database.py +++ b/golem/database/database.py @@ -55,7 +55,7 @@ def execute_sql(self, sql, params=None, require_commit=True): class Database: - SCHEMA_VERSION = 28 + SCHEMA_VERSION = 29 def __init__(self, # noqa pylint: disable=too-many-arguments db: peewee.Database, diff --git a/golem/database/schemas/029_wallet_operation_type.py b/golem/database/schemas/029_wallet_operation_type.py new file mode 100644 index 0000000000..2cbd1770ba --- /dev/null +++ b/golem/database/schemas/029_wallet_operation_type.py @@ -0,0 +1,16 @@ +# pylint: disable=no-member +# pylint: disable=unused-argument +import peewee as pw + +SCHEMA_VERSION = 29 + + +def migrate(migrator, database, fake=False, **kwargs): + migrator.add_fields( + 'walletoperation', + operation_type=pw.CharField(default='task_payment'), + ) + + +def rollback(migrator, database, fake=False, **kwargs): + migrator.remove_fields('walletoperation', 'task_payment')