def set_couch_sql_migration_complete(domain):
from corehq.apps.couch_sql_migration.couchsqlmigration import commit_migration
commit_migration(domain)
set_migration_complete(domain, COUCH_TO_SQL_SLUG)
# we get this for free
set_tz_migration_complete(domain)
_notify_dimagi_users_on_domain(domain)
def set_couch_sql_migration_complete(domain):
from corehq.apps.couch_sql_migration.couchsqlmigration import commit_migration
commit_migration(domain)
# no need to keep this around anymore since state is kept on domain model
DomainMigrationProgress.objects.filter(domain=domain, migration_slug=COUCH_TO_SQL_SLUG).delete()
# we get this for free
set_tz_migration_complete(domain)
def set_couch_sql_migration_complete(domain):
from corehq.apps.couch_sql_migration.couchsqlmigration import commit_migration
commit_migration(domain)
set_migration_complete(domain, COUCH_TO_SQL_SLUG)
# we get this for free
set_tz_migration_complete(domain)
_notify_dimagi_users_on_domain(domain)
def set_couch_sql_migration_complete(domain):
from corehq.apps.couch_sql_migration.couchsqlmigration import commit_migration
commit_migration(domain)
# no need to keep this around anymore since state is kept on domain model
DomainMigrationProgress.objects.filter(domain=domain, migration_slug=COUCH_TO_SQL_SLUG).delete()
# we get this for free
set_tz_migration_complete(domain)
_notify_dimagi_users_on_domain(domain)
def test_pause(self):
xform = self.get_xml('form')
set_tz_migration_started(self.domain)
with self.assertRaisesRegexp(LocalSubmissionError, 'status code 503'):
submit_form_locally(xform, self.domain)
_run_timezone_migration_for_domain(self.domain)
set_tz_migration_complete(self.domain)
# no issue
submit_form_locally(xform, self.domain)
def test_pause(self):
xform = self.get_xml('form')
set_tz_migration_started(self.domain)
with self.assertRaisesRegexp(LocalSubmissionError, 'status code 503'):
submit_form_locally(xform, self.domain)
_run_timezone_migration_for_domain(self.domain)
set_tz_migration_complete(self.domain)
# no issue
submit_form_locally(xform, self.domain)
def save(self, **params):
self.last_modified = datetime.utcnow()
if not self._rev:
# mark any new domain as timezone migration complete
set_tz_migration_complete(self.name)
super(Domain, self).save(**params)
from corehq.apps.domain.signals import commcare_domain_post_save
results = commcare_domain_post_save.send_robust(sender='domain', domain=self)
log_signal_errors(results, "Error occurred during domain post_save (%s)", {'domain': self.name})
def save(self, **params):
self.last_modified = datetime.utcnow()
if not self._rev:
# mark any new domain as timezone migration complete
set_tz_migration_complete(self.name)
super(Domain, self).save(**params)
from corehq.apps.domain.signals import commcare_domain_post_save
results = commcare_domain_post_save.send_robust(sender='domain', domain=self)
log_signal_errors(results, "Error occurred during domain post_save (%s)", {'domain': self.name})
def save(self, **params):
from corehq.apps.domain.dbaccessors import domain_or_deleted_domain_exists
self.last_modified = datetime.utcnow()
if not self._rev:
if domain_or_deleted_domain_exists(self.name):
raise NameUnavailableException(self.name)
# mark any new domain as timezone migration complete
set_tz_migration_complete(self.name)
super(Domain, self).save(**params)
from corehq.apps.domain.signals import commcare_domain_post_save
results = commcare_domain_post_save.send_robust(sender='domain', domain=self)
log_signal_errors(results, "Error occurred during domain post_save (%s)", {'domain': self.name})
def save(self, **params):
self.last_modified = datetime.utcnow()
if not self._rev:
# mark any new domain as timezone migration complete
set_tz_migration_complete(self.name)
super(Domain, self).save(**params)
from corehq.apps.domain.signals import commcare_domain_post_save
results = commcare_domain_post_save.send_robust(sender='domain', domain=self)
for result in results:
# Second argument is None if there was no error
if result[1]:
notify_exception(
None,
message="Error occured during domain post_save %s: %s" %
(self.name, str(result[1]))
)
def handle(self, domain, **options):
if should_use_sql_backend(domain):
raise CommandError(
'This command only works for couch-based domains.')
filepath = get_planning_db_filepath(domain)
self.stdout.write('Using file {}\n'.format(filepath))
if options['BEGIN']:
self.require_only_option('BEGIN', options)
set_tz_migration_started(domain)
if options['ABORT']:
self.require_only_option('ABORT', options)
set_tz_migration_not_started(domain)
if options['blow_away']:
delete_planning_db(domain)
self.stdout.write('Removed file {}\n'.format(filepath))
if options['prepare']:
self.planning_db = prepare_planning_db(domain)
self.stdout.write('Created and loaded file {}\n'.format(filepath))
else:
self.planning_db = get_planning_db(domain)
if options['COMMIT']:
self.require_only_option('COMMIT', options)
assert get_tz_migration_status(
domain, strict=True) == MigrationStatus.IN_PROGRESS
commit_plan(domain, self.planning_db)
set_tz_migration_complete(domain)
if options['prepare_case_json']:
prepare_case_json(self.planning_db)
if options['stats']:
self.valiate_forms_and_cases(domain)
if options['show_diffs']:
self.show_diffs()
if options['play']:
from corehq.apps.tzmigration.planning import *
session = self.planning_db.Session() # noqa
try:
import ipdb as pdb
except ImportError:
import pdb
pdb.set_trace()
def handle(self, domain, **options):
if should_use_sql_backend(domain):
raise CommandError('This command only works for couch-based domains.')
filepath = get_planning_db_filepath(domain)
self.stdout.write('Using file {}\n'.format(filepath))
if options['BEGIN']:
self.require_only_option('BEGIN', options)
set_tz_migration_started(domain)
if options['ABORT']:
self.require_only_option('ABORT', options)
set_tz_migration_not_started(domain)
if options['blow_away']:
delete_planning_db(domain)
self.stdout.write('Removed file {}\n'.format(filepath))
if options['prepare']:
self.planning_db = prepare_planning_db(domain)
self.stdout.write('Created and loaded file {}\n'.format(filepath))
else:
self.planning_db = get_planning_db(domain)
if options['COMMIT']:
self.require_only_option('COMMIT', options)
assert get_tz_migration_status(domain, strict=True) == MigrationStatus.IN_PROGRESS
commit_plan(domain, self.planning_db)
set_tz_migration_complete(domain)
if options['prepare_case_json']:
prepare_case_json(self.planning_db)
if options['stats']:
self.valiate_forms_and_cases(domain)
if options['show_diffs']:
self.show_diffs()
if options['play']:
from corehq.apps.tzmigration.planning import *
session = self.planning_db.Session() # noqa
try:
import ipdb as pdb
except ImportError:
import pdb
pdb.set_trace()
def test_complete(self):
set_tz_migration_complete('green')
self.assertEqual(get_tz_migration_status('green'),
MigrationStatus.COMPLETE)
self.assertTrue(get_tz_migration_complete('green'))
def run_timezone_migration_for_domain(domain):
set_tz_migration_started(domain)
_run_timezone_migration_for_domain(domain)
set_tz_migration_complete(domain)
def run_timezone_migration_for_domain(domain):
set_tz_migration_started(domain)
_run_timezone_migration_for_domain(domain)
set_tz_migration_complete(domain)
