1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
import argparse
import logging
import os
import sys
import django
def django_setup():
try:
import patchwork
except:
# try and guess location based on how ansible deploys patchwork
here = os.path.abspath(os.path.dirname(__file__))
patchwork = os.path.join(here, '../project')
if not os.path.exists(patchwork):
sys.exit('patchwork not found in PYTHONPATH or at: %s' % patchwork)
sys.path.append(patchwork)
if 'DJANGO_SETTINGS_MODULE' not in os.environ:
os.environ['DJANGO_SETTINGS_MODULE'] = 'local_settings'
if getattr(django, 'setup', None):
django.setup()
def _init_logging(level):
handler = logging.StreamHandler(stream=sys.stderr)
formatter = logging.Formatter(
'%(asctime)s %(levelname)-5s %(name)s: %(message)s',
datefmt='%H:%M:%S')
handler.setFormatter(formatter)
l = logging.getLogger('')
l.addHandler(handler)
l.setLevel(getattr(logging, level))
class LoggingAction(argparse.Action):
def __call__(self, parser, namespace, value, option_string=None):
log = logging.getLogger()
log.setLevel(getattr(logging, value))
log.debug('logging level set to: %s', value)
def add_logging_arguments(parser, level='WARN'):
_init_logging(level)
parser.add_argument('--log', default=level, action=LoggingAction,
choices=('WARN', 'INFO', 'DEBUG'),
help='Logging level to use. Default=%(default)s')
|