from django.contrib import admin from bluebottle.utils.model_dispatcher import get_organization_model, get_organizationdocument_model, get_organizationmember_model from .forms import OrganizationDocumentForm ORGANIZATION_MODEL = get_organization_model() MEMBER_MODEL = get_organizationmember_model() DOCUMENT_MODEL = get_organizationdocument_model() class OrganizationDocumentInline(admin.StackedInline): model = DOCUMENT_MODEL form = OrganizationDocumentForm extra = 0 raw_id_fields = ('author', ) readonly_fields = ('download_url',) fields = readonly_fields + ('file', 'author') def download_url(self, obj): return "<a href='{0}'>{1}</a>".format(str(obj.document_url), 'Download') download_url.allow_tags = True class OrganizationMemberInline(admin.StackedInline): model = MEMBER_MODEL raw_id_fields = ('user', ) extra = 0 class OrganizationAdmin(admin.ModelAdmin): prepopulated_fields = {'slug': ('name',)}
import factory from bluebottle.utils.model_dispatcher import get_organization_model, \ get_organizationmember_model from .geo import CountryFactory from .accounts import BlueBottleUserFactory ORGANIZATION_MODEL = get_organization_model() ORGANIZATION_MEMBER_MODEL = get_organizationmember_model() class OrganizationFactory(factory.DjangoModelFactory): FACTORY_FOR = ORGANIZATION_MODEL name = factory.Sequence(lambda n: 'Organization_{0}'.format(n)) slug = factory.Sequence(lambda n: 'organization_{0}'.format(n)) address_line1 = "'s Gravenhekje 1a" address_line2 = '1011 TG' city = 'Amsterdam' state = 'North Holland' country = factory.SubFactory(CountryFactory, name='Netherlands') postal_code = '1011TG' # Contact phone_number = '(+31) 20 715 8980' website = 'http://onepercentclub.com' email = '*****@*****.**' class OrganizationMemberFactory(factory.DjangoModelFactory):