def get_by_customer_id(self, customer_id):
all_payments = []
try:
db_config = read_db_config()
conn = MySQLConnection(**db_config)
cursor = conn.cursor()
args = [customer_id]
cursor.callproc('getAllPaymentInfoByCustomerID', args)
for result in cursor.stored_results():
payments = result.fetchall()
for x in payments:
currentpayment = PaymentInfo()
currentpayment.card_id = x[0]
currentpayment.last_four = x[1]
currentpayment.expir_date = x[2]
currentpayment.card_issuer = x[3]
currentpayment.customer_id = x[4]
currentpayment.billing_address.address_id = x[5]
currentpayment.billing_address.street = x[6]
currentpayment.billing_address.city = x[7]
currentpayment.billing_address.state_code = x[8]
currentpayment.billing_address.zip_code = x[9]
currentpayment.billing_address.address_type = x[10]
all_payments.append(currentpayment)
cursor.close()
conn.close()
except Error as error:
print(error)
except Exception as e:
print(e)
return all_payments
def get_by_address_id(
self,
p_address_id,
p_customer_id,
):
all_payments = []
try:
db_config = read_db_config()
conn = MySQLConnection(**db_config)
cursor = conn.cursor()
args = [p_address_id, p_customer_id]
cursor.callproc('getPaymentInfoByAddressID', args)
for result in cursor.stored_results():
payments = result.fetchall()
for x in payments:
currentpayment = PaymentInfo()
currentpayment.card_id = x[0]
currentpayment.last_four = x[2]
currentpayment.expir_date = x[3]
currentpayment.card_issuer = x[4]
currentpayment.customer_id = x[5]
currentpayment.billing_address_id = x[6]
all_payments.append(currentpayment)
cursor.close()
conn.close()
except Error as error:
print(error)
except Exception as e:
print(e)
return all_payments
def get_byid(self, order_id):
try:
# Setup connection to the DB
db_config = read_db_config()
conn = MySQLConnection(**db_config)
cursor = conn.cursor()
order = None
args = [order_id]
cadao = CustomerAddressDao()
pdao = PaymentInfoDao()
# Calls the stored procedure
cursor.callproc('getRetailOrderByOrderID', args)
# This loop iterates through the resultsets
for result in cursor.stored_results():
# This loop iterates through the rows in each resultset
for x in result.fetchall():
order = RetailOrder()
order.order_id = x[0]
order.date_ordered = x[1]
order.discount = x[2]
order.total_price = x[3]
order.status = x[4]
u = User()
u.id = x[5]
u.first_name = x[6]
u.last_name = x[7]
order.customer = u
p = PaymentInfo()
p.card_id = x[8]
p.last_four = x[9]
p.card_issuer = x[10]
order.card = p
a = CustomerAddress()
a.address_id = x[11]
a.street = x[12]
a.city = x[13]
a.state_code = x[14]
a.zip_code = x[15]
order.shipping_address = a
# Close the connection to the DB
cursor.close()
conn.close()
except Error as error:
print(error)
except Exception as e:
print(e)
return order
def post(self,request,address_id):
user_id = request.session['user_id']
username = request.session['username']
eaddress = EditAddressForm(request.POST)
apayment = AddPaymentInfoForm2(request.POST)
daddress = DeleteAddressForm(request.POST)
address = self.cadao.get_byid(address_id)
user_id = address.customer_id
context = {}
if 'edit-address' in request.POST:
if eaddress.is_valid():
a = CustomerAddress()
a.address_id = address_id
a.customer_id = user_id
a.street = eaddress.cleaned_data['street']
a.city = eaddress.cleaned_data['city']
a.state_code = eaddress.cleaned_data['state_code']
a.zip_code = eaddress.cleaned_data['zip_code']
a.address_type = eaddress.cleaned_data['address_type']
self.cadao.update(a)
context['user_id'] = request.session['user_id'],
context['username'] = request.session['username']
return redirect(reverse(('customeraddress'), kwargs={'address_id': address_id}))
elif 'add-card' in request.POST:
if apayment.is_valid():
p = PaymentInfo()
p.card_number = apayment.cleaned_data['card_number']
p.cvc = apayment.cleaned_data['cvc']
p.expir_date = apayment.cleaned_data['expir_date']
p.card_issuer = apayment.cleaned_data['card_issuer']
p.customer_id = user_id
p.billing_address.address_id = address_id
self.padao.create(p)
context['user_id'] = request.session['user_id'],
context['username'] = request.session['username']
return redirect(reverse(('customeraddress'), kwargs={'address_id': address_id}))
elif 'delete-address' in request.POST:
if daddress.is_valid():
a = CustomerAddress()
a.address_id = daddress.cleaned_data['address_id']
a.customer_id = user_id
self.cadao.delete(a)
context['user_id'] = request.session['user_id'],
context['username'] = request.session['username']
return redirect(reverse('customeraccount'))
else:
return redirect(reverse(('customeraddress'), kwargs={'address_id': address_id}))
def post(self,request):
context={}
user_id = request.session['user_id']
username = request.session['username']
bill_addresses = self.cadao.get_by_customer_and_type(user_id, "Billing")
bill_address_choices = []
for address in bill_addresses:
address_val = (str(address.address_id), str(address.street) + " " + str(address.city) + ", "
+ str(address.state_code) + " " + str(address.zip_code))
bill_address_choices.append(address_val)
addcard = AddPaymentInfoForm(request.POST,bill_address_choices=bill_address_choices)
aaddress = AddAddressForm2(request.POST)
if 'add-card' in request.POST:
if addcard.is_valid():
p = PaymentInfo()
p.customer_id = user_id
p.card_number = addcard.cleaned_data['card_number']
p.cvc = addcard.cleaned_data['cvc']
p.expir_date = addcard.cleaned_data['expir_date']
p.card_issuer = addcard.cleaned_data['card_issuer']
p.billing_address.address_id = addcard.cleaned_data['billing_addresses']
self.pdao.create(p)
context['user_id'] = request.session['user_id'],
context['username'] = request.session['username']
return redirect(reverse('customeraccount'))
elif 'add-address' in request.POST:
if aaddress.is_valid():
a = CustomerAddress()
a.customer_id = user_id
a.street = aaddress.cleaned_data['street']
a.city = aaddress.cleaned_data['city']
a.state_code = aaddress.cleaned_data['state_code']
a.zip_code = aaddress.cleaned_data['zip_code']
a.address_type = 'Billing'
self.cadao.create(a)
context['user_id'] = request.session['user_id']
context['username'] = request.session['username']
return redirect(reverse('customeraddcard'))
else:
return redirect(reverse('customeraccount'))