| Server IP : 85.214.239.14 / Your IP : 18.227.46.87 Web Server : Apache/2.4.62 (Debian) System : Linux h2886529.stratoserver.net 4.9.0 #1 SMP Tue Jan 9 19:45:01 MSK 2024 x86_64 User : www-data ( 33) PHP Version : 7.4.18 Disable Function : pcntl_alarm,pcntl_fork,pcntl_waitpid,pcntl_wait,pcntl_wifexited,pcntl_wifstopped,pcntl_wifsignaled,pcntl_wifcontinued,pcntl_wexitstatus,pcntl_wtermsig,pcntl_wstopsig,pcntl_signal,pcntl_signal_get_handler,pcntl_signal_dispatch,pcntl_get_last_error,pcntl_strerror,pcntl_sigprocmask,pcntl_sigwaitinfo,pcntl_sigtimedwait,pcntl_exec,pcntl_getpriority,pcntl_setpriority,pcntl_async_signals,pcntl_unshare, MySQL : OFF | cURL : OFF | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : OFF Directory : /srv/modoboa/env/lib64/python3.5/site-packages/modoboa_contacts/ |
Upload File : |
"""Contacts factories."""
import factory
from modoboa.admin import factories as admin_factories
from . import models
class AddressBookFactory(factory.django.DjangoModelFactory):
"""Address book factory."""
class Meta:
model = models.AddressBook
user = factory.SubFactory(admin_factories.UserFactory)
name = "Contacts"
_path = "contacts"
class CategoryFactory(factory.django.DjangoModelFactory):
"""Category factory."""
class Meta:
model = models.Category
class EmailAddressFactory(factory.django.DjangoModelFactory):
"""Email address factory."""
class Meta:
model = models.EmailAddress
type = "home"
class PhoneNumberFactory(factory.django.DjangoModelFactory):
"""Phone number factory."""
class Meta:
model = models.PhoneNumber
type = "home"
class ContactFactory(factory.django.DjangoModelFactory):
"""Contact factory."""
class Meta:
model = models.Contact
first_name = "Homer"
last_name = "Simpson"
display_name = factory.LazyAttribute(
lambda c: "{}{}".format(c.first_name, c.last_name))
@factory.post_generation
def categories(self, create, extracted, **dummy_kwargs):
"""Add categories to contact."""
if not create or not extracted:
return
for item in extracted:
self.categories.add(item)
@factory.post_generation
def emails(self, create, extracted, **dummy_kwargs):
"""Add emails to contact."""
if not create or not extracted:
return
for item in extracted:
EmailAddressFactory(contact=self, address=item)
@factory.post_generation
def phone_numbers(self, create, extracted, **dummy_kwargs):
"""Add phone numbers to contact."""
if not create or not extracted:
return
for item in extracted:
PhoneNumberFactory(contact=self, number=item)