1
0
Fork 0

Update registers names & imports

This commit is contained in:
Adam Dobrawy 2018-05-15 00:54:46 +02:00
parent 6ded51bcde
commit d4e052bb79
3 changed files with 7 additions and 6 deletions

View file

@ -35,7 +35,7 @@ from django_mailbox.signals import message_received
from django_mailbox.transports import Pop3Transport, ImapTransport, \
MaildirTransport, MboxTransport, BabylTransport, MHTransport, \
MMDFTransport, GmailImapTransport
from django_mailbox.transports.registry import transport_registry
from django_mailbox.transports.registry import registry
logger = logging.getLogger(__name__)
@ -224,7 +224,7 @@ class Mailbox(models.Model):
conn = MHTransport(self.location)
elif self.type == 'mmdf':
conn = MMDFTransport(self.location)
return transport_registry[self.type].from_url(self.uri)
return registry[self.type].from_url(self.uri)
def process_incoming_message(self, message):
"""Process a message incoming to this mailbox."""

View file

@ -8,3 +8,4 @@ from django_mailbox.transports.babyl import BabylTransport
from django_mailbox.transports.mh import MHTransport
from django_mailbox.transports.mmdf import MMDFTransport
from django_mailbox.transports.gmail import GmailImapTransport
from django_mailbox.transports.registry import registry, register

View file

@ -1,8 +1,8 @@
transport_registry = {}
registry = {}
def register_transport(transport_type):
def register(transport_type):
def decorator(f):
transport_registry[transport_type] = f
registry[transport_type] = f
return f
return decorator