def handle(self, *url_lists, **options):
from django.conf import settings
from debug_logging.models import TestRun
from debug_logging.utils import (get_project_name, get_hostname,
get_revision)
verbosity = int(options.get('verbosity', 1))
self.quiet = verbosity < 1
self.verbose = verbosity > 1
# Dtermine if the DBHandler is used
if True in [isinstance(handler, DBHandler) for handler in
LOGGING_CONFIG["LOGGING_HANDLERS"]]:
self.has_dbhandler = True
else:
self.has_dbhandler = False
# Check for a username without a password, or vice versa
if options['username'] and not options['password']:
raise CommandError('If a username is provided, a password must '
'also be provided.')
if options['password'] and not options['username']:
raise CommandError('If a password is provided, a username must '
'also be provided.')
# Create a TestRun object to track this run
filters = {}
panels = settings.DEBUG_TOOLBAR_PANELS
if 'debug_logging.panels.identity.IdentityLoggingPanel' in panels:
filters['project_name'] = get_project_name()
filters['hostname'] = get_hostname()
if 'debug_logging.panels.revision.RevisionLoggingPanel' in panels:
filters['revision'] = get_revision()
if self.has_dbhandler:
# Check to see if there is already a TestRun object open
existing_runs = TestRun.objects.filter(end__isnull=True, **filters)
if existing_runs:
if options['manual_start']:
# If the --manual-start option was specified, error out
# because there is already an open TestRun
raise CommandError('There is already an open TestRun.')
# Otherwise, close it so that we can open a new one
for existing_run in existing_runs:
existing_run.end = datetime.now()
existing_run.save()
if options['manual_end']:
# If the --manual-end option was specified, we can now exit
self.status_update('The TestRun was successfully closed.')
return
if options['manual_end']:
# The --manual-end option was specified, but there was no
# existing run to close.
raise CommandError('There is no open TestRun to end.')
filters['start'] = datetime.now()
test_run = TestRun(**filters)
if options['name']:
test_run.name = options['name']
if options['description']:
test_run.description = options['description']
test_run.save()
if options['manual_start']:
# The TestRun was successfully created
self.status_update('A new TestRun was successfully opened.')
return
urls = []
for url_list in url_lists:
with open(url_list) as f:
urls.extend([l.strip() for l in f.readlines()
if not l.startswith('#')])
if options['sitemap']:
sitemaps = import_from_string(options['sitemap'])
if isinstance(sitemaps, dict):
for sitemap in sitemaps.values():
urls.extend(map(sitemap.location, sitemap.items()))
elif isinstance(sitemaps, Sitemap):
urls.extend(map(sitemaps.location, sitemaps.items()))
else:
raise CommandError(
'Sitemaps should be a Sitemap object or a dict, got %s '
'instead' % type(sitemaps)
)
self.status_update('Beginning debug logging run...')
client = Client()
if options['username'] and options['password']:
client.login(username=options['username'],
password=options['password'])
for url in urls:
try:
response = client.get(url, DJANGO_DEBUG_LOGGING=True)
except KeyboardInterrupt as e:
if self.has_dbhandler:
# Close out the log entry
test_run.end = datetime.now()
test_run.save()
raise CommandError('Debug logging run cancelled.')
except Exception as e:
if self.verbose:
self.status_update('\nSkipped %s because of error: %s'
% (url, e))
continue
if response and response.status_code == 200:
self.status_ticker()
else:
if self.verbose:
try:
self.status_update('\nURL %s responded with code %s'
% (url, response.status_code))
except NameError as e:
self.status_update('\nSkipped %s because of error: %s'
% (url, e))
if self.has_dbhandler:
# Close out the log entry
test_run.end = datetime.now()
test_run.save()
self.status_update('done!\n')
def handle(self, *url_lists, **options):
from django.conf import settings
from debug_logging.models import TestRun
from debug_logging.utils import (get_project_name, get_hostname,
get_revision)
verbosity = int(options.get('verbosity', 1))
self.quiet = verbosity < 1
self.verbose = verbosity > 1
# Check for a username without a password, or vice versa
if options['username'] and not options['password']:
raise CommandError('If a username is provided, a password must '
'also be provided.')
if options['password'] and not options['username']:
raise CommandError('If a password is provided, a username must '
'also be provided.')
# Create a TestRun object to track this run
filters = {}
panels = settings.DEBUG_TOOLBAR_PANELS
if 'debug_logging.panels.identity.IdentityLoggingPanel' in panels:
filters['project_name'] = get_project_name()
filters['hostname'] = get_hostname()
if 'debug_logging.panels.revision.RevisionLoggingPanel' in panels:
filters['revision'] = get_revision()
# Check to see if there is already a TestRun object open
existing_runs = TestRun.objects.filter(end__isnull=True, **filters)
if existing_runs:
if options['manual_start']:
# If the --manual-start option was specified, error out because
# there is already an open TestRun
raise CommandError('There is already an open TestRun.')
# Otherwise, close it so that we can open a new one
for existing_run in existing_runs:
existing_run.end = datetime.now()
existing_run.save()
if options['manual_end']:
# If the --manual-end option was specified, we can now exit
self.status_update('The TestRun was successfully closed.')
return
if options['manual_end']:
# The --manual-end option was specified, but there was no existing
# run to close.
raise CommandError('There is no open TestRun to end.')
filters['start'] = datetime.now()
test_run = TestRun(**filters)
if options['name']:
test_run.name = options['name']
if options['description']:
test_run.description = options['description']
test_run.save()
if options['manual_start']:
# The TestRun was successfully created
self.status_update('A new TestRun was successfully opened.')
return
urls = []
for url_list in url_lists:
with open(url_list) as f:
urls.extend([l.strip() for l in f.readlines()
if not l.startswith('#')])
self.status_update('Beginning debug logging run...')
client = Client()
if options['username'] and options['password']:
client.login(username=options['username'],
password=options['password'])
for url in urls:
try:
response = client.get(url)
except KeyboardInterrupt, e:
# Close out the log entry
test_run.end = datetime.now()
test_run.save()
raise CommandError('Debug logging run cancelled.')
except:
def handle(self, *url_lists, **options):
from django.conf import settings
from debug_logging.models import TestRun
from debug_logging.utils import (get_project_name, get_hostname,
get_revision)
verbosity = int(options.get('verbosity', 1))
self.quiet = verbosity < 1
self.verbose = verbosity > 1
# Dtermine if the DBHandler is used
if True in [
isinstance(handler, DBHandler)
for handler in LOGGING_CONFIG["LOGGING_HANDLERS"]
]:
self.has_dbhandler = True
else:
self.has_dbhandler = False
# Check for a username without a password, or vice versa
if options['username'] and not options['password']:
raise CommandError('If a username is provided, a password must '
'also be provided.')
if options['password'] and not options['username']:
raise CommandError('If a password is provided, a username must '
'also be provided.')
# Create a TestRun object to track this run
filters = {}
panels = settings.DEBUG_TOOLBAR_PANELS
if 'debug_logging.panels.identity.IdentityLoggingPanel' in panels:
filters['project_name'] = get_project_name()
filters['hostname'] = get_hostname()
if 'debug_logging.panels.revision.RevisionLoggingPanel' in panels:
filters['revision'] = get_revision()
if self.has_dbhandler:
# Check to see if there is already a TestRun object open
existing_runs = TestRun.objects.filter(end__isnull=True, **filters)
if existing_runs:
if options['manual_start']:
# If the --manual-start option was specified, error out
# because there is already an open TestRun
raise CommandError('There is already an open TestRun.')
# Otherwise, close it so that we can open a new one
for existing_run in existing_runs:
existing_run.end = datetime.now()
existing_run.save()
if options['manual_end']:
# If the --manual-end option was specified, we can now exit
self.status_update('The TestRun was successfully closed.')
return
if options['manual_end']:
# The --manual-end option was specified, but there was no
# existing run to close.
raise CommandError('There is no open TestRun to end.')
filters['start'] = datetime.now()
test_run = TestRun(**filters)
if options['name']:
test_run.name = options['name']
if options['description']:
test_run.description = options['description']
test_run.save()
if options['manual_start']:
# The TestRun was successfully created
self.status_update('A new TestRun was successfully opened.')
return
urls = []
for url_list in url_lists:
with open(url_list) as f:
urls.extend([
l.strip() for l in f.readlines() if not l.startswith('#')
])
if options['sitemap']:
sitemaps = import_from_string(options['sitemap'])
if isinstance(sitemaps, dict):
for sitemap in sitemaps.values():
urls.extend(map(sitemap.location, sitemap.items()))
elif isinstance(sitemaps, Sitemap):
urls.extend(map(sitemaps.location, sitemaps.items()))
else:
raise CommandError(
'Sitemaps should be a Sitemap object or a dict, got %s '
'instead' % type(sitemaps))
self.status_update('Beginning debug logging run...')
client = Client()
if options['username'] and options['password']:
client.login(username=options['username'],
password=options['password'])
for url in urls:
try:
response = client.get(url, DJANGO_DEBUG_LOGGING=True)
except KeyboardInterrupt as e:
if self.has_dbhandler:
# Close out the log entry
test_run.end = datetime.now()
test_run.save()
raise CommandError('Debug logging run cancelled.')
except Exception as e:
if self.verbose:
self.status_update('\nSkipped %s because of error: %s' %
(url, e))
continue
if response and response.status_code == 200:
self.status_ticker()
else:
if self.verbose:
try:
self.status_update('\nURL %s responded with code %s' %
(url, response.status_code))
except NameError as e:
self.status_update(
'\nSkipped %s because of error: %s' % (url, e))
if self.has_dbhandler:
# Close out the log entry
test_run.end = datetime.now()
test_run.save()
self.status_update('done!\n')