reader = csv.reader(csvfile)
for row in reader:
_, ticket_types, first_name, last_name, orig_pid, email, status = row
email = email.lower()
if not status == "Redo för import":
continue
pid = format_pid(orig_pid)
if try_get_person(pid=pid):
error('Person exists')
continue
person = try_create_manually(first_name, last_name, pid, email)
if not person:
error('Could not create person')
continue
person.save()
print ' '.join((first_name, last_name))
invoice = Invoice(person=person, is_verified=True)
invoice.generate_data()
invoice.save()
create_tickets(invoice, ticket_types)
invoice.send_as_email()
def main():
edited = []
manual_visitors = 0
liu_visitors = 0
failed = 0
invalid_pid = 0
with open('people.csv', 'rb') as csvfile:
def error(msg):
print
print msg
print name
print ', '.join(row)
print '-------------------------------'
reader = csv.reader(csvfile)
for row in reader:
_, _, liu_id, name, orig_pid, email, reason, ticket_types, _ = row
name = unicode(name, 'utf-8')
name_split = name.split()
fname, lname = name_split[0], ' '.join(name_split[1:])
liu_id = liu_id.lower().strip()
pid = format_pid(orig_pid)
if '@' in email and '.' in email:
email = email.lower()
else:
email = None
# if not reason:
# error('No reason given')
# no_reason += 1
# continue
person = try_get_person(pid=pid, liu_id=liu_id)
if not person:
visitor = try_get_from_kobra(pid=pid, liu_id=liu_id)
if visitor:
person = try_get_person(pid=visitor.pid,
liu_id=visitor.liu_id)
if person:
visitor = None
if email:
if visitor:
visitor.email = email.lower()
elif person:
person.email = email.lower()
liu_visitors += 1
else:
visitor = try_create_manually(fname, lname, pid, email)
if visitor:
manual_visitors += 1
if not (visitor or person):
if orig_pid and not pid:
error('Invalid PID')
invalid_pid += 1
continue
error('Could not find or create user')
failed += 1
continue
if visitor:
visitor.save()
person = visitor.person_ptr
elif person:
person.save()
if Invoice.objects.filter(person=person).exists():
ticket_types = ticket_types.replace(' (tors, fre, lör)', '')
if [TYPES[tt] for tt in ticket_types.split(', ')] != [1]:
edited.append(person)
continue
invoice = Invoice(person=person, is_verified=True)
invoice.generate_data()
invoice.save()
create_tickets(invoice, ticket_types)
invoice.send_as_email()
print '\n========================================================'
print '%d manual visitors' % manual_visitors
print '%d liu visitors' % liu_visitors
print '%d failed' % failed
print '%d invalid PID' % invalid_pid
print '\n========================================================'
print '%d edited ticket types' % len(edited)
print
for p in edited:
print unicode(p)
print p.liu_id or p.pid
print
