37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
import re
|
|
|
|
from .forms import (POP3EmailSetupForm, IMAPEmailSetupForm,
|
|
StagingFolderSetupForm, WatchFolderSetupForm,
|
|
WebFormSetupForm)
|
|
from .literals import (SOURCE_CHOICE_EMAIL_IMAP, SOURCE_CHOICE_EMAIL_POP3,
|
|
SOURCE_CHOICE_STAGING, SOURCE_CHOICE_WATCH,
|
|
SOURCE_CHOICE_WEB_FORM)
|
|
from .models import (IMAPEmail, POP3Email, StagingFolderSource,
|
|
WatchFolderSource, WebFormSource)
|
|
|
|
|
|
def get_class(source_type):
|
|
if source_type == SOURCE_CHOICE_WEB_FORM:
|
|
return WebFormSource
|
|
elif source_type == SOURCE_CHOICE_STAGING:
|
|
return StagingFolderSource
|
|
elif source_type == SOURCE_CHOICE_WATCH:
|
|
return WatchFolderSource
|
|
elif source_type == SOURCE_CHOICE_EMAIL_POP3:
|
|
return POP3Email
|
|
elif source_type == SOURCE_CHOICE_EMAIL_IMAP:
|
|
return IMAPEmail
|
|
|
|
|
|
def get_form_class(source_type):
|
|
if source_type == SOURCE_CHOICE_WEB_FORM:
|
|
return WebFormSetupForm
|
|
elif source_type == SOURCE_CHOICE_STAGING:
|
|
return StagingFolderSetupForm
|
|
elif source_type == SOURCE_CHOICE_WATCH:
|
|
return WatchFolderSetupForm
|
|
elif source_type == SOURCE_CHOICE_EMAIL_POP3:
|
|
return POP3EmailSetupForm
|
|
elif source_type == SOURCE_CHOICE_EMAIL_IMAP:
|
|
return IMAPEmailSetupForm
|