python_assignment_peewee.py

import peewee

database = peewee.SqliteDatabase('customers.db')
database.connect()
database.execute_sql('PRAGMA foreign_keys = ON;')
database.execute_sql('drop table if exists customer;')

class BaseModel(peewee.Model):
    class Meta:
        database = database

class Customer(BaseModel):
    """
        This class defines Customer, which maintains details of someone
        for whom we want to research consumer habits to date.

        Customer ID
        Name
        Lastname
        Home address
        Phone number
        Email address
        Status (active or inactive customer)
        Credit limit
    """

    customer_id = peewee.CharField(primary_key = True, max_length=30)
    name = peewee.CharField(max_length=50)
    lastname = peewee.CharField(max_length=50, null=True)
    homeaddress = peewee.CharField(max_length=75)
    phone_number = peewee.CharField(max_length=15)
    email = peewee.CharField(max_length=320) # based on max email address length search
    status = peewee.CharField(max_length=10) # "inactive" or "active"
    credit_limit = peewee.FloatField()


Customer.create_table()

def add_customer(customer_id, name, lastname, home_address, phone_number, email_address, status, credit_limit): 
    """This function will add a new customer to the sqlite3 database."""
    try:
        with database.transaction():
            customer = Customer.create(
                customer_id = customer_id,
                name = name,
                lastname = lastname,
                homeaddress = home_address,
                phone_number = phone_number,
                email = email_address,
                status = status,
                credit_limit = credit_limit,
            )
            customer.save()
            print('Database add successful')

    except Exception as e:
        print(f'jError creating = {customer_id}')
        print(e)
        print('See how the database protects our data')


def search_customer(customer_id): 
    """This function will return a dictionary object with name, lastname, email address and phone number of a customer or an empty dictionary object if no customer was found."""
    try:
        customer = Customer.get(Customer.customer_id == customer_id)
        return {
            "name": customer.name,
            "lastname": customer.lastname,
            "email": customer.email,
            "phone_number": customer.phone_number,
        }
    except Exception as e:
        print(f"Error retrieving customer {customer_id}")
        print(e)
        return {}



def delete_customer(customer_id): 
    """This function will delete a customer from the sqlite3 database."""
    try:
        with database.transaction():
            customer = Customer.get(Customer.customer_id == customer_id)
            customer.delete_instance()
            customer.save()
            return True
    except Exception as e:
        print("transaction failed while deleting customer")
        print(e)
        return False


def update_customer_credit(customer_id, credit_limit): 
    """This function will search an existing customer by customer_id and update their credit limit or raise a ValueError exception if the customer does not exist."""
    try:
        with database.transaction():
            customer = Customer.get(Customer.customer_id == customer_id)
            customer.credit_limit = credit_limit
            customer.save()
    except Exception as e:
        raise ValueError("NoCustomer") # required to pass test
        print("transaction failed while updating customer ({customer_id}) credit limit")
        print(e)


def list_active_customers(): 
    """This function will return an integer with the number of customers whose status is currently active."""
    try:
        active_customers = Customer.select().where(Customer.status == "active")# .count()
        for customer in active_customers:
            print(f"{customer.name} {customer.lastname}'s status is {customer.status}")
        return len(active_customers)
    except Exception as e:
        print("Retrieval of active customers failed")
        print(e)


# if __name__ == "__main__":
# data = []
# (add_customer(c) for c in data)

test_basic_ops.py

"""
    Autograde Lesson 3 assignment
    Run pytest
    Run cobverage and linitng using standard batch file
    Student should submit an empty database

"""

import pytest

# import basic_ops as l
import basic_operations as l

@pytest.fixture
def _add_customers():
    return [
        ("1","Tybalt","Heinsh","8 Lien Trail","876935123","theinsh0@unicef.org","active","2420"),
        ("2","Gwennie","Tennock","4608 Steensland Lane","134552899","gtennock1@time.com","active","2343"),
        ("3","Euell","Fidgeon","8 Lerdahl Park","009363240","efidgeon2@yandex.ru","inactive","1077"),
        ("4","Cairistiona","Ginnell","3 Harbort Drive","880717312","cginnell3@wisc.edu","active","1484"),
        ("5","Mariel","Wolfarth","47 Loomis Hill","069169873","mwolfarth4@a8.net","inactive","113"),
        ("6","Mallorie","Sammut","3614 Fisk Avenue","828591715","msammut5@mayoclinic.com","active","3255"),
        ("7","Seamus","Pittford","2136 Rigney Park","807689452","spittford6@nbcnews.com","active","3468"),
        ("8","Piotr","Gauson","4145 Kensington Park","133398937","pgauson7@europa.eu","inactive","3625"),
        ("9","Teodora","Loxton","40829 Bartillon Crossing","194346240","tloxton8@tripod.com","inactive","444"),
        ("10","Nanine","Eadmeades","8691 Calypso Circle","367839372","neadmeades9@addthis.com","inactive","2459"),
        ("11","Marieann","Calcutt","5678 Rutledge Point","245977057","mcalcutta@blogtalkradio.com","inactive","2962"),
        ("12","Wandis","Dewdney","095 Packers Park","135126295","wdewdneyb@phpbb.com","active","4742"),
        ("13","Tommy","Wannes","59889 Myrtle Avenue","969027317","twannesc@4shared.com","inactive","4067"),
        ("14","Letti","Barkworth","7 Sundown Place","243310785","lbarkworthd@mozilla.com","inactive","1697"),
        ("15","Danit","Fayerman","9107 Oakridge Point","596550663","dfayermane@ustream.tv","active","2382"),
        ("16","Susanna","Grigaut","87 Cascade Avenue","839048392","sgrigautf@goo.gl","inactive","3982"),
        ("17","Donal","Thraves","2048 Porter Place","303563873","dthravesg@hud.gov","inactive","3597"),
        ("18","Erina","Whightman","0 Columbus Park","380312722","ewhightmanh@bloglines.com","active","2007"),
        ("19","Rogers","Dumbare","2 Shopko Hill","510504125","rdumbarei@google.fr","active","371"),
        ("20","Kayla","Coyte","4134 Atwood Pass","468104655","kcoytej@paginegialle.it","active","4400"),
    ]

