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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
# -*- coding: utf-8 -*-
# Copyright (c) 2012 Roberto Alsina y otros.
# Permission is hereby granted, free of charge, to any
# person obtaining a copy of this software and associated
# documentation files (the "Software"), to deal in the
# Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the
# Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice
# shall be included in all copies or substantial portions of
# the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
# KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
# PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS
# OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
# OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
# OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
from __future__ import print_function, unicode_literals
import sys
from operator import attrgetter
from doit.loader import generate_tasks
from doit.cmd_base import TaskLoader
from doit.reporter import ExecutedOnlyReporter
from doit.doit_cmd import DoitMain
from doit.cmd_help import Help as DoitHelp
from doit.cmd_run import Run as DoitRun
from .nikola import Nikola
def main(args):
sys.path.append('')
try:
import conf
if sys.version_info[0] > 2:
from imp import reload as _reload
else:
_reload = reload # NOQA
_reload(conf)
config = conf.__dict__
except ImportError:
config = {}
site = Nikola(**config)
return DoitNikola(site).run(args)
class Help(DoitHelp):
"""show Nikola usage instead of doit """
@staticmethod
def print_usage(cmds):
"""print nikola "usage" (basic help) instructions"""
print("Nikola")
print("Available commands:")
for cmd in sorted(cmds.values(), key=attrgetter('name')):
print(" nikola %s \t\t %s" % (cmd.name, cmd.doc_purpose))
print("")
print(" nikola help show help / reference")
print(" nikola help <command> show command usage")
print(" nikola help <task-name> show task usage")
class Build(DoitRun):
"""expose "run" command as "build" for backward compatibility"""
pass
class NikolaTaskLoader(TaskLoader):
"""custom task loader to get tasks from Nikola instead of dodo.py file"""
def __init__(self, nikola):
self.nikola = nikola
def load_tasks(self, cmd, opt_values, pos_args):
DOIT_CONFIG = {
'reporter': ExecutedOnlyReporter,
'default_tasks': ['render_site'],
}
tasks = generate_tasks('render_site', self.nikola.gen_tasks())
return tasks, DOIT_CONFIG
class DoitNikola(DoitMain):
# overwite help command
DOIT_CMDS = list(DoitMain.DOIT_CMDS) + [Help, Build]
TASK_LOADER = NikolaTaskLoader
def __init__(self, nikola):
self.nikola = nikola
self.task_loader = self.TASK_LOADER(nikola)
def get_commands(self):
# core doit commands
cmds = DoitMain.get_commands(self)
# load nikola commands
for name, cmd in self.nikola.commands.items():
cmds[name] = cmd
return cmds
def run(self, cmd_args):
sub_cmds = self.get_commands()
args = self.process_args(cmd_args)
if len(args) == 0 or args == ["--help"]:
cmd_args = ['help']
args = ['help']
if len(args) == 0 or args[0] not in sub_cmds.keys() or \
args[0] in ('run', 'build'):
# Check for conf.py before launching run
if not self.nikola.configured:
print("This command needs to run inside an "
"existing Nikola site.")
return False
super(DoitNikola, self).run(cmd_args)
|
