"""Setup data generator."""
from mixer.backend.peewee import Mixer
mixer = Mixer(commit=True)
gen_tc_id = lambda: mixer.faker.md5()[:24]
@mixer.middleware("salesmeter.models.Operator")
def set_password(operator):
"""Encrypt user's password."""
operator.set_password(operator.password)
return operator
mixer.register("salesmeter.models.Ticket", tc_id=gen_tc_id, barcode=lambda: mixer.faker.random_number(12))
mixer.register("salesmeter.models.Partner", tc_id=gen_tc_id)
mixer.register("salesmeter.models.Event", tc_id=gen_tc_id)
mixer.register("salesmeter.models.TicketSet", tc_id=gen_tc_id)
mixer.register("salesmeter.models.Order", tc_id=gen_tc_id, buyer=mixer.faker.name)
