def __init__(self):
BaseCommand.__init__(self)
self.style = no_style()
# Keep a count of the installed objects and ontologies
self.count = [0, 0]
self.models = set()
self.verbosity, self.show_traceback, self.fail_gracefully = 1, False, False
def __init__(self, *args, **kwargs):
self.verbosity = 0
self.stats = {}
self._ignore = []
BaseCommand.__init__(self, *args, **kwargs)
def __init__(self, *args, **kwargs):
"""
Creation of the ejabberd atuh bridge service.
:Parameters:
- `args`: all non-keyword arguments
- `kwargs`: all keyword arguments
"""
BaseCommand.__init__(self, *args, **kwargs)
try:
log_level = int(logging.DEBUG)
except:
log_level = logging.INFO
if os.access("/var/log/ejabberd/kauth.log", os.W_OK):
logging.basicConfig(level=log_level,
format='%(asctime)s %(levelname)s %(message)s',
filename="/var/log/ejabberd/kauth.log",
filemode='a')
else:
logging.basicConfig(level=log_level,
format='%(asctime)s %(levelname)s %(message)s',
stream=sys.stderr)
logging.warn(
('Could not write to ' + '/var/log/ejabberd/kauth.log' +
'. Falling back to stderr ...'))
logging.info(
('ejabberd kauth process started' + ' (more than one is common)'))
def __init__(self, *args, **kwargs):
class NiceContext(Context):
"""
Same as the real context class, but override some methods in order to gain
nice colored output.
"""
def __init__(s, *args, **kwargs):
kwargs['insert_debug_symbols'] = self.insert_debug_symbols
Context.__init__(s, *args, **kwargs)
def compile_media_callback(s, compress_tag, media_files):
"""
When the compiler notifies us that compiling of this file begins.
"""
if compress_tag:
print self.colored('Compiling media files from', 'yellow'),
print self.colored(' "%s" ' % compress_tag.path, 'green'),
print self.colored(' (line %s, column %s)" ' % (compress_tag.line, compress_tag.column), 'yellow')
else:
print self.colored('Compiling media files', 'yellow')
for m in media_files:
print self.colored(' * %s' % m, 'green')
def compile_media_progress_callback(s, compress_tag, media_file, current, total, file_size):
"""
Print progress of compiling media files.
"""
print self.colored(' (%s / %s):' % (current, total), 'yellow'),
print self.colored(' %s (%s bytes)' % (media_file, file_size), 'green')
self.NiceContext = NiceContext
BaseCommand.__init__(self, *args, **kwargs)
def __init__(self, *args, **kwargs):
self.verbosity = 0
self.stats = {}
self._ignore = []
BaseCommand.__init__(self, *args, **kwargs)
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (make_option(
'--queues',
'-q',
help='Queues to include (default: all). Use queue slugs'), )
def add_arguments(self, parser):
BaseCommand.add_arguments(self, parser)
parser.add_argument('addrport', nargs='?',
help='Optional port number, or ipaddr:port')
# The below flags are for legacy compat.
# 2017-06-08 added addrport positional argument,
# because:
# - more consistent with runserver.
# - don't have to remember the name of the flags (is it --bind or --addr etc)
# - quicker to type
# - we don't need positional args for anything else
parser.add_argument(
'--addr', action='store', type=str, dest='addr', default=None,
help='The host/address to bind to (default: {})'.format(DEFAULT_ADDR))
ahelp = (
'Port number to listen on. Defaults to the environment variable '
'$PORT (if defined), or {}.'.format(DEFAULT_PORT)
)
parser.add_argument(
'--port', action='store', type=int, dest='port', default=None,
help=ahelp)
def __init__(self, *args, **kwargs):
"""
Creation of the ejabberd atuh bridge service.
:Parameters:
- `args`: all non-keyword arguments
- `kwargs`: all keyword arguments
"""
BaseCommand.__init__(self, *args, **kwargs)
try:
log_level = int(settings.TUNNEL_EJABBERD_AUTH_GATEWAY_LOG_LEVEL)
except:
log_level = logging.INFO
# If we can write to the log do so, else fail back to the console
if os.access(settings.TUNNEL_EJABBERD_AUTH_GATEWAY_LOG, os.W_OK):
logging.basicConfig(
level=log_level,
format='%(asctime)s %(levelname)s %(message)s',
filename=settings.TUNNEL_EJABBERD_AUTH_GATEWAY_LOG,
filemode='a')
else:
logging.basicConfig(
level=log_level,
format='%(asctime)s %(levelname)s %(message)s',
stream=sys.stderr)
logging.warn(('Could not write to ' +
settings.TUNNEL_EJABBERD_AUTH_GATEWAY_LOG +
'. Falling back to stderr ...'))
logging.info(('ejabberd_auth_bridge process started' +
' (more than one is common)'))
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (make_option('--quiet',
'-q',
default=False,
action='store_true',
help='Hide all command output'), )
def add_arguments(self, parser):
BaseCommand.add_arguments(self, parser)
parser.add_argument('addrport', nargs='?',
help='Optional port number, or ipaddr:port')
# The below flags are for legacy compat.
# 2017-06-08 added addrport positional argument,
# because:
# - more consistent with runserver.
# - don't have to remember the name of the flags (is it --bind or --addr etc)
# - quicker to type
# - we don't need positional args for anything else
parser.add_argument(
'--addr', action='store', type=str, dest='addr', default=None,
help='The host/address to bind to (default: {})'.format(DEFAULT_ADDR))
ahelp = (
'Port number to listen on. Defaults to the environment variable '
'$PORT (if defined), or {}.'.format(DEFAULT_PORT)
)
parser.add_argument(
'--port', action='store', type=int, dest='port', default=None,
help=ahelp)
ahelp = (
'Run an SSL server directly in Daphne with a self-signed cert/key'
)
parser.add_argument(
'--dev-https', action='store_true', dest='dev_https', default=False,
help=ahelp)
def add_arguments(self, parser):
BaseCommand.add_arguments(self, parser)
ahelp = ('TODO')
parser.add_argument('--reload',
action='store_true',
dest='use_reloader',
default=False,
help=ahelp)
ahelp = (
'The port that the http server should run on. It defaults to '
'5000. This value can be set by the environment variable $PORT.')
parser.add_argument('--port',
action='store',
type=int,
dest='port',
default=None,
help=ahelp)
parser.add_argument('--addr',
action='store',
type=str,
dest='addr',
default='0.0.0.0',
help=ahelp)
def add_arguments(self, parser):
BaseCommand.add_arguments(self, parser)
ahelp = ('TODO')
parser.add_argument('--reload',
action='store_true',
dest='use_reloader',
default=False,
help=ahelp)
ahelp = (
'Port number to listen on. Defaults to the environment variable '
'$PORT (if defined), or 8000.')
parser.add_argument('--port',
action='store',
type=int,
dest='port',
default=None,
help=ahelp)
parser.add_argument(
'--addr',
action='store',
type=str,
dest='addr',
default='0.0.0.0',
help='The host/address to bind to (default: 0.0.0.0)')
parser.add_argument(
'--botworker',
action='store_true',
dest='botworker',
default=False,
help='--botworker flag is deprecated and has no effect')
def __init__(self, *args, **kwargs):
"""
Creation of the ejabberd atuh bridge service.
:Parameters:
- `args`: all non-keyword arguments
- `kwargs`: all keyword arguments
"""
BaseCommand.__init__(self, *args, **kwargs)
try:
log_level = int(logging.DEBUG)
except:
log_level = logging.INFO
if os.access("/var/log/ejabberd/kauth.log", os.W_OK):
logging.basicConfig(
level=log_level,
format='%(asctime)s %(levelname)s %(message)s',
filename="/var/log/ejabberd/kauth.log",
filemode='a')
else:
logging.basicConfig(
level=log_level,
format='%(asctime)s %(levelname)s %(message)s',
stream=sys.stderr)
logging.warn(('Could not write to ' +
'/var/log/ejabberd/kauth.log' +
'. Falling back to stderr ...'))
logging.info(('ejabberd kauth process started' +' (more than one is common)'))
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option("--queues", "-q", help="Queues to include (default: all). Use queue slugs"),
make_option("--verbose", "-v", action="store_true", default=False, help="Display a list of dates excluded"),
)
def add_arguments(self, parser: argparse.ArgumentParser) -> None:
BaseCommand.add_arguments(self, parser)
parser.add_argument('--discovery',
action='store',
type=str,
required=True,
help=pgettext_lazy(
'Scanner Custom',
'Discovery to execute'))
parser.add_argument('--disabled',
action='store_true',
default=False,
help=pgettext_lazy(
'Scanner Custom',
'Launch also disabled discoveries'))
parser.add_argument('--failing',
action='store_true',
default=False,
help=pgettext_lazy(
'Scanner Custom',
'Save results also for failing hosts'))
parser.add_argument('--destinations',
action='store',
type=str,
required=False,
help=pgettext_lazy(
'Scanner Custom',
'Execute the discovery only to the selected '
'destinations'))
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option("--days",
"-d",
help="Days of week (monday, tuesday, etc)"),
make_option(
"--occurrences",
"-o",
type="int",
default=1,
help="Occurrences: How many weeks ahead to exclude this day",
),
make_option(
"--queues",
"-q",
help="Queues to include (default: all). Use queue slugs",
),
make_option(
"--escalate-verbosely",
"-x",
action="store_true",
default=False,
help="Display a list of dates excluded",
),
)
def __init__(self, *args, **kwargs):
"""
Creation of the ejabberd atuh bridge service.
:Parameters:
- `args`: all non-keyword arguments
- `kwargs`: all keyword arguments
"""
BaseCommand.__init__(self, *args, **kwargs)
# try:
# log_level = int(settings.TUNNEL_EJABBERD_AUTH_GATEWAY_LOG_LEVEL)
# except:
log_level = logging.INFO
# If we can write to the log do so, else fail back to the console
#if os.access(settings.TUNNEL_EJABBERD_AUTH_GATEWAY_LOG, os.W_OK):
# logging.basicConfig(
# level=log_level,
# format='%(asctime)s %(levelname)s %(message)s',
# filename=settings.TUNNEL_EJABBERD_AUTH_GATEWAY_LOG,
# filemode='a')
#else:
logging.basicConfig(level=log_level,
format='%(asctime)s %(levelname)s %(message)s',
stream=sys.stderr)
#logging.warn(('Could not write to ' +
# settings.TUNNEL_EJABBERD_AUTH_GATEWAY_LOG +
# '. Falling back to stderr ...'))
logging.info(('ejabberd_auth_bridge process started' +
' (more than one is common)'))
def __init__(self):
BaseCommand.__init__(self)
self.skipped_users = []
self.skipped_products = []
self.skipped_logos = []
self.skipped_pics = []
signal.signal(signal.SIGINT, self.handle_interrupt)
def __init__(self, *args, **kwargs):
BaseCommand.__init__(self, *args, **kwargs)
self.target = None
self.files_in_export_dir = []
self.exported_files = []
self.compare_checksums = False
self.use_filename_format = False
self.delete = False
def __init__(self, *args, **kwargs):
self.verbosity = 0
self.file_consumer = None
self.mail_fetcher = None
BaseCommand.__init__(self, *args, **kwargs)
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option(
'--queues', '-q',
help='Queues to include (default: all). Use queue slugs'),
)
def __init__(self, *args, **kwargs):
self.verbosity = 0
self.file_consumer = None
self.mail_fetcher = None
BaseCommand.__init__(self, *args, **kwargs)
def __init__(self):
BaseCommand.__init__(self)
self.found = 0
self.no_match = 0
from_coord = SpatialReference(900913)
to_coord = SpatialReference(4326)
self.trans = CoordTransform(from_coord, to_coord)
self.rtrans = CoordTransform(to_coord, from_coord)
def __init__(self, *args, **kwargs):
self.logger = logging.getLogger(__name__)
BaseCommand.__init__(self, *args, **kwargs)
self.username = settings.MAIL_POLLING["username"]
self.password = settings.MAIL_POLLING["password"]
self.host = settings.MAIL_POLLING["host"]
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option(
'--quiet', '-q',
default=False,
action='store_true',
help='Hide all command output'),
)
def __init__(self, *largs, **kwargs):
BaseCommand.__init__(self, *largs, **kwargs)
from django.conf import settings
self.cascade = False
if settings.DATABASE_ENGINE:
self.cascade = settings.DATABASE_ENGINE.find('postgresql') >= 0
elif settings.DATABASES:
db_settings = settings.DATABASES.get('default')
self.cascade = db_settings.get('ENGINE').find('postgresql') >= 0
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (make_option(
'--quiet',
'-q',
default=False,
action='store_true',
help='Hide details about each message as they are processed.'), )
def __init__(self, *args, **kwargs):
BaseCommand.__init__(self, *args, **kwargs)
self.local_enabled = False
# Change this to True to log requests for debug *logs may include passwords*
self.LOGGING_ENABLED = False
if get_config("JABBER_LOCAL_ENABLED",None).value == "1":
self.local_enabled = True
self.local_user = get_config("JABBER_FROM_JID", False).value.split('@')[0]
self.local_pass = get_config("JABBER_FROM_PASSWORD", False).value
self.space_char = get_config("JABBER_LOCAL_SPACE_CHAR", False).value
def __init__(self, *args, **kwargs):
self.verbosity = 0
self.logger = logging.getLogger(__name__)
self.file_consumer = None
self.mail_fetcher = None
self.first_iteration = True
BaseCommand.__init__(self, *args, **kwargs)
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option(
'--quiet', '-q',
default=False,
action='store_true',
help='Hide details about each queue/message as they are processed'),
)
def __init__(self, *args, **kwargs):
BaseCommand.__init__(self, *args, **kwargs)
self.local_enabled = False
# Change this to True to log requests for debug *logs may include passwords*
self.LOGGING_ENABLED = False
if get_config("JABBER_LOCAL_ENABLED",None).value == "1":
self.local_enabled = True
self.local_user = get_config("JABBER_FROM_JID", False).value.split('@')[0]
self.local_pass = get_config("JABBER_FROM_PASSWORD", False).value
self.space_char = get_config("JABBER_LOCAL_SPACE_CHAR", False).value
def __init__(self):
"""
Host base command for all management host commands
"""
BaseCommand.__init__(self)
# Automatically save a DiscoveryResult record for each successful
# discovery result
self.save_results = True
# Verbosity level for printing results
self.verbosity = 0
def __init__(self):
BaseCommand.__init__(self)
self.address = '127.0.0.1'
#TODO: add some function for server/crawler address creation;
# now it can be bugged with large numbers of servers/crawlers
self.server_port = max([int(server.address.split(':')[2]) for server in TaskServer.objects.all()] + [INIT_SERVER_PORT]) + 1
self.crawler_port = max([int(crawler.address.split(':')[2]) for crawler in Crawler.objects.all()] + [INIT_CRAWLER_PORT]) + 1
self.last_scaling = time.time()
self.old_crawlers = [crawler.address for crawler in Crawler.objects.all()]
self.changed = False
def __init__(self, *args, **kwargs):
BaseCommand.__init__(self, *args, **kwargs)
# TODO: use Django settings to configure the logger
# https://docs.djangoproject.com/en/3.1/topics/logging/
logger = logging.getLogger()
logger.setLevel(logging.INFO)
handler = logging.StreamHandler()
formatter = logging.Formatter("{levelname:8} {message}", style="{")
handler.setFormatter(formatter)
logger.addHandler(handler)
def __init__(self):
"""
Discovery base command for all management discovery commands
Use scanner_tool string to choose the desider scanner to use
"""
BaseCommand.__init__(self)
# Automatically save a DiscoveryResult record for each successful
# discovery result
self.save_results = True
# Verbosity level for printing results
self.verbosity = 0
def __init__(self):
BaseCommand.__init__(self)
self.logger = getLogger("Review Progress")
self.logger.setLevel(logging.DEBUG)
stream_handler = StreamHandler()
file_handler = FileHandler('crawl_progress.log', 'a')
handler_format = Formatter('%(asctime)s - %(message)s')
stream_handler.setFormatter(handler_format)
file_handler.setFormatter(handler_format)
self.logger.addHandler(stream_handler)
self.logger.addHandler(file_handler)
stream_handler.setLevel(logging.DEBUG)
def __init__(self):
BaseCommand.__init__(self)
self.option_list = (
make_option(
'--queues',
help='Queues to include (default: all). Use queue slugs'),
make_option('--verboseescalation',
action='store_true',
default=False,
help='Display a list of dates excluded'),
)
def add_arguments(self, parser: argparse.ArgumentParser) -> None:
BaseCommand.add_arguments(self, parser)
parser.add_argument('--host',
action='store',
type=str,
required=True,
help=pgettext_lazy('Ping', 'Host to ping'))
parser.add_argument('--count',
action='store',
type=int,
default=0,
help=pgettext_lazy('Ping', 'Max ping count'))
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option(
"--quiet",
"-q",
default=False,
action="store_true",
help="Hide details about each message as they are processed.",
),
)
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option(
'--queues',
help='Queues to include (default: all). Use queue slugs'),
make_option(
'--verboseescalation',
action='store_true',
default=False,
help='Display a list of dates excluded'),
)
def __init__(self):
BaseCommand.__init__(self)
self.address = '127.0.0.1' # TODO - ENVIRONMENT DEPENDANT: change management address
signal.signal(signal.SIGINT, sigint_signal_handler)
#TODO - FUTURE WORKS: add some function for server/crawler address creation;
# now it can be bugged with large numbers of servers/crawlers
self.server_port = max([int(server.address.split(':')[2]) for server in TaskServer.objects.all()]
+ [INIT_SERVER_PORT]) + 1
self.crawler_port = max([int(crawler.address.split(':')[2]) for crawler in Crawler.objects.all()]
+ [INIT_CRAWLER_PORT]) + 1
self.last_scaling = time.time()
self.old_crawlers = [crawler.address for crawler in Crawler.objects.all()]
self.changed = False
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option(
"--queues",
help="Queues to include (default: all). Use queue slugs"),
make_option(
"--verboseescalation",
action="store_true",
default=False,
help="Display a list of dates excluded",
),
)
def make_wsgi_application():
# validate models
s = StringIO()
import django
from django.core.management.base import BaseCommand
django.setup()
cmd = BaseCommand()
import sys
cmd.stdout, cmd.stderr = sys.stdout, sys.stderr
cmd.check()
translation.activate(settings.LANGUAGE_CODE)
if django14:
return get_internal_wsgi_application()
return WSGIHandler()
def add_arguments(self, parser: argparse.ArgumentParser) -> None:
BaseCommand.add_arguments(self, parser)
parser.add_argument('--filename',
action='store',
type=str,
required=True,
help=pgettext('Load OUI IEEE MA-L', 'filename'))
parser.add_argument('--count',
action='store',
type=int,
required=False,
default=1000,
help=pgettext('Load OUI IEEE MA-L',
'items count to insert per batch'))
def __init__(self):
BaseCommand.__init__(self)
self.tracking = []
self.streams = {}
self.first_pass_completed = False
self.verbosity = 1
self._wait_for_db()
self.socialapp = SocialApp.objects.get(pk=1)
os.makedirs(os.path.join(Archive.ARCHIVES_DIR, "raw"), exist_ok=True)
os.makedirs(os.path.join(Archive.ARCHIVES_DIR, "map"), exist_ok=True)
def __init__(self, *args, **kwargs):
BaseCommand.__init__(self, *args, **kwargs)
self.actions = {
"help": "show help information",
ReviewMethod.name: [
ReviewMethod,
],
PracticeMethod.name: [
PracticeMethod,
],
HardMethod.name: [
HardMethod,
],
"dictation": [HardMethod, ReviewMethod],
}
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (make_option('--quiet',
'-q',
default=False,
action='store_true',
help='Hide all command output.'), )
self.option_list += (make_option(
'--prod-indexes',
'-p',
default=False,
action='store_true',
help=
'Delete all production databases indexes in CouchDB. e.g. "dmscouch" and "mdtcouch" databases.'
), )
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option(
'--quiet', '-q',
default=False,
action='store_true',
help='Hide all command output.'),
)
self.option_list += (
make_option(
'--prod-indexes', '-p',
default=False,
action='store_true',
help='Delete all production databases indexes in CouchDB. e.g. "dmscouch" and "mdtcouch" databases.'),
)
def test_import_csv_user_basic(self, requests_post, utils_get_landowner):
"""Check if parsing a wellformed CSV works fine"""
cmd = BaseCommand()
grp = Group(name='theuser')
grp.save()
ubi = UserImporter(cmd, './legacy/tests/basic-users-mapping.csv')
self.assertEqual(User.objects.count(), 4)
def validate_models(self):
s = io.StringIO()
try:
from django.core.management.validation import get_validation_errors
except ImportError:
import django
from django.core.management.base import BaseCommand
django.setup()
cmd = BaseCommand()
cmd.stdout, cmd.stderr = sys.stdout, sys.stderr
cmd.check()
else:
num_errors = get_validation_errors(s, None)
if num_errors:
raise RuntimeError(
'One or more Django models did not validate:\n{0}'.format(
s.getvalue()))
def test_import_csv_categories_clean(self, requests_post,
utils_get_landowner):
"""Check if clean removes existing objects"""
cat = Category(name='Testcategory')
cat.save()
cmd = BaseCommand()
cati = CategoryImporter(cmd, './legacy/tests/basic-cat.csv')
self.assertEqual(Category.objects.count(), 11)
def add_arguments(self, parser):
BaseCommand.add_arguments(self, parser)
ahelp = ('TODO')
parser.add_argument(
'--reload', action='store_true', dest='use_reloader',
default=False, help=ahelp)
ahelp = (
'The port that the http server should run on. It defaults to '
'5000. This value can be set by the environment variable $PORT.')
parser.add_argument(
'--port', action='store', type=int, dest='port', default=None,
help=ahelp)
parser.add_argument(
'--addr', action='store', type=str, dest='addr', default='0.0.0.0',
help=ahelp)
def create_parser(self, prog_name, subcommand):
# Overwrite parser creation with a LaxOptionParser that will
# ignore arguments it doesn't know, allowing us to pass those
# along to the webassets command.
# Hooking into run_from_argv() would be another thing to try
# if this turns out to be problematic.
parser = BaseCommand.create_parser(self, prog_name, subcommand)
parser.__class__ = LaxOptionParser
return parser
def __init__(self, *args, **kwargs):
class NiceContext(Context):
"""
Same as the real context class, but override some methods in order to gain
nice colored output.
"""
def __init__(s, *args, **kwargs):
kwargs['insert_debug_symbols'] = self.insert_debug_symbols
Context.__init__(s, *args, **kwargs)
def compile_media_callback(s, compress_tag, media_files):
"""
When the compiler notifies us that compiling of this file begins.
"""
if compress_tag:
print self.colored('Compiling media files from', 'yellow'),
print self.colored(' "%s" ' % compress_tag.path, 'green'),
print self.colored(' (line %s, column %s)" ' % (compress_tag.line, compress_tag.column), 'yellow')
else:
print self.colored('Compiling media files', 'yellow')
for m in media_files:
print self.colored(' * %s' % m, 'green')
@contextmanager
def time_operation(self, name):
if self.verbosity >= 3:
begin = time.clock()
yield
end = time.clock()
print termcolor.colored(name, "blue"), (termcolor.colored("%.2fs" % (end - begin), "yellow"))
else:
yield
def compile_media_progress_callback(s, compress_tag, media_file, current, total, file_size):
"""
Print progress of compiling media files.
"""
print self.colored(' (%s / %s):' % (current, total), 'yellow'),
print self.colored(' %s (%s bytes)' % (media_file, file_size), 'green')
self.NiceContext = NiceContext
BaseCommand.__init__(self, *args, **kwargs)
def add_arguments(self, parser):
BaseCommand.add_arguments(self, parser)
ahelp = ('TODO')
parser.add_argument(
'--reload', action='store_true', dest='use_reloader',
default=False, help=ahelp)
ahelp = (
'Port number to listen on. Defaults to the environment variable '
'$PORT (if defined), or 8000.'
)
parser.add_argument(
'--port', action='store', type=int, dest='port', default=None,
help=ahelp)
parser.add_argument(
'--addr', action='store', type=str, dest='addr', default='0.0.0.0',
help='The host/address to bind to (default: 0.0.0.0)')
parser.add_argument(
'--botworker', action='store_true',
dest='botworker', default=False,
help='Run botworker (for browser bots)')
def __init__(self):
BaseCommand.__init__(self)
self.option_list += (
make_option(
'--days', '-d',
help='Days of week (monday, tuesday, etc)'),
make_option(
'--occurrences', '-o',
type='int',
default=1,
help='Occurrences: How many weeks ahead to exclude this day'),
make_option(
'--queues', '-q',
help='Queues to include (default: all). Use queue slugs'),
make_option(
'--escalate-verbosely', '-x',
action='store_true',
default=False,
help='Display a list of dates excluded'),
)
def validate_models(self):
import django
try:
django_setup = django.setup
except AttributeError:
pass
else:
django_setup()
s = io.StringIO()
try:
from django.core.management.validation import get_validation_errors
except ImportError:
from django.core.management.base import BaseCommand
cmd = BaseCommand()
try:
# since django 1.5
from django.core.management.base import OutputWrapper
cmd.stdout = OutputWrapper(sys.stdout)
cmd.stderr = OutputWrapper(sys.stderr)
except ImportError:
cmd.stdout, cmd.stderr = sys.stdout, sys.stderr
cmd.check()
else:
num_errors = get_validation_errors(s, None)
if num_errors:
raise RuntimeError(
'One or more Django models did not validate:\n{0}'.format(
s.getvalue()))
def __init__(self, *args, **kwargs): BaseCommand.__init__(self, *args, **kwargs) # Create cleaners self.readMoreCleaner = HTMLCleaner() self.contentCleaner = HTMLCleaner() for tag in ALLOWED_HTML_TAGS: attributes = [] if tag in TAG_ATTRIBUTES: attributes = TAG_ATTRIBUTES[tag] self.readMoreCleaner.addAllowedTag(tag, attributes) for tag in ALLOWED_HTML_TAGS_STRICT: attributes = [] if tag in TAG_ATTRIBUTES: attributes = TAG_ATTRIBUTES[tag] self.contentCleaner.addAllowedTag(tag, attributes) for key, value in TRANSFORM_TAGS.items(): self.contentCleaner.addTransformationRule(key, value) self.readMoreCleaner.addTransformationRule(key, value)
def __init__(self):
BaseCommand.__init__(self)
