SchemaMigration

Complexity

Total Complexity

5

Size/Duplication

Total Lines	41
Duplicated Lines	0 %

Test Coverage

Coverage

95.24%

3 Methods

Rating	Name	Duplication	Size	Complexity
A	run()	0	15	2
A	reset()	0	20	2
A	_build_router()	0	3	1

from plugin.core.database import Database
from plugin.models.core import db, migrations_path
from plugin.modules.migrations.core.base import Migration

from peewee_migrate.core import Router
import logging

log = logging.getLogger(__name__)


class SchemaMigration(Migration):
    def run(self):
        log.debug('migrations_path: %r', migrations_path)

        router = self._build_router()

        # Validate current schema
        if not router.validate():
            log.error('Detected corrupt/invalid database schema, resetting database...')
            return self.reset()
        else:
            log.info('Database schema is valid')

        # Run schema migrations
        router.run()
        return True

    @classmethod
    def reset(cls):
        if not Database.reset('main', db, 'invalid-schema'):
            # Unable to reset database
            return False

        # Run migrations
        router = cls._build_router()
        router.run()

        # Log message to channel menu
        from plugin.managers.message import MessageManager
        from plugin.models import Message

        MessageManager.get.from_message(logging.WARNING,

It seems like a value for argument message is missing in the unbound method call.

            "Plugin database has been automatically reset due to schema corruption, see http://bit.ly/TFPx90101 for more details",

This line is too long as per the coding-style (130/120).

            code=Message.Code.DatabaseSchemaCorruptionReset
        )

        return True

    @staticmethod
    def _build_router():
        return Router(migrations_path, DATABASE=db)