@pytest.fixture
def _search_customers(): # needs to del with database
    return [
        [
            ("15","Danit","Fayerman","9107 Oakridge Point","596550663","dfayermane@ustream.tv","active","2382"),
            ("16","Susanna","Grigaut","87 Cascade Avenue","839048392","sgrigautf@goo.gl","inactive","3982"),
            ("17","Donal","Thraves","2048 Porter Place","303563873","dthravesg@hud.gov","inactive","3597"),
            ("18","Erina","Whightman","0 Columbus Park","380312722","ewhightmanh@bloglines.com","active","2007"),
            ("19","Rogers","Dumbare","2 Shopko Hill","510504125","rdumbarei@google.fr","active","371"),
        ],
        ("16", "000")
    ]


@pytest.fixture
def _delete_customers(): # needs to del with database
    return [
        ("3","Euell","Fidgeon","8 Lerdahl Park","009363240","efidgeon2@yandex.ru","inactive","1077"),
        ("4","Cairistiona","Ginnell","3 Harbort Drive","880717312","cginnell3@wisc.edu","active","1484"),
        ("5","Mariel","Wolfarth","47 Loomis Hill","069169873","mwolfarth4@a8.net","inactive","113"),
    ]

@pytest.fixture
def _update_customer_credit(): # needs to del with database
    return [
        ("8","Piotr","Gauson","4145 Kensington Park","133398937","pgauson7@europa.eu","inactive","3625"),
        ("9","Teodora","Loxton","40829 Bartillon Crossing","194346240","tloxton8@tripod.com","inactive","444"),
        ("10","Nanine","Eadmeades","8691 Calypso Circle","367839372","neadmeades9@addthis.com","inactive","2459"),
    ]

@pytest.fixture
def _list_active_customers():
    return [
        ("1","Tybalt","Heinsh","8 Lien Trail","876935123","theinsh0@unicef.org","active","2420"),
        ("2","Gwennie","Tennock","4608 Steensland Lane","134552899","gtennock1@time.com","active","2343"),
        ("4","Cairistiona","Ginnell","3 Harbort Drive","880717312","cginnell3@wisc.edu","active","1484"),
        ("6","Mallorie","Sammut","3614 Fisk Avenue","828591715","msammut5@mayoclinic.com","active","3255"),
        ("7","Seamus","Pittford","2136 Rigney Park","807689452","spittford6@nbcnews.com","active","3468"),
        ("12","Wandis","Dewdney","095 Packers Park","135126295","wdewdneyb@phpbb.com","active","4742"),
        ("15","Danit","Fayerman","9107 Oakridge Point","596550663","dfayermane@ustream.tv","active","2382"),
    ]

def test_list_active_customers(_list_active_customers):
    """ actives """
    for customer in _list_active_customers:
        l.add_customer(
            customer[0],
            customer[1],
            customer[2],
            customer[3],
            customer[4],
            customer[5],
            customer[6],
            customer[7],
        )

    actives = l.list_active_customers()

    assert actives == 7

     



def test_add_customer(_add_customers):
    """ additions """
    print("type(_add_customers) == ", type(_add_customers))
    
    for customer in _add_customers:
        l.add_customer(
            customer[0],
            customer[1],
            customer[2],
            customer[3],
            customer[4],
            customer[5],
            customer[6],
            customer[7],
        )
        added = l.search_customer(customer[0])
        assert added["name"] == customer[1]
        assert added["lastname"] == customer[2]
        assert added["email"] == customer[5]
        assert added["phone_number"] == customer[4]

    for customer in _add_customers:
        l.delete_customer(customer[0])



def test_search_customer(_search_customers):
    """ search """
    for customer in _search_customers[0]:
        l.add_customer(
            customer[0],
            customer[1],
            customer[2],
            customer[3],
            customer[4],
            customer[5],
            customer[6],
            customer[7]
        )

    result = l.search_customer(_search_customers[1][1])
    assert result == {}

    result = l.search_customer(_search_customers[1][0])
    assert result["name"] == _search_customers[0][1][1]
    assert result["lastname"] == _search_customers[0][1][2]
    assert result["email"] == _search_customers[0][1][5]
    assert result["phone_number"] == _search_customers[0][1][4]

    for customer in _search_customers[0]:
        l.delete_customer(customer[0])


def test_delete_customer(_delete_customers):
    """ delete """
    for customer in _delete_customers:
        l.add_customer(customer[0],
                       customer[1],
                       customer[2],
                       customer[3],
                       customer[4],
                       customer[5],
                       customer[6],
                       customer[7]
                       )

        response = l.delete_customer(customer[0])
        assert response is True

        deleted = l.search_customer(customer[0])
        assert deleted == {}

def test_update_customer_credit(_update_customer_credit):
    """ update """

    for customer in _update_customer_credit:
        l.add_customer(
            customer[0],
            customer[1],
            customer[2],
            customer[3],
            customer[4],
            customer[5],
            customer[6],
            customer[7]
        )

    l.update_customer_credit("8", 0)
    l.update_customer_credit("9", 1000)
    l.update_customer_credit("10", -42)
    with pytest.raises(ValueError) as excinfo:
        l.update_customer_credit("00100", 1000) # error
        assert 'NoCustomer' in str(excinfo.value)