def handle(self, **options):
from tendenci.addons.corporate_memberships.models import (
CorpProfile, CorpMembership, CorpMembershipRep,
CorpMembershipAuthDomain, CorporateMembership)
from tendenci.addons.corporate_memberships.utils import corp_membership_update_perms
from tendenci.addons.memberships.models import MembershipDefault
verbosity = int(options['verbosity'])
corp_profile_field_names = [smart_str(field.name) for field \
in CorpProfile._meta.fields \
if not field.__class__ == AutoField]
corp_membership_field_names = [smart_str(field.name) for field \
in CorpMembership._meta.fields \
if not field.__class__ == AutoField]
corp_profile_fields = dict([(field.name, field) \
for field in CorpProfile._meta.fields \
if field.get_internal_type() != 'AutoField' and \
field.name not in ['guid']])
corp_membership_fields = dict([(field.name, field) \
for field in CorpMembership._meta.fields \
if field.get_internal_type() != 'AutoField' and \
field.name not in ['user', 'guid',
'corp_profile']])
def get_default_value(field):
# if allows null or has default, return None
if field.null or field.has_default():
return None
field_type = field.get_internal_type()
if field_type == 'BooleanField':
return False
if field_type == 'DateField':
return date
if field_type == 'DateTimeField':
return datetime.now()
if field_type == 'DecimalField':
return Decimal(0)
if field_type == 'IntegerField':
return 0
if field_type == 'FloatField':
return 0
if field_type == 'ForeignKey':
if not field.name in ['creator', 'owner']:
[
value
] = field.related.parent_model.objects.all()[:1] or [None]
return value
return None
return ''
corporates = CorporateMembership.objects.all()
for corporate in corporates:
# check if corp_profile already exists
[corp_profile
] = CorpProfile.objects.filter(name=corporate.name)[:1] or [None]
if not corp_profile:
corp_profile = CorpProfile()
for field_name in corp_profile_field_names:
if hasattr(corporate, field_name):
value = getattr(corporate, field_name)
else:
value = None
if value is None and field_name in corp_profile_fields:
value = get_default_value(
corp_profile_fields[field_name])
if not value is None:
setattr(corp_profile, field_name, value)
corp_profile.save()
if verbosity >= 2:
print 'Insert corp_profile: ', corp_profile
corp_membership = None
else:
# check if corp_membership exists
[corp_membership
] = corp_profile.corp_memberships.all().exclude(
status_detail='archive').order_by(
'-expiration_dt')[:1] or [None]
if not corp_membership:
corp_membership = CorpMembership()
for field_name in corp_membership_field_names:
if hasattr(corporate, field_name):
value = getattr(corporate, field_name)
else:
value = None
if value is None and field_name in corp_membership_fields:
value = get_default_value(
corp_membership_fields[field_name])
if not value is None:
setattr(corp_membership, field_name, value)
corp_membership.corp_profile = corp_profile
corp_membership.save()
# update invoice object_type and object_id
if corp_membership.invoice:
corp_membership.invoice.object_type = ContentType.objects.get(
app_label=corp_membership._meta.app_label,
model=corp_membership._meta.module_name)
corp_membership.invoice.object_id = corp_membership.id
corp_membership.invoice.save()
if verbosity >= 2:
print 'Insert corp_membership (id=%d) for: ' % corp_membership.id, corp_profile
# dues reps
reps = corporate.reps.all()
if reps:
for rep in reps:
if not corp_profile.reps.filter(user=rep.user).exists():
CorpMembershipRep.objects.create(
corp_profile=corp_profile,
user=rep.user,
is_dues_rep=rep.is_dues_rep,
is_member_rep=rep.is_member_rep)
corp_membership_update_perms(corp_membership)
# auth domains
auth_domains = corporate.auth_domains.all()
if auth_domains:
for auth_domain in auth_domains:
if not corp_profile.authorized_domains.filter(
name=auth_domain.name).exists():
CorpMembershipAuthDomain.objects.create(
corp_profile=corp_profile, name=auth_domain.name)
# update individual membership entries
memberships = MembershipDefault.objects.filter(
corporate_membership_id=corporate.pk, corp_profile_id=0)
for membership in memberships:
membership.corp_profile_id = corp_profile.pk
membership.corporate_membership_id = corp_membership.pk
membership.save()
if verbosity >= 2:
print 'Updated membership (id=%d):' % membership.id, membership
def handle(self, **options):
from tendenci.addons.corporate_memberships.models import (
CorpProfile,
CorpMembership,
CorporateMembership)
from tendenci.addons.memberships.models import MembershipDefault
verbosity = int(options['verbosity'])
corp_profile_field_names = [smart_str(field.name) for field \
in CorpProfile._meta.fields \
if not field.__class__ == AutoField]
corp_membership_field_names = [smart_str(field.name) for field \
in CorpMembership._meta.fields \
if not field.__class__ == AutoField]
corp_profile_fields = dict([(field.name, field) \
for field in CorpProfile._meta.fields \
if field.get_internal_type() != 'AutoField' and \
field.name not in ['guid']])
corp_membership_fields = dict([(field.name, field) \
for field in CorpMembership._meta.fields \
if field.get_internal_type() != 'AutoField' and \
field.name not in ['user', 'guid',
'corp_profile']])
def get_default_value(field):
# if allows null or has default, return None
if field.null or field.has_default():
return None
field_type = field.get_internal_type()
if field_type == 'BooleanField':
return False
if field_type == 'DateField':
return date
if field_type == 'DateTimeField':
return datetime.now()
if field_type == 'DecimalField':
return Decimal(0)
if field_type == 'IntegerField':
return 0
if field_type == 'FloatField':
return 0
if field_type == 'ForeignKey':
if not field.name in ['creator', 'owner']:
[value] = field.related.parent_model.objects.all(
)[:1] or [None]
return value
return None
return ''
corporates = CorporateMembership.objects.all()
for corporate in corporates:
# check if corp_profile already exists
[corp_profile] = CorpProfile.objects.filter(name=corporate.name)[:1] or [None]
if not corp_profile:
corp_profile = CorpProfile()
for field_name in corp_profile_field_names:
if hasattr(corporate, field_name):
value = getattr(corporate, field_name)
else:
value = None
if value is None and field_name in corp_profile_fields:
value = get_default_value(corp_profile_fields[field_name])
if not value is None:
setattr(corp_profile, field_name,
value)
corp_profile.save()
if verbosity >= 2:
print 'Insert corp_profile: ', corp_profile
corp_membership = None
else:
# check if corp_membership exists
corp_membership = corp_profile.corp_membership
if not corp_membership:
corp_membership = CorpMembership()
for field_name in corp_membership_field_names:
if hasattr(corporate, field_name):
value = getattr(corporate, field_name)
else:
value = None
if value is None and field_name in corp_membership_fields:
value = get_default_value(corp_membership_fields[field_name])
if not value is None:
setattr(corp_membership, field_name,
value)
corp_membership.corp_profile = corp_profile
corp_membership.save()
# update invoice object_type and object_id
if corp_membership.invoice:
corp_membership.invoice.object_type = ContentType.objects.get(
app_label=corp_membership._meta.app_label,
model=corp_membership._meta.module_name)
corp_membership.invoice.object_id = corp_membership.id
corp_membership.invoice.save()
if verbosity >= 2:
print 'Insert corp_membership (id=%d) for: ' % corp_membership.id, corp_profile
# update individual membership entries
memberships = MembershipDefault.objects.filter(
corporate_membership_id=corporate.pk,
corp_profile_id=0
)
for membership in memberships:
membership.corp_profile_id = corp_profile.pk
membership.corporate_membership_id = corp_membership.pk
membership.save()
if verbosity >= 2:
print 'Updated membership (id=%d):' % membership.id, membership
def do_import_corp_membership(self, corp_profile, corp_memb, action_info):
"""
Database import here - insert or update
"""
# handle corp_profile
if not corp_profile:
corp_profile = CorpProfile()
self.assign_import_values_from_dict(corp_profile, action_info['corp_profile_action'])
if corp_profile.status == None or corp_profile.status == '' or \
self.cmemb_data.get('status', '') == '':
corp_profile.status = True
if not corp_profile.status_detail:
corp_profile.status_detail = 'active'
else:
corp_profile.status_detail = corp_profile.status_detail.lower()
if not corp_profile.creator:
corp_profile.creator = self.request_user
corp_profile.creator_username = self.request_user.username
if not corp_profile.owner:
corp_profile.owner = self.request_user
corp_profile.owner_username = self.request_user.username
corp_profile.save()
# corpmembership
if not corp_memb:
corp_memb = CorpMembership(
corp_profile=corp_profile,
creator=self.request_user,
creator_username=self.request_user.username,
owner=self.request_user,
owner_username=self.request_user.username,
)
self.assign_import_values_from_dict(corp_memb, action_info['corp_memb_action'])
if corp_memb.status == None or corp_memb.status == '' or \
self.cmemb_data.get('status', '') == '':
corp_memb.status = True
if not corp_memb.status_detail:
corp_memb.status_detail = 'active'
else:
corp_memb.status_detail = corp_memb.status_detail.lower()
# set to approved for active memberships
if not corp_memb.approved:
if corp_memb.status and corp_memb.status_detail == 'active':
corp_memb.approved = True
# corporate membership type
if not hasattr(corp_memb, "corporate_membership_type") or \
not corp_memb.corporate_membership_type:
# last resort - pick the first available membership type
corp_memb.corporate_membership_type = CorporateMembershipType.objects.all(
).order_by('id')[0]
# no join_dt - set one
if not hasattr(corp_memb, 'join_dt') or not corp_memb.join_dt:
if corp_memb.status and corp_memb.status_detail == 'active':
corp_memb.join_dt = datetime.now()
# no expire_dt - get it via corporate_membership_type
if not hasattr(corp_memb, 'expiration_dt') or not corp_memb.expiration_dt:
if corp_memb.corporate_membership_type:
expiration_dt = corp_memb.corporate_membership_type.get_expiration_dt(
join_dt=corp_memb.join_dt)
setattr(corp_memb, 'expiration_dt', expiration_dt)
if not corp_memb.creator:
corp_memb.creator = self.request_user
corp_memb.creator_username = self.request_user.username
if not corp_memb.owner:
corp_memb.owner = self.request_user
corp_memb.owner_username = self.request_user.username
corp_memb.save()
# bind members to their corporations by company names
if self.mimport.bind_members:
self.bind_members_to_corp_membership(corp_memb)
def do_import_corp_membership(self, corp_profile, corp_memb, action_info):
"""
Database import here - insert or update
"""
# handle corp_profile
if not corp_profile:
corp_profile = CorpProfile()
self.assign_import_values_from_dict(corp_profile,
action_info['corp_profile_action'])
if corp_profile.status == None or corp_profile.status == '' or \
self.cmemb_data.get('status', '') == '':
corp_profile.status = True
if not corp_profile.status_detail:
corp_profile.status_detail = 'active'
else:
corp_profile.status_detail = corp_profile.status_detail.lower()
if not corp_profile.creator:
corp_profile.creator = self.request_user
corp_profile.creator_username = self.request_user.username
if not corp_profile.owner:
corp_profile.owner = self.request_user
corp_profile.owner_username = self.request_user.username
corp_profile.save()
# corpmembership
if not corp_memb:
corp_memb = CorpMembership(
corp_profile=corp_profile,
creator=self.request_user,
creator_username=self.request_user.username,
owner=self.request_user,
owner_username=self.request_user.username,
)
self.assign_import_values_from_dict(corp_memb,
action_info['corp_memb_action'])
if corp_memb.status == None or corp_memb.status == '' or \
self.cmemb_data.get('status', '') == '':
corp_memb.status = True
if not corp_memb.status_detail:
corp_memb.status_detail = 'active'
else:
corp_memb.status_detail = corp_memb.status_detail.lower()
# set to approved for active memberships
if not corp_memb.approved:
if corp_memb.status and corp_memb.status_detail == 'active':
corp_memb.approved = True
# corporate membership type
if not hasattr(corp_memb, "corporate_membership_type") or \
not corp_memb.corporate_membership_type:
# last resort - pick the first available membership type
corp_memb.corporate_membership_type = CorporateMembershipType.objects.all(
).order_by('id')[0]
# no join_dt - set one
if not hasattr(corp_memb, 'join_dt') or not corp_memb.join_dt:
if corp_memb.status and corp_memb.status_detail == 'active':
corp_memb.join_dt = datetime.now()
# no expire_dt - get it via corporate_membership_type
if not hasattr(corp_memb,
'expiration_dt') or not corp_memb.expiration_dt:
if corp_memb.corporate_membership_type:
expiration_dt = corp_memb.corporate_membership_type.get_expiration_dt(
join_dt=corp_memb.join_dt)
setattr(corp_memb, 'expiration_dt', expiration_dt)
if not corp_memb.creator:
corp_memb.creator = self.request_user
corp_memb.creator_username = self.request_user.username
if not corp_memb.owner:
corp_memb.owner = self.request_user
corp_memb.owner_username = self.request_user.username
corp_memb.save()
# bind members to their corporations by company names
if self.mimport.bind_members:
self.bind_members_to_corp_membership(corp_memb)