def delete_customer(customer_id):
'''delete a customer by id.'''
try:
with DB.atomic():
delete_query = Customer.delete().where(Customer.customer_id == customer_id)
delete_query.execute()
except IntegrityError:
LOGGER.warning("Customer ID %s not found.", customer_id)
def delete_customer(delete_name):
'''delete a customer by id.'''
try:
with DB.atomic():
Customer.delete().where(Customer.name == delete_name).execute()
result = f'Succesfully deleted {delete_name}'
except IntegrityError:
result = f'{delete_name} does not exist'
return result
def update_customer_credit(customer_id, credit_limit):
"""Update an existing customer's credit limit"""
try:
with DB.atomic():
customer = Customer.get(Customer.customer_id == customer_id)
customer.credit_limit = credit_limit
customer.save()
except pw.DoesNotExist:
LOGGER.error("Failed to update ID '%s': not found in database",
customer_id)
raise ValueError
def add_customer(name, lastname, home_address, phone_number, email_address,
status, poverty_score):
'''atomic() is easier than transaction()'''
try:
with DB.atomic():
Customer.create(name=name,
lastname=lastname,
home_address=home_address,
phone_number=phone_number,
email_address=email_address,
status=status,
credit_limit=poverty_score)
except IntegrityError:
LOGGER.warning("Name %s is already taken.", name)
def add_customer(customer_id, name, lastname, home_address, phone_number,
email_address, status, credit_limit):
'''add new customer function.'''
try:
with DB.atomic():
Customer.create(customer_id=customer_id,
name=name,
lastname=lastname,
home_address=home_address,
phone_number=phone_number,
email_address=email_address,
status=status,
credit_limit=credit_limit)
LOGGER.info("Adding customer %s to database.", customer_id)
except IntegrityError:
LOGGER.warning("Customer ID %s is already taken.", customer_id)
def add_customer(customer_id, name, lastname, home_address, phone_number,
email_address, status, credit_limit):
#pylint: disable=too-many-arguments
'''
This function adds a customer and its data to the customer database
'''
logging.debug('Attempting to add customer %s to database', customer_id)
with DB.atomic():
try:
added_customer = Customer.create(customer_id=customer_id,
name=name,
lastname=lastname,
home_address=home_address,
phone_number=phone_number,
email_address=email_address,
status=status,
credit_limit=credit_limit)
added_customer.save()
logging.debug('Successfully added customer %s to database',
customer_id)
except p.DatabaseError:
logging.error('Could not add customer %s to database', customer_id)