def cases_case(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 3,
'tenant': kwargs.get('tenant', self.tenant),
'status': kwargs.get('status') if kwargs.get('status') else iterator(self.cases_case_status),
'type': kwargs.get('type') if kwargs.get('type') else iterator(self.cases_case_type),
'assigned_to': kwargs.get('assigned_to') if kwargs.get('assigned_to') else iterator(self.users_user),
'teams': kwargs.get('teams') if kwargs.get('teams') else iterator(self.users_team),
'account': kwargs.get('account') if kwargs.get('account') else iterator(self.accounts_account),
'contact': kwargs.get('contact') if kwargs.get('contact') else iterator(self.contacts_contact),
})
contact = kwargs.pop('contact') # Remove contact for now.
cases_with_account = CaseFactory.create_batch(**kwargs)
kwargs['account'] = None # Replace account with contact.
kwargs['contact'] = contact
cases_with_contact = CaseFactory.create_batch(**kwargs)
kwargs['teams'] = None # Remove all connections.
kwargs['contact'] = None
cases_without = CaseFactory.create_batch(**kwargs)
self.stdout.write('Done with cases_case.')
return cases_with_account + cases_with_contact + cases_without
def cases_case(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 3,
'tenant': kwargs.get('tenant', self.tenant),
'status': kwargs.get('status') if kwargs.get('status') else iterator(self.cases_case_status),
'type': kwargs.get('type') if kwargs.get('type') else iterator(self.cases_case_type),
'assigned_to': kwargs.get('assigned_to') if kwargs.get('assigned_to') else iterator(self.users_user),
'teams': kwargs.get('teams') if kwargs.get('teams') else iterator(self.users_team),
'account': kwargs.get('account') if kwargs.get('account') else iterator(self.accounts_account),
'contact': kwargs.get('contact') if kwargs.get('contact') else iterator(self.contacts_contact),
})
contact = kwargs.pop('contact') # Remove contact for now.
cases_with_account = CaseFactory.create_batch(**kwargs)
kwargs['account'] = None # Replace account with contact.
kwargs['contact'] = contact
cases_with_contact = CaseFactory.create_batch(**kwargs)
kwargs['teams'] = None # Remove all connections.
kwargs['contact'] = None
cases_without = CaseFactory.create_batch(**kwargs)
self.stdout.write('Done with cases_case.')
return cases_with_account + cases_with_contact + cases_without
def cases_case(self, **kwargs):
kwargs.update(
{
"size": kwargs.get("size", self.batch_size) / 3,
"tenant": kwargs.get("tenant", self.tenant),
"status": kwargs.get("status") if kwargs.get("status") else iterator(self.cases_case_status),
"type": kwargs.get("type") if kwargs.get("type") else iterator(self.cases_case_type),
"assigned_to": kwargs.get("assigned_to") if kwargs.get("assigned_to") else iterator(self.users_user),
"teams": kwargs.get("teams") if kwargs.get("teams") else iterator(self.users_team),
"account": kwargs.get("account") if kwargs.get("account") else iterator(self.accounts_account),
"contact": kwargs.get("contact") if kwargs.get("contact") else iterator(self.contacts_contact),
}
)
contact = kwargs.pop("contact") # Remove contact for now.
cases_with_account = CaseFactory.create_batch(**kwargs)
kwargs["account"] = None # Replace account with contact.
kwargs["contact"] = contact
cases_with_contact = CaseFactory.create_batch(**kwargs)
kwargs["teams"] = None # Remove all connections.
kwargs["contact"] = None
cases_without = CaseFactory.create_batch(**kwargs)
self.stdout.write("Done with cases_case.")
return cases_with_account + cases_with_contact + cases_without
def notes_note(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size),
'tenant': kwargs.get('tenant', self.tenant),
'author': kwargs.get('author') if kwargs.get('author') else iterator(self.users_user),
'subject': kwargs.get('subject') if kwargs.get('subject') else iterator(self._notes_subject),
})
self.stdout.write('Done with notes_note.')
return NoteFactory.create_batch(**kwargs)
def accounts_function(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 2,
'account': kwargs.get('account') if kwargs.get('account') else iterator(self.accounts_account),
'contact': kwargs.get('contact') if kwargs.get('contact') else iterator(self.contacts_contact),
})
functions = FunctionFactory.create_batch(**kwargs)
self.stdout.write('Done with accounts_function.')
return functions
def accounts_function(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 2,
'account': kwargs.get('account') if kwargs.get('account') else iterator(self.accounts_account),
'contact': kwargs.get('contact') if kwargs.get('contact') else iterator(self.contacts_contact),
})
functions = FunctionFactory.create_batch(**kwargs)
self.stdout.write('Done with accounts_function.')
return functions
def notes_note(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size),
'tenant': kwargs.get('tenant', self.tenant),
'author': kwargs.get('author') if kwargs.get('author') else iterator(self.users_user),
'subject': kwargs.get('subject') if kwargs.get('subject') else iterator(self._notes_subject),
})
self.stdout.write('Done with notes_note.')
return NoteFactory.create_batch(**kwargs)
def notes_note(self, **kwargs):
kwargs.update(
{
"size": kwargs.get("size", self.batch_size),
"tenant": kwargs.get("tenant", self.tenant),
"author": kwargs.get("author") if kwargs.get("author") else iterator(self.users_user),
"subject": kwargs.get("subject") if kwargs.get("subject") else iterator(self._notes_subject),
}
)
self.stdout.write("Done with notes_note.")
return NoteFactory.create_batch(**kwargs)
def accounts_function(self, **kwargs):
kwargs.update(
{
"size": kwargs.get("size", self.batch_size) / 2,
"account": kwargs.get("account") if kwargs.get("account") else iterator(self.accounts_account),
"contact": kwargs.get("contact") if kwargs.get("contact") else iterator(self.contacts_contact),
}
)
functions = FunctionFactory.create_batch(**kwargs)
self.stdout.write("Done with accounts_function.")
return functions
def users_login(self, **kwargs):
kwargs.update(
{
"tenant": kwargs.get("tenant", self.tenant),
"teams": kwargs.get("teams") if kwargs.get("teams") else iterator(self.users_team),
"is_active": kwargs.get("is_active", True),
"email": "*****@*****.**" % self.tenant.pk,
}
)
user = LilyUserFactory.create(**kwargs)
self.stdout.write(
"\nYou can now login as a normal user in %(tenant)s with:\n%(email)s\n%(password)s\n"
% {"tenant": self.tenant, "email": user.email, "password": "******"}
)
kwargs["email"] = "*****@*****.**" % self.tenant.pk
superuser = LilySuperUserFactory.create(**kwargs)
self.stdout.write(
"\nYou can now login as a superuser in %(tenant)s with:\n%(email)s\n%(password)s\n\n"
% {"tenant": self.tenant, "email": superuser.email, "password": "******"}
)
return [user, superuser]
def users_login(self, **kwargs):
kwargs.update({
'tenant': kwargs.get('tenant', self.tenant),
'teams': kwargs.get('teams') if kwargs.get('teams') else iterator(self.users_team),
'is_active': kwargs.get('is_active', True),
'email': '*****@*****.**' % self.tenant.pk,
})
user = LilyUserFactory.create(**kwargs)
self.stdout.write('\nYou can now login as a normal user in %(tenant)s with:\n%(email)s\n%(password)s\n' % {
'tenant': self.tenant,
'email': user.email,
'password': '******'
})
kwargs['email'] = '*****@*****.**' % self.tenant.pk
superuser = LilySuperUserFactory.create(**kwargs)
self.stdout.write('\nYou can now login as a superuser in %(tenant)s with:\n%(email)s\n%(password)s\n\n' % {
'tenant': self.tenant,
'email': superuser.email,
'password': '******'
})
return [user, superuser]
def users_login(self, **kwargs):
kwargs.update({
'tenant': kwargs.get('tenant', self.tenant),
'teams': kwargs.get('teams') if kwargs.get('teams') else iterator(self.users_team),
'is_active': kwargs.get('is_active', True),
'email': '*****@*****.**' % self.tenant.pk,
})
user = LilyUserFactory.create(**kwargs)
self.stdout.write('\nYou can now login as a normal user in %(tenant)s with:\n%(email)s\n%(password)s\n' % {
'tenant': self.tenant,
'email': user.email,
'password': '******'
})
kwargs['email'] = '*****@*****.**' % self.tenant.pk
superuser = LilySuperUserFactory.create(**kwargs)
self.stdout.write('\nYou can now login as a superuser in %(tenant)s with:\n%(email)s\n%(password)s\n\n' % {
'tenant': self.tenant,
'email': superuser.email,
'password': '******'
})
return [user, superuser]
class PlayFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = Play
strategy = BUILD_STRATEGY
sqlalchemy_get_or_create = (
'game',
'player',
)
sqlalchemy_session = scoped_session(lambda: db.session,
scopefunc=lambda: db.session)
turn_number = factory.Faker('random_number')
score = factory.Faker('random_number')
game = factory.iterator(games)
player = factory.iterator(players)
word = factory.SubFactory(WordFactory)
def deals_deal(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 2,
'tenant': kwargs.get('tenant', self.tenant),
'account': kwargs.get('account') if kwargs.get('account') else iterator(self.accounts_account),
'contact': kwargs.get('contact') if kwargs.get('contact') else iterator(self.contacts_contact),
'assigned_to': kwargs.get('assigned_to') if kwargs.get('assigned_to') else iterator(self.users_user),
'contacted_by': kwargs.get('contacted_by') if kwargs.get('contacted_by') else
iterator(self.deals_deal_contacted_by),
'found_through': kwargs.get('found_through') if kwargs.get('found_through') else
iterator(self.deals_deal_found_through),
'next_step': kwargs.get('next_step') if kwargs.get('next_step') else
iterator(self.deals_deal_next_step),
'status': kwargs.get('status') if kwargs.get('status') else iterator(self.deals_deal_status),
'why_customer': kwargs.get('why_customer') if kwargs.get('why_customer') else
iterator(self.deals_deal_why_customer),
})
contact = kwargs.pop('contact')
kwargs['contact'] = None
deals_with_accounts = DealFactory.create_batch(**kwargs)
kwargs['contact'] = contact
deals_with_contacts = DealFactory.create_batch(**kwargs)
self.stdout.write('Done with deals_deal.')
return deals_with_accounts + deals_with_contacts
def deals_deal(self, **kwargs):
kwargs.update(
{
"size": kwargs.get("size", self.batch_size) / 2,
"tenant": kwargs.get("tenant", self.tenant),
"account": kwargs.get("account") if kwargs.get("account") else iterator(self.accounts_account),
"contact": kwargs.get("contact") if kwargs.get("contact") else iterator(self.contacts_contact),
"assigned_to": kwargs.get("assigned_to") if kwargs.get("assigned_to") else iterator(self.users_user),
"contacted_by": kwargs.get("contacted_by")
if kwargs.get("contacted_by")
else iterator(self.deals_deal_contacted_by),
"found_through": kwargs.get("found_through")
if kwargs.get("found_through")
else iterator(self.deals_deal_found_through),
"next_step": kwargs.get("next_step")
if kwargs.get("next_step")
else iterator(self.deals_deal_next_step),
"status": kwargs.get("status") if kwargs.get("status") else iterator(self.deals_deal_status),
"why_customer": kwargs.get("why_customer")
if kwargs.get("why_customer")
else iterator(self.deals_deal_why_customer),
}
)
contact = kwargs.pop("contact")
kwargs["contact"] = None
deals_with_accounts = DealFactory.create_batch(**kwargs)
kwargs["contact"] = contact
deals_with_contacts = DealFactory.create_batch(**kwargs)
self.stdout.write("Done with deals_deal.")
return deals_with_accounts + deals_with_contacts
def deals_deal(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 2,
'tenant': kwargs.get('tenant', self.tenant),
'account': kwargs.get('account') if kwargs.get('account') else iterator(self.accounts_account),
'contact': kwargs.get('contact') if kwargs.get('contact') else iterator(self.contacts_contact),
'assigned_to': kwargs.get('assigned_to') if kwargs.get('assigned_to') else iterator(self.users_user),
'contacted_by': kwargs.get('contacted_by') if kwargs.get('contacted_by') else
iterator(self.deals_deal_contacted_by),
'found_through': kwargs.get('found_through') if kwargs.get('found_through') else
iterator(self.deals_deal_found_through),
'next_step': kwargs.get('next_step') if kwargs.get('next_step') else
iterator(self.deals_deal_next_step),
'status': kwargs.get('status') if kwargs.get('status') else iterator(self.deals_deal_status),
'why_customer': kwargs.get('why_customer') if kwargs.get('why_customer') else
iterator(self.deals_deal_why_customer),
})
contact = kwargs.pop('contact')
kwargs['contact'] = None
deals_with_accounts = DealFactory.create_batch(**kwargs)
kwargs['contact'] = contact
deals_with_contacts = DealFactory.create_batch(**kwargs)
self.stdout.write('Done with deals_deal.')
return deals_with_accounts + deals_with_contacts
class DepartmentFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = models.Department
sqlalchemy_session = models.db.session
sqlalchemy_session_persistence = 'flush'
abbreviation = factory.Sequence(lambda n: 'SMPL' + str(n))
name = 'Sample Department'
school = factory.iterator(schools)
class AdvertisementImageFactory(factory.django.DjangoModelFactory):
""" Generates dummy advertisement image object """
description = factory.Faker('word')
alternate_text = factory.Faker('word')
parent_advertisement = factory.iterator(Advertisement.objects.all)
image = factory.django.ImageField()
class Meta:
model = AdvertisementImage
class AdvertisementFactory(factory.django.DjangoModelFactory):
""" Generates dummy advertisement object """
name = factory.Faker('sentence')
owner = factory.iterator(
User.objects.all) # !TODO: If iterator is empty then creation crashes
category = factory.iterator(Category.objects.all)
contact_email = factory.Faker('ascii_free_email')
contact_phone = factory.Faker('phone_number')
content = factory.Faker('text')
views = factory.Faker('random_int')
importance = factory.Faker('random_int')
expires_date = factory.Faker('date_time_this_month',
tzinfo=timezone.get_current_timezone())
price = factory.Faker('random_int')
location_city = factory.Faker('address')
status = 'published'
class Meta:
model = Advertisement
class WordFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = Word
sqlalchemy_get_or_create = ('word', )
sqlalchemy_session = scoped_session(lambda: db.session,
scopefunc=lambda: db.session)
word = factory.Faker('word')
times_used = factory.Faker('random_number')
language = factory.iterator(languages)
def users_user(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 2,
'tenant': kwargs.get('tenant', self.tenant),
'teams': kwargs.get('teams') if kwargs.get('teams') else iterator(self.users_team)
})
users = LilyUserFactory.create_batch(**kwargs)
superusers = LilySuperUserFactory.create_batch(**kwargs)
self.stdout.write('Done with users_user.')
return users + superusers
def users_user(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 2,
'tenant': kwargs.get('tenant', self.tenant),
'teams': kwargs.get('teams') if kwargs.get('teams') else iterator(self.users_team)
})
users = LilyUserFactory.create_batch(**kwargs)
superusers = LilySuperUserFactory.create_batch(**kwargs)
self.stdout.write('Done with users_user.')
return users + superusers
def users_user(self, **kwargs):
kwargs.update(
{
"size": kwargs.get("size", self.batch_size) / 2,
"tenant": kwargs.get("tenant", self.tenant),
"teams": kwargs.get("teams") if kwargs.get("teams") else iterator(self.users_team),
}
)
users = LilyUserFactory.create_batch(**kwargs)
superusers = LilySuperUserFactory.create_batch(**kwargs)
self.stdout.write("Done with users_user.")
return users + superusers
def accounts_account(self, **kwargs):
kwargs.update({
'size':
kwargs.get('size', self.batch_size) / 2,
'tenant':
kwargs.get('tenant', self.tenant),
'assigned_to':
kwargs.get('assigned_to')
if kwargs.get('assigned_to') else iterator(self.users_user),
})
accounts_with_users = AccountFactory.create_batch(**kwargs)
del kwargs['assigned_to']
accounts_without_users = AccountFactory.create_batch(**kwargs)
self.stdout.write('Done with accounts_account.')
return accounts_with_users + accounts_without_users
def accounts_account(self, **kwargs):
kwargs.update(
{
"size": kwargs.get("size", self.batch_size) / 2,
"tenant": kwargs.get("tenant", self.tenant),
"assigned_to": kwargs.get("assigned_to") if kwargs.get("assigned_to") else iterator(self.users_user),
}
)
accounts_with_users = AccountFactory.create_batch(**kwargs)
del kwargs["assigned_to"]
accounts_without_users = AccountFactory.create_batch(**kwargs)
self.stdout.write("Done with accounts_account.")
return accounts_with_users + accounts_without_users
def accounts_account(self, **kwargs):
kwargs.update({
'size': kwargs.get('size', self.batch_size) / 2,
'tenant': kwargs.get('tenant', self.tenant),
'assigned_to': kwargs.get('assigned_to') if kwargs.get('assigned_to') else iterator(self.users_user),
})
accounts_with_users = AccountFactory.create_batch(**kwargs)
del kwargs['assigned_to']
accounts_without_users = AccountFactory.create_batch(**kwargs)
self.stdout.write('Done with accounts_account.')
return accounts_with_users + accounts_without_users
