#!/usr/bin/env python

#see license.txt for license details

"""Generate documentation for reportlab.graphics classes.

Type the following for usage info:

  python graphdocpy.py -h

"""

__version__ = '0.8'

import sys

sys.path.insert(0, '.')

import os, re, types, string, getopt, pickle, copy, time, pprint, traceback

from string import find, join, split, replace, expandtabs, rstrip

import reportlab

from reportlab import rl_config

from docpy import PackageSkeleton0, ModuleSkeleton0

from docpy import DocBuilder0, PdfDocBuilder0, HtmlDocBuilder0

from docpy import htmlescape, htmlrepr, defaultformat, \

     getdoc, reduceDocStringLength

from docpy import makeHtmlSection, makeHtmlSubSection, \

     makeHtmlInlineImage

from reportlab.lib.units import inch, cm

from reportlab.lib.pagesizes import A4

from reportlab.lib import colors

from reportlab.lib.enums import TA_CENTER, TA_LEFT

from reportlab.lib.utils import getStringIO

#from StringIO import StringIO

#getStringIO=StringIO

from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle

from reportlab.pdfgen import canvas

from reportlab.platypus.flowables import Flowable, Spacer

from reportlab.platypus.paragraph import Paragraph

from reportlab.platypus.tableofcontents import TableOfContents

from reportlab.platypus.flowables \

     import Flowable, Preformatted,Spacer, Image, KeepTogether, PageBreak

from reportlab.platypus.xpreformatted import XPreformatted

from reportlab.platypus.frames import Frame

from reportlab.platypus.doctemplate \

     import PageTemplate, BaseDocTemplate

from reportlab.platypus.tables import TableStyle, Table

from reportlab.graphics.shapes import NotImplementedError

import inspect

# Needed to draw Widget/Drawing demos.

from reportlab.graphics.widgetbase import Widget

from reportlab.graphics.shapes import Drawing

from reportlab.graphics import shapes

from reportlab.graphics import renderPDF

VERBOSE = rl_config.verbose

VERIFY = 1

_abstractclasserr_re = re.compile(r'^\s*abstract\s*class\s*(\w+)\s*instantiated',re.I)

####################################################################

#

# Stuff needed for building PDF docs.

#

####################################################################

def mainPageFrame(canvas, doc):

    "The page frame used for all PDF documents."

    canvas.saveState()

    pageNumber = canvas.getPageNumber()

    canvas.line(2*cm, A4[1]-2*cm, A4[0]-2*cm, A4[1]-2*cm)

    canvas.line(2*cm, 2*cm, A4[0]-2*cm, 2*cm)

    if pageNumber > 1:

        canvas.setFont('Times-Roman', 12)

        canvas.drawString(4 * inch, cm, "%d" % pageNumber)

        if hasattr(canvas, 'headerLine'): # hackish

            headerline = string.join(canvas.headerLine, ' \xc2\x8d ')

            canvas.drawString(2*cm, A4[1]-1.75*cm, headerline)

    canvas.setFont('Times-Roman', 8)

    msg = "Generated with docpy. See http://www.reportlab.com!"

    canvas.drawString(2*cm, 1.65*cm, msg)

    canvas.restoreState()

class MyTemplate(BaseDocTemplate):

    "The document template used for all PDF documents."

    _invalidInitArgs = ('pageTemplates',)

    def __init__(self, filename, **kw):

        frame1 = Frame(2.5*cm, 2.5*cm, 15*cm, 25*cm, id='F1')

        self.allowSplitting = 0

        BaseDocTemplate.__init__(self, filename, **kw)

        self.addPageTemplates(PageTemplate('normal', [frame1], mainPageFrame))

    def afterFlowable(self, flowable):

        "Takes care of header line, TOC and outline entries."

        if flowable.__class__.__name__ == 'Paragraph':

            f = flowable

            # Build a list of heading parts.

            # So far, this is the *last* item on the *previous* page...

            if f.style.name[:8] == 'Heading0':

                self.canv.headerLine = [f.text] # hackish

            elif f.style.name[:8] == 'Heading1':

                if len(self.canv.headerLine) == 2:

                    del self.canv.headerLine[-1]

                elif len(self.canv.headerLine) == 3:

                    del self.canv.headerLine[-1]

                    del self.canv.headerLine[-1]

                self.canv.headerLine.append(f.text)

            elif f.style.name[:8] == 'Heading2':

                if len(self.canv.headerLine) == 3:

                    del self.canv.headerLine[-1]

                self.canv.headerLine.append(f.text)

            if f.style.name[:7] == 'Heading':

                # Register TOC entries.

                headLevel = int(f.style.name[7:])

                self.notify('TOCEntry', (headLevel, flowable.getPlainText(), self.page))

                # Add PDF outline entries.

                c = self.canv

                title = f.text

                key = str(hash(f))

                lev = int(f.style.name[7:])

                try:

                    if lev == 0:

                        isClosed = 0

                    else:

                        isClosed = 1

                    c.bookmarkPage(key)

                    c.addOutlineEntry(title, key, level=lev, closed=isClosed)

                    c.showOutline()

                except:

                    if VERBOSE:

                        # AR hacking in exception handlers

                        print 'caught exception in MyTemplate.afterFlowable with heading text %s' % f.text

                        traceback.print_exc()

                    else:

                        pass

####################################################################

#

# Utility functions

#

####################################################################

def indentLevel(line, spacesPerTab=4):

    """Counts the indent levels on the front.

    It is assumed that one tab equals 4 spaces.

    """

    x = 0

    nextTab = 4

    for ch in line:

        if ch == ' ':

            x = x + 1

        elif ch == '\t':

            x = nextTab

            nextTab = x + spacesPerTab

        else:

            return x

assert indentLevel('hello') == 0, 'error in indentLevel'

assert indentLevel(' hello') == 1, 'error in indentLevel'

assert indentLevel('  hello') == 2, 'error in indentLevel'

assert indentLevel('   hello') == 3, 'error in indentLevel'

assert indentLevel('\thello') == 4, 'error in indentLevel'

assert indentLevel(' \thello') == 4, 'error in indentLevel'

assert indentLevel('\t hello') == 5, 'error in indentLevel'

####################################################################

#

# Special-purpose document builders

#

####################################################################

class GraphPdfDocBuilder0(PdfDocBuilder0):

    """A PDF document builder displaying widgets and drawings.

    This generates a PDF file where only methods named 'demo' are

    listed for any class C. If C happens to be a subclass of Widget

    and has a 'demo' method, this method is assumed to generate and

    return a sample widget instance, that is then appended graphi-

    cally to the Platypus story.

    Something similar happens for functions. If their names start

    with 'sample' they are supposed to generate and return a sample

    drawing. This is then taken and appended graphically to the

    Platypus story, as well.

    """

    fileSuffix = '.pdf'

    def begin(self, name='', typ=''):

        styleSheet = getSampleStyleSheet()

        self.code = styleSheet['Code']

        self.bt = styleSheet['BodyText']

        self.story = []

        # Cover page

        t = time.gmtime(time.time())

        timeString = time.strftime("%Y-%m-%d %H:%M", t)

        self.story.append(Paragraph('<font size=18>Documentation for %s "%s"</font>' % (typ, name), self.bt))

        self.story.append(Paragraph('<font size=18>Generated by: graphdocpy.py version %s</font>' %  __version__, self.bt))

        self.story.append(Paragraph('<font size=18>Date generated: %s</font>' % timeString, self.bt))

        self.story.append(Paragraph('<font size=18>Format: PDF</font>', self.bt))

        self.story.append(PageBreak())

        # Table of contents

        toc = TableOfContents()

        self.story.append(toc)

        self.story.append(PageBreak())

    def end(self, fileName=None):

        if fileName:  # overrides output path

            self.outPath = fileName

        elif self.packageName:

            self.outPath = self.packageName + self.fileSuffix

        elif self.skeleton:

            self.outPath = self.skeleton.getModuleName() + self.fileSuffix

        else:

            self.outPath = ''

        if self.outPath:

            doc = MyTemplate(self.outPath)

            doc.multiBuild(self.story)

    def beginModule(self, name, doc, imported):

        story = self.story

        bt = self.bt

        # Defer displaying the module header info to later...

        self.shouldDisplayModule = (name, doc, imported)

        self.hasDisplayedModule = 0

    def endModule(self, name, doc, imported):

        if self.hasDisplayedModule:

            DocBuilder0.endModule(self, name, doc, imported)

    def beginClasses(self, names):

        # Defer displaying the module header info to later...

        if self.shouldDisplayModule:

            self.shouldDisplayClasses = names

    # Skip all methods.

    def beginMethod(self, name, doc, sig):

        pass

    def endMethod(self, name, doc, sig):

        pass

    def beginClass(self, name, doc, bases):

        "Append a graphic demo of a Widget or Drawing at the end of a class."

        if VERBOSE:

            print 'GraphPdfDocBuilder.beginClass(%s...)' % name

        aClass = eval('self.skeleton.moduleSpace.' + name)

        if issubclass(aClass, Widget):

            if self.shouldDisplayModule:

                modName, modDoc, imported = self.shouldDisplayModule

                self.story.append(Paragraph(modName, self.makeHeadingStyle(self.indentLevel-2, 'module')))

                self.story.append(XPreformatted(modDoc, self.bt))

                self.shouldDisplayModule = 0

                self.hasDisplayedModule = 1

                if self.shouldDisplayClasses:

                    self.story.append(Paragraph('Classes', self.makeHeadingStyle(self.indentLevel-1)))

                    self.shouldDisplayClasses = 0

            PdfDocBuilder0.beginClass(self, name, doc, bases)

            self.beginAttributes(aClass)

        elif issubclass(aClass, Drawing):

            if self.shouldDisplayModule:

                modName, modDoc, imported = self.shouldDisplayModule

                self.story.append(Paragraph(modName, self.makeHeadingStyle(self.indentLevel-2, 'module')))

                self.story.append(XPreformatted(modDoc, self.bt))

                self.shouldDisplayModule = 0

                self.hasDisplayedModule = 1

                if self.shouldDisplayClasses:

                    self.story.append(Paragraph('Classes', self.makeHeadingStyle(self.indentLevel-1)))

                    self.shouldDisplayClasses = 0

            PdfDocBuilder0.beginClass(self, name, doc, bases)

    def beginAttributes(self, aClass):

        "Append a list of annotated attributes of a class."

        self.story.append(Paragraph(

            'Public Attributes',

            self.makeHeadingStyle(self.indentLevel+1)))

        map = aClass._attrMap

        if map:

            map = map.items()

            map.sort()

        else:

            map = []

        for name, typ in map:

            if typ != None:

                if hasattr(typ, 'desc'):

                    desc = typ.desc

                else:

                    desc = '<i>%s</i>' % typ.__class__.__name__

            else:

                desc = '<i>None</i>'

            self.story.append(Paragraph(

                "<b>%s</b> %s" % (name, desc), self.bt))

        self.story.append(Paragraph("", self.bt))

    def endClass(self, name, doc, bases):

        "Append a graphic demo of a Widget or Drawing at the end of a class."

        PdfDocBuilder0.endClass(self, name, doc, bases)

        aClass = eval('self.skeleton.moduleSpace.' + name)

        if hasattr(aClass, '_nodoc'):

            pass

        elif issubclass(aClass, Widget):

            try:

                widget = aClass()

            except AssertionError, err:

                if _abstractclasserr_re.match(str(err)): return

                raise

            self.story.append(Spacer(0*cm, 0.5*cm))

            self._showWidgetDemoCode(widget)

            self.story.append(Spacer(0*cm, 0.5*cm))

            self._showWidgetDemo(widget)

            self.story.append(Spacer(0*cm, 0.5*cm))

            self._showWidgetProperties(widget)

            self.story.append(PageBreak())

        elif issubclass(aClass, Drawing):

            drawing = aClass()

            self.story.append(Spacer(0*cm, 0.5*cm))

            self._showDrawingCode(drawing)

            self.story.append(Spacer(0*cm, 0.5*cm))

            self._showDrawingDemo(drawing)

            self.story.append(Spacer(0*cm, 0.5*cm))

    def beginFunctions(self, names):

        srch = string.join(names, ' ')

        if string.find(string.join(names, ' '), ' sample') > -1:

            PdfDocBuilder0.beginFunctions(self, names)

    # Skip non-sample functions.

    def beginFunction(self, name, doc, sig):

        "Skip function for 'uninteresting' names."

        if name[:6] == 'sample':

            PdfDocBuilder0.beginFunction(self, name, doc, sig)

    def endFunction(self, name, doc, sig):

        "Append a drawing to the story for special function names."

        if name[:6] != 'sample':

            return

        if VERBOSE:

            print 'GraphPdfDocBuilder.endFunction(%s...)' % name

        PdfDocBuilder0.endFunction(self, name, doc, sig)

        aFunc = eval('self.skeleton.moduleSpace.' + name)

        drawing = aFunc()

        self.story.append(Spacer(0*cm, 0.5*cm))

        self._showFunctionDemoCode(aFunc)

        self.story.append(Spacer(0*cm, 0.5*cm))

        self._showDrawingDemo(drawing)

        self.story.append(PageBreak())

    def _showFunctionDemoCode(self, function):

        """Show a demo code of the function generating the drawing."""

        # Heading

        self.story.append(Paragraph("<i>Example</i>", self.bt))

        self.story.append(Paragraph("", self.bt))

        # Sample code

        codeSample = inspect.getsource(function)

        self.story.append(Preformatted(codeSample, self.code))

    def _showDrawingCode(self, drawing):

        """Show code of the drawing class."""

        # Heading

        #className = drawing.__class__.__name__

        self.story.append(Paragraph("<i>Example</i>", self.bt))

        # Sample code

        codeSample = inspect.getsource(drawing.__class__.__init__)

        self.story.append(Preformatted(codeSample, self.code))

    def _showDrawingDemo(self, drawing):

        """Show a graphical demo of the drawing."""

        # Add the given drawing to the story.

        # Ignored if no GD rendering available

        # or the demo method does not return a drawing.

        try:

            flo = renderPDF.GraphicsFlowable(drawing)

            self.story.append(Spacer(6,6))

            self.story.append(flo)

            self.story.append(Spacer(6,6))

        except:

            if VERBOSE:

                print 'caught exception in _showDrawingDemo'

                traceback.print_exc()

            else:

                pass

    def _showWidgetDemo(self, widget):

        """Show a graphical demo of the widget."""

        # Get a demo drawing from the widget and add it to the story.

        # Ignored if no GD rendering available

        # or the demo method does not return a drawing.

        try:

            if VERIFY:

                widget.verify()

            drawing = widget.demo()

            flo = renderPDF.GraphicsFlowable(drawing)

            self.story.append(Spacer(6,6))

            self.story.append(flo)

            self.story.append(Spacer(6,6))

        except:

            if VERBOSE:

                print 'caught exception in _showWidgetDemo'

                traceback.print_exc()

            else:

                pass

    def _showWidgetDemoCode(self, widget):

        """Show a demo code of the widget."""

        # Heading

        #className = widget.__class__.__name__

        self.story.append(Paragraph("<i>Example</i>", self.bt))

        # Sample code

        codeSample = inspect.getsource(widget.__class__.demo)

        self.story.append(Preformatted(codeSample, self.code))

    def _showWidgetProperties(self, widget):

        """Dump all properties of a widget."""

        props = widget.getProperties()

        keys = props.keys()

        keys.sort()

        lines = []

        for key in keys:

            value = props[key]

            f = getStringIO()

            pprint.pprint(value, f)

            value = f.getvalue()[:-1]

            valueLines = string.split(value, '\n')

            for i in range(1, len(valueLines)):

                valueLines[i] = ' '*(len(key)+3) + valueLines[i]

            value = string.join(valueLines, '\n')

            lines.append('%s = %s' % (key, value))

        text = join(lines, '\n')

        self.story.append(Paragraph("<i>Properties of Example Widget</i>", self.bt))

        self.story.append(Paragraph("", self.bt))

        self.story.append(Preformatted(text, self.code))

class GraphHtmlDocBuilder0(HtmlDocBuilder0):

    "A class to write the skeleton of a Python source."

    fileSuffix = '.html'

    def beginModule(self, name, doc, imported):

        # Defer displaying the module header info to later...

        self.shouldDisplayModule = (name, doc, imported)

        self.hasDisplayedModule = 0

    def endModule(self, name, doc, imported):

        if self.hasDisplayedModule:

            HtmlDocBuilder0.endModule(self, name, doc, imported)

    def beginClasses(self, names):

        # Defer displaying the module header info to later...

        if self.shouldDisplayModule:

            self.shouldDisplayClasses = names

    # Skip all methods.

    def beginMethod(self, name, doc, sig):

        pass

    def endMethod(self, name, doc, sig):

        pass

    def beginClass(self, name, doc, bases):

        "Append a graphic demo of a widget at the end of a class."

        aClass = eval('self.skeleton.moduleSpace.' + name)

        if issubclass(aClass, Widget):

            if self.shouldDisplayModule:

                modName, modDoc, imported = self.shouldDisplayModule

                self.outLines.append('<H2>%s</H2>' % modName)

                self.outLines.append('<PRE>%s</PRE>' % modDoc)