def seed_relationship_types(apps, schema_editor):
    """Seed the relationship types."""
    MilestoneRelationshipType = apps.get_model("milestones", "MilestoneRelationshipType")
    for name in fetch_milestone_relationship_types().values():
        MilestoneRelationshipType.objects.get_or_create(
            name=name,
            description='Autogenerated milestone relationship type "{}"'.format(name),
        )
def delete_relationship_types(apps, schema_editor):
    """Clean up any relationships we made."""
    MilestoneRelationshipType = apps.get_model("milestones",
                                               "MilestoneRelationshipType")
    db_alias = schema_editor.connection.alias
    for name in fetch_milestone_relationship_types().values():
        MilestoneRelationshipType.objects.using(db_alias).filter(
            name=name).delete()
示例#3
0
def seed_relationship_types(apps, schema_editor):
    """Seed the relationship types."""
    MilestoneRelationshipType = apps.get_model("milestones",
                                               "MilestoneRelationshipType")
    for name in fetch_milestone_relationship_types().values():
        MilestoneRelationshipType.objects.get_or_create(
            name=name,
            description='Autogenerated milestone relationship type "{}"'.
            format(name),
        )
 def forwards(self, orm):
     """
     Adds database entries for milestone relationship types defined in data.py
     Performs existence checks before adding in order to avoid integrity errors
     """
     for name in fetch_milestone_relationship_types().values():
         try:
             orm.MilestoneRelationshipType.objects.get(name=name)
         except orm.MilestoneRelationshipType.DoesNotExist:
             orm.MilestoneRelationshipType.objects.create(
                 name=name,
                 description='Autogenerated milestone relationship type "{}"'.format(name),
             )
def delete_relationship_types(apps, schema_editor):
    """Clean up any relationships we made."""
    MilestoneRelationshipType = apps.get_model("milestones", "MilestoneRelationshipType")
    db_alias = schema_editor.connection.alias
    for name in fetch_milestone_relationship_types().values():
        MilestoneRelationshipType.objects.using(db_alias).filter(name=name).delete()