def register_member(self, reg):
random_string = generate_random_string()
while Member.objects.filter(temporary_web_id=random_string).count() > 0:
# Preventing duplication errors
random_string = generate_random_string()
# Make the 'added' date timezone-aware.
added_date = datetime.strptime(reg['date'], '%Y-%m-%d %H:%M:%S')
# Create and configure member
member = Member()
member.added = pytz.timezone(settings.TIME_ZONE).localize(added_date)
member.ssn = reg['ssn']
member.name = reg['name']
member.email = reg['email']
merge_national_registry_info(member, reg['national'], timezone.now())
member.temporary_web_id = random_string
member.temporary_web_id_timing = timezone.now()
# Save member to database
stdout.write('* Registering member...')
stdout.flush()
member.save()
stdout.write(' done\n')
self.process_groups(reg, member)
# Send confirmation message
message = InteractiveMessage.objects.get(interactive_type='registration_received', active=True)
body = message.produce_links(member.temporary_web_id)
# Save the start of the delivery attempt
delivery = InteractiveMessageDelivery()
delivery.interactive_message = message
delivery.member = member
delivery.email = member.email
delivery.timing_start = timezone.now()
delivery.save()
# Actually send the message
stdout.write('* Sending confirmation email...')
stdout.flush()
quick_mail(member.email, message.subject, body)
stdout.write(' done\n')
# Update the delivery
delivery.timing_end = timezone.now()
delivery.save()
def handle(self, *args, **options):
# These arguments let us override the Django settings from
# the CLI - necessary while we don't want to enable the new
# gateway completely.
for arg in ('url', 'password', 'username', 'xml_namespace'):
if options.get(arg):
settings.NATIONAL_REGISTRY[arg] = options.get(arg)[0]
local_update_db = {}
if options.get('import-db-state'):
for fn in options['import-db-state']:
with open(fn, 'r') as fd:
local_update_db.update(json.load(fd))
# Each of the selection arguments creates a set of users...
# These are the users that lack a legal municipality code.
user_sets = []
if options.get('all', False):
user_sets.append(Member.objects.all())
if options.get('lack-muni', False):
user_sets.append(
set(Member.objects.filter(legal_municipality_code=None)) |
set(Member.objects.filter(legal_municipality_code='')))
# These are users whose SSN we have specified manually.
ssn = []
for ssns in (options.get('ssn') or []):
ssn.extend([s.strip().replace('-', '') for s in ssns.split(',')])
if ssn:
user_sets.append(
set(Member.objects.filter(ssn__in=ssn)))
# These are users in a particular location
locs = []
for loc in (options.get('loc-code') or []):
locs.extend([lc.strip() for lc in loc.split(',')])
if locs:
users = []
for lc in locs:
users.extend(LocationCode.objects.get(
location_code=lc).get_members())
user_sets.append(set(users))
# These are users whose last legal-lookup happened before a deadline.
# An example of usage: --refresh=$(date +%s '6 months ago')
if options.get('refresh'):
deadline_set = set()
for deadline in options.get('refresh'):
when = datetime.datetime.utcfromtimestamp(int(deadline))
when = when.replace(tzinfo=pytz.utc)
deadline_set |= (
set(Member.objects.filter(legal_lookup_timing__lt=when)) |
set(Member.objects.filter(legal_lookup_timing__isnull=True)))
user_sets.append(deadline_set)
# The specified user sets are combined, either by intersecting them
# (constraint AND constraint AND ...) or combining (OR).
users = user_sets.pop(0)
for uset in user_sets:
if options.get('intersect', False):
users &= uset
else:
users |= uset
if not users:
sys.stderr.write('No users, nothing to do.\n')
else:
sys.stderr.write('Processing %d users...\n' % len(users))
users = list(users)
if options.get('shuffle'):
random.shuffle(users)
else:
users.sort(key=lambda u: u.ssn)
if options.get('dump-ssns'):
print '\n'.join(user.ssn for user in users)
dump = {}
if (options.get('update-db')
or options.get('update-as-json')
or options.get('export-db-state')):
for user in users:
if options.get('export-db-state'):
nr_info = self.member_nr_info(user)
else:
if options.get('import-db-state'):
nr_info = local_update_db.get(user.ssn, {})
else:
nr_info = lookup_national_registry(user.ssn)
if not nr_info:
sys.stderr.write('Lookup failed for %s\n' % user.ssn)
if nr_info and options.get('update-db'):
sys.stderr.write('Updating %s in DB\n' % user.ssn)
now = timezone.now()
try:
merge_national_registry_info(user, nr_info, now)
user.save()
except AssertionError:
pass # Just keen on truckin'
if nr_info and (
options.get('update-as-json') or
options.get('export-db-state')):
dump[nr_info['ssn']] = nr_info
if dump:
json.dump(dump, sys.stdout, indent=1)
