def check_national_registry(self, reg):
stdout.write('- Checking if %s is an individual in the national registry...' % reg['ssn'])
stdout.flush()
national = lookup_national_registry(reg['ssn'])
if national['is_valid'] and national['is_individual']:
reg['national'] = national
stdout.write(' yes\n')
return True
else:
stdout.write(' no\n')
return False
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)