75 lines
1.8 KiB
Python
75 lines
1.8 KiB
Python
from zim.plugins import PluginClass
|
|
from zim.command import Command
|
|
from zim.notebook import resolve_notebook, build_notebook
|
|
|
|
usagehelp = '''\
|
|
usage: zim --plugin linklist [OPTIONS] <notebook>
|
|
|
|
--help, -h Print this help
|
|
--existing-only List only pages that exist
|
|
--missing-only List only pages that don't exist
|
|
'''
|
|
|
|
|
|
class LinkListPlugin(PluginClass):
|
|
plugin_info = {
|
|
'name': 'Link List',
|
|
'description': 'List all links throughout the pages',
|
|
'author': 'Gergely Polonkai',
|
|
}
|
|
|
|
|
|
class LinkListCommand(Command):
|
|
options = (
|
|
('help', 'h', 'Print this help text and exit'),
|
|
('existing-only', 'e', 'List only existing pages'),
|
|
('missing-only', 'm', 'List only missing pages'),
|
|
)
|
|
|
|
def parse_options(self, *args):
|
|
Command.parse_options(self, *args)
|
|
|
|
def _all_links(self):
|
|
for page in self.nb.index.walk():
|
|
yield page
|
|
|
|
def run(self):
|
|
if self.opts.get('help'):
|
|
print usagehelp
|
|
|
|
return
|
|
|
|
nbi = None
|
|
|
|
if len(self.args) > 0:
|
|
nbi = resolve_notebook(self.args[0])
|
|
|
|
if nbi is None:
|
|
print("Notebook must be specified!")
|
|
|
|
return
|
|
|
|
self.nb, ns = build_notebook(nbi)
|
|
|
|
listing = None
|
|
|
|
if self.opts.get('missing-only'):
|
|
listing = 'M'
|
|
|
|
if self.opts.get('existing-only'):
|
|
if listing == 'M':
|
|
print """\
|
|
--missing-only and --existing-only are mutually exclusive!"""
|
|
|
|
return
|
|
|
|
listing = 'E'
|
|
|
|
for page in self._all_links():
|
|
exists = page.exists()
|
|
|
|
if listing is None or \
|
|
(listing == 'M' and exists == 0) or \
|
|
(listing == 'E' and exists == 1):
|
|
print "%c %s" % ('E' if exists == 1 else 'M', page.name)
|