def run_footprint_init(self, *args, **options):
if not settings.CELERY_ALWAYS_EAGER:
raise Exception('This command must run with settings.CELERY_ALWAYS_EQUAL = True. '
'Add --settings=footprint.settings_init to the command line.')
db_entity_keys = options.get('db_entity_keys').split(',') if options.get('db_entity_keys') else None
# Replace so we can use options as kwargs
options['db_entity_keys'] = db_entity_keys
config_entity_keys = options.get('config_entity_keys').split(',') if options.get('config_entity_keys') else None
# Replace so we can use options as kwargs
options['config_entity_keys'] = config_entity_keys
if not options.get('run_analysis'):
AnalysisModule._no_post_save_task_run_global = True
limit_to_classes = map(
lambda cls: resolve_model('main.%s' % cls), (options.get('class').split(',') if options.get('class') else [])
)
options['limit_to_classes'] = limit_to_classes
# Perforance testing
if options.get('memory'):
ConfigEntity.init_heapy()
ConfigEntity.start_heapy_diagnosis()
# Delete all ConfigEntity intances so they can be recreated.
# This will cascade delete related models, but it doesn't delete
# BuiltForms and other independent models
if options.get('recreate'):
for cls in filter_classes(limit_to_classes):
cls.objects.all().delete()
# Delete deleted config_entities
if options.get('recycle'):
for cls in filter_classes(limit_to_classes):
cls.objects.filter(deleted=True).delete()
if options.get('delete_clones') or options.get('delete_scenario_clones'):
# Delete clones and uploads
for cls in filter_classes(limit_to_classes):
all_config_entities = cls.objects.all()
for config_entity in all_config_entities:
if options.get('delete_clones'):
db_entities = map(
lambda db_entity_interest: db_entity_interest.db_entity,
DbEntityInterest.objects.filter(
config_entity=config_entity,
db_entity__origin_instance__isnull=False)
) +\
filter(
lambda db_entity: db_entity.feature_class_configuration and \
(isinstance(db_entity.feature_class_configuration, dict) or
db_entity.feature_class_configuration.generated),
config_entity.computed_db_entities())
layers_to_remove = Layer.objects.filter(layer_libraries__config_entity__in=[config_entity], db_entity_interest__db_entity__key__in=map(lambda db_entity: db_entity.key, db_entities))
for layer in layers_to_remove:
# Drop the layer_selection classes
layer_selection_class = get_or_create_layer_selection_class_for_layer(layer)
drop_tables_for_dynamic_classes(
layer_selection_class,
layer_selection_class.features.field.rel.through
)
layers_to_remove.delete()
for layer in Layer.objects.all():
try:
layer.db_entity_interest.db_entity
except:
# orphan
try:
# Drop the layer_selection classes
layer_selection_class = get_or_create_layer_selection_class_for_layer(layer)
drop_tables_for_dynamic_classes(
layer_selection_class,
layer_selection_class.features.field.rel.through
)
layer.delete()
except:
pass
# DbEntities
for db_entity in db_entities:
feature_class = None
try:
feature_class = FeatureClassCreator(config_entity, db_entity).dynamic_model_class()
except Exception, e:
logger.warn("No feature class for db_entity %s could be created. Exception: %s" % (db_entity.name, e.message))
DeleteImportProcessor().drop_data(config_entity, db_entity)
db_entity.delete()
if issubclass(cls, Scenario):
cloned_config_entities = cls.objects.filter(origin_instance__isnull=False)
# ConfigEntities and their schemas
if options.get('delete_clones') or options.get('delete_scenario_clones'):
for config_entity in cloned_config_entities:
PGNamespace.objects.drop_schema(config_entity.schema())
for db_entity in config_entity.owned_db_entities():
db_entity.delete()
cloned_config_entities.delete()
if options.get('delete_clones') and False:
for built_form_set in BuiltFormSet.objects.all():
built_form_set.built_forms.remove(*built_form_set.built_forms.filter(origin_instance__isnull=False))
# BuiltForms
BuiltForm.objects.filter(origin_instance__isnull=False).delete()
# Orphaned BuiltForm assets (only an issue when corrupt saves have happened)
BuildingAttributeSet.objects.annotate(
num_buildings=Count('building'), num_buildingtypes=Count('buildingtype'), num_placetypes=Count('building')).filter(
num_buildings=0, num_buildingtypes=0, num_placetypes=0).delete()
Medium.objects.annotate(num_built_form_sets=Count('builtform')).filter(num_built_form_sets=0, key__startswith='built_form').delete()
BuiltFormExample.objects.annotate(num_built_form_sets=Count('builtform')).filter(num_built_form_sets=0).delete()