54 lines
2 KiB
Python
54 lines
2 KiB
Python
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from orchestra.core import accounts
|
|
from orchestra.models.fields import MultiSelectField
|
|
|
|
from . import settings
|
|
|
|
|
|
class ContactQuerySet(models.QuerySet):
|
|
def filter(self, *args, **kwargs):
|
|
usages = kwargs.pop('email_usages', [])
|
|
qs = models.Q()
|
|
for usage in usages:
|
|
qs = qs | models.Q(email_usage__regex=r'.*(^|,)+%s($|,)+.*' % usage)
|
|
return super(ContactQuerySet, self).filter(qs, *args, **kwargs)
|
|
|
|
|
|
class Contact(models.Model):
|
|
BILLING = 'BILLING'
|
|
EMAIL_USAGES = (
|
|
('SUPPORT', _("Support tickets")),
|
|
('ADMIN', _("Administrative")),
|
|
(BILLING, _("Billing")),
|
|
('TECH', _("Technical")),
|
|
('ADDS', _("Announcements")),
|
|
('EMERGENCY', _("Emergency contact")),
|
|
)
|
|
|
|
objects = ContactQuerySet.as_manager()
|
|
|
|
account = models.ForeignKey('accounts.Account', verbose_name=_("Account"),
|
|
related_name='contacts', null=True)
|
|
short_name = models.CharField(_("short name"), max_length=128)
|
|
full_name = models.CharField(_("full name"), max_length=256, blank=True)
|
|
email = models.EmailField()
|
|
email_usage = MultiSelectField(_("email usage"), max_length=256, blank=True,
|
|
choices=EMAIL_USAGES,
|
|
default=settings.CONTACTS_DEFAULT_EMAIL_USAGES)
|
|
phone = models.CharField(_("phone"), max_length=32, blank=True)
|
|
phone2 = models.CharField(_("alternative phone"), max_length=32, blank=True)
|
|
address = models.TextField(_("address"), blank=True)
|
|
city = models.CharField(_("city"), max_length=128, blank=True,
|
|
default=settings.CONTACTS_DEFAULT_CITY)
|
|
zipcode = models.PositiveIntegerField(_("zip code"), null=True, blank=True)
|
|
country = models.CharField(_("country"), max_length=20, blank=True,
|
|
default=settings.CONTACTS_DEFAULT_COUNTRY)
|
|
|
|
def __unicode__(self):
|
|
return self.short_name
|
|
|
|
|
|
accounts.register(Contact)
|