Browse Source

Rename function.

feature/python-error-tracking
Drew Larson 8 years ago
parent
commit
c993de85bd
  1. 10
      bikeshop_project/bikeshop/utils/member_import.py

10
bikeshop_project/bikeshop/utils/member_import.py

@ -23,10 +23,10 @@ def email_generator():
return '{0}.{1}@example.com'.format(*local) return '{0}.{1}@example.com'.format(*local)
def payment_type(pt): def get_payment_type(pt):
types = Payment.payment_choices payment_types = Payment.payment_choices
try: try:
return [type for type in types if type[1].lower() == pt.lower()][0] return [payment_type for payment_type in payment_types if payment_type[1].lower() == pt.lower()][0]
except IndexError: except IndexError:
return 'UNKNOWN', 'Unknown' return 'UNKNOWN', 'Unknown'
@ -70,10 +70,10 @@ def member_import():
) )
payment = Payment.objects.create( payment = Payment.objects.create(
type=payment_type(row.get('payment'))[0], type=get_payment_type(row.get('payment'))[0],
) )
membership = Membership.objects.create( Membership.objects.create(
renewed_at=renewed_at, renewed_at=renewed_at,
self_identification=row.get('self_identification', None), self_identification=row.get('self_identification', None),
gender=row.get('gender', None), gender=row.get('gender', None),

Loading…
Cancel
Save