VMM/emailaddress: Class DestinationEmailAddress accepts also v0.6.x
authorPascal Volk <neverseen@users.sourceforge.net>
Sat, 03 Sep 2011 22:07:55 +0000
branchv0.6.x
changeset 427 8e41e77b84e6
parent 426 933b9debbec1
child 428 0d8e7977ae63
VMM/emailaddress: Class DestinationEmailAddress accepts also something@localhost addresses now.
VirtualMailManager/emailaddress.py
--- a/VirtualMailManager/emailaddress.py	Sat Sep 03 20:42:05 2011 +0000
+++ b/VirtualMailManager/emailaddress.py	Sat Sep 03 22:07:55 2011 +0000
@@ -11,8 +11,9 @@
 
 from VirtualMailManager.domain import check_domainname, get_gid
 from VirtualMailManager.constants import \
-     DOMAIN_NO_NAME, INVALID_ADDRESS, LOCALPART_INVALID, LOCALPART_TOO_LONG
-from VirtualMailManager.errors import EmailAddressError as EAErr
+     DOMAIN_NO_NAME, INVALID_ADDRESS, LOCALPART_INVALID, LOCALPART_TOO_LONG, \
+     DOMAIN_INVALID
+from VirtualMailManager.errors import DomainError, EmailAddressError as EAErr
 
 
 RE_LOCALPART = re.compile(r"[^\w!#$%&'\*\+-\.\/=?^_`{\|}~]")
@@ -23,12 +24,13 @@
     """Simple class for validated e-mail addresses."""
     __slots__ = ('_localpart', '_domainname')
 
-    def __init__(self, address):
+    def __init__(self, address, _validate=True):
         """Creates a new instance from the string/unicode ``address``."""
         assert isinstance(address, basestring)
         self._localpart = None
         self._domainname = None
-        self._chk_address(address)
+        if _validate:
+            self._chk_address(address)
 
     @property
     def localpart(self):
@@ -86,9 +88,9 @@
 class DestinationEmailAddress(EmailAddress):
     """Provides additionally the domains group ID - when the domain is known
     in the database."""
-    __slots__ = ('_gid')
+    __slots__ = ('_gid', '_localhost')
 
-    def __init__(self, address, dbh):
+    def __init__(self, address, dbh, _validate=False):
         """Creates a new DestinationEmailAddress instance
 
         Arguments:
@@ -98,9 +100,23 @@
         `dbh`: pyPgSQL.PgSQL.Connection/pyPgSQL.PgSQL.connection
           a database connection for the database access
         """
-        super(DestinationEmailAddress, self).__init__(address)
+        super(DestinationEmailAddress, self).__init__(address, _validate)
+        self._localhost = False
+        if not _validate:
+            try:
+                self._chk_address(address)
+            except DomainError, err:
+                if err.code is DOMAIN_INVALID and \
+                   address.split('@')[1] == 'localhost':
+                    self._localhost = True
+                    self._domainname = 'localhost'
+                else:
+                    raise
         self._gid = 0
-        self._find_domain(dbh)
+        if not self._localhost:
+            self._find_domain(dbh)
+        else:
+            self._localpart = self._localpart.lower()
 
     def _find_domain(self, dbh):
         """Checks if the domain is known"""
@@ -109,6 +125,11 @@
             self._localpart = self._localpart.lower()
 
     @property
+    def at_localhost(self):
+        """True when the address is something@localhost."""
+        return self._localhost
+
+    @property
     def gid(self):
         """The domains group ID. 0 if the domain is not known."""
         return self._gid