Python MigrationStatus Examples

Example #1

def sync_phone_numbers_for_domain(domain):
    for user_id in CouchUser.ids_by_domain(domain, is_active=True):
        _sync_user_phone_numbers(user_id)

    for user_id in CouchUser.ids_by_domain(domain, is_active=False):
        _sync_user_phone_numbers(user_id)

    case_ids = CaseAccessors(domain).get_case_ids_in_domain()
    for case in CaseAccessors(domain).iter_cases(case_ids):
        _sync_case_phone_number(case)

    MigrationStatus.set_migration_completed('phone_sync_domain_%s' % domain)

Example #2

File: tasks.py Project: dimagi/commcare-hq

def sync_phone_numbers_for_domain(domain):
    for user_id in CouchUser.ids_by_domain(domain, is_active=True):
        _sync_user_phone_numbers(user_id)

    for user_id in CouchUser.ids_by_domain(domain, is_active=False):
        _sync_user_phone_numbers(user_id)

    case_ids = CaseAccessors(domain).get_case_ids_in_domain()
    for case in CaseAccessors(domain).iter_cases(case_ids):
        _sync_case_phone_number(case)

    MigrationStatus.set_migration_completed('phone_sync_domain_%s' % domain)

Example #3

def assert_messaging_migration_complete(info):
    migrations_have_not_run = any(
        [not MigrationStatus.has_migration_completed(name) for name in info.migration_names]
    )
    is_fresh_install = os.environ.get('CCHQ_IS_FRESH_INSTALL') == '1'
    if migrations_have_not_run and not (settings.UNIT_TESTING or is_fresh_install):
        raise MigrationException(EXCEPTION_MESSAGE.render(info.context))

Example #4

File: migration_status.py Project: saketkanth/commcare-hq

def assert_messaging_migration_complete(info):
    migrations_have_not_run = any(
        [not MigrationStatus.has_migration_completed(name) for name in info.migration_names]
    )
    is_fresh_install = os.environ.get('CCHQ_IS_FRESH_INSTALL') == '1'
    if migrations_have_not_run and not (settings.UNIT_TESTING or is_fresh_install):
        raise MigrationException(EXCEPTION_MESSAGE.render(info.context))

Example #5

def assert_messaging_migration_complete(info):
    migrations_have_not_run = any([
        not MigrationStatus.has_migration_completed(name)
        for name in info.migration_names
    ])
    if migrations_have_not_run:
        raise MigrationException(EXCEPTION_MESSAGE.render(info.context))

Example #6

File: migrate_logs_to_sql.py Project: ansarbek/commcare-hq

 def handle(self, *args, **options):
     if not options['balance_only']:
         self.run_migration()
         MigrationStatus.set_migration_completed(MigrationStatus.MIGRATION_LOGS)
     self.balance()

Example #7

 def handle(self, *args, **options):
     with open('keyword_migration.log', 'w') as f:
         if not options['balance_only']:
             self.migrate(f)
             MigrationStatus.set_migration_completed(MigrationStatus.MIGRATION_KEYWORDS)
         self.balance(f)