def db_init():
"""
Function to initialize DB, create
and add tables
"""
DB.init('customer.db')
DB.create_tables([Customer])
add_customer(CUSTOMER_LIST)
def setup_db():
""" Sets up database for test cases """
DB.drop_tables([Customer])
# DB.close()
DB.create_tables([Customer])
add_customer(1, "Leo", "Messi", "100 Las Ramblas", 1234567890,
"*****@*****.**", True, 5000000.00)
add_customer(2, "Ivan", "Rakitic", "111 Las Ramblas", 2345678901,
"*****@*****.**", True, 2000000.00)
add_customer(3, "Gerard", "Pique", "123 Las Ramblas", 3333333333,
"*****@*****.**", True, 300000.00)
def db_init():
"""
Function to initialize DB, create/add
tables and add customer data to the DB
"""
DB.init('customer.db')
DB.create_tables([Customer])
[
add_customer(customer) for customer in CUSTOMER_LIST
if customer is not None
]
def setUp(self):
DB.create_tables([Customer])
"""Creates a customer database with peewee ORM, sqlite and python"""
import logging
from customer_model import Customer, DB
logging.basicConfig(level=logging.INFO)
logging.info("Creating the Customer table")
DB.create_tables([Customer])
logging.info("Closing the customer database")
#DB.close()
def setup_db():
""" Sets up database for test cases """
DB.drop_tables([Customer])
DB.close()
DB.create_tables([Customer])
DB.close()