devscripts/prepare_manpage.py

   1 from __future__ import unicode_literals
   2
   3 import io
   4 import optparse
   5 import os.path
   6 import re
   7
   8 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
   9 README_FILE = os.path.join(ROOT_DIR, 'README.md')
  10
  11 PREFIX = '''%YOUTUBE-DL(1)
  12
  13 # NAME
  14
  15 youtube\-dl \- download videos from youtube.com or other video platforms
  16
  17 # SYNOPSIS
  18
  19 **youtube-dl** \[OPTIONS\] URL [URL...]
  20
  21 '''
  22
  23
  24 def main():
  25     parser = optparse.OptionParser(usage='%prog OUTFILE.md')
  26     options, args = parser.parse_args()
  27     if len(args) != 1:
  28         parser.error('Expected an output filename')
  29
  30     outfile, = args
  31
  32     with io.open(README_FILE, encoding='utf-8') as f:
  33         readme = f.read()
  34
  35     readme = re.sub(r'(?s)^.*?(?=# DESCRIPTION)', '', readme)
  36     readme = re.sub(r'\s+youtube-dl \[OPTIONS\] URL \[URL\.\.\.\]', '', readme)
  37     readme = PREFIX + readme
  38
  39     readme = filter_options(readme)
  40
  41     with io.open(outfile, 'w', encoding='utf-8') as outf:
  42         outf.write(readme)
  43
  44
  45 def filter_options(readme):
  46     ret = ''
  47     in_options = False
  48     for line in readme.split('\n'):
  49         if line.startswith('# '):
  50             if line[2:].startswith('OPTIONS'):
  51                 in_options = True
  52             else:
  53                 in_options = False
  54
  55         if in_options:
  56             if line.lstrip().startswith('-'):
  57                 option, description = re.split(r'\s{2,}', line.lstrip())
  58                 split_option = option.split(' ')
  59
  60                 if not split_option[-1].startswith('-'):  # metavar
  61                     option = ' '.join(split_option[:-1] + ['*%s*' % split_option[-1]])
  62
  63                 # Pandoc's definition_lists. See http://pandoc.org/README.html
  64                 # for more information.
  65                 ret += '\n%s\n:   %s\n' % (option, description)
  66             else:
  67                 ret += line.lstrip() + '\n'
  68         else:
  69             ret += line + '\n'
  70
  71     return ret
  72
  73 if __name__ == '__main__':
  74     main()