commit 73a7b6d8aa6e4bcf7b8a27d60472624ca59b07c6 · cduck.me/drawsvg

+40 -2
elements.py
···

       3
       3
        
       import os.path

     

       4
       4
        
       import base64

     

       5
       5
        
       import warnings

     

       6
       6
       +
       import xml.sax.saxutils as xml

     

       6
       7
        
       

     

       7
       8
        
       # TODO: Support drawing ellipses without manually using Path

     

       8
       9
        
       

     
···

       20
       21
        
           ''' Base class for SVG drawing elements that are a single node with no

     

       21
       22
        
               child nodes '''

     

       22
       23
        
           TAG_NAME = '_'

     

       24
       24
       +
           hasContent = False

     

       23
       25
        
           def __init__(self, **args):

     

       24
       26
        
               self.args = args

     

       25
       27
        
           def writeSvgElement(self, outputFile):

     
···

       30
       32
        
                   k = k.replace('__', ':')

     

       31
       33
        
                   k = k.replace('_', '-')

     

       32
       34
        
                   outputFile.write('{}="{}" '.format(k,v))

     

       33
       33
       -
               outputFile.write('/>')

     

       35
       35
       +
               if not self.hasContent:

     

       36
       36
       +
                   outputFile.write('/>')

     

       37
       37
       +
               else:

     

       38
       38
       +
                   outputFile.write('>')

     

       39
       39
       +
                   self.writeContent(outputFile)

     

       40
       40
       +
                   outputFile.write('</')

     

       41
       41
       +
                   outputFile.write(self.TAG_NAME)

     

       42
       42
       +
                   outputFile.write('>')

     

       43
       43
       +
           def writeContent(self, outputFile):

     

       44
       44
       +
               ''' Override in a subclass to add data between the start and end

     

       45
       45
       +
                   tags.  This will not be called if hasContent is False. '''

     

       46
       46
       +
               raise RuntimeError('This element has no content')

     

       34
       47
        
           def __eq__(self, other):

     

       35
       48
        
               if isinstance(other, type(self)):

     

       36
       49
        
                   return (self.tagName == other.tagName and

     
···

       90
       103
        
                   encData = base64.b64encode(data).decode()

     

       91
       104
        
                   uri = 'data:{};base64,{}'.format(mimeType, encData)

     

       92
       105
        
               super().__init__(x=x, y=-y-height, width=width, height=height,

     

       93
       93
       -
                                href=uri, **kwargs)

     

       106
       106
       +
                                xlink__href=uri, **kwargs)

     

       107
       107
       +
       

     

       108
       108
       +
       class Text(DrawingBasicElement):

     

       109
       109
       +
           ''' Text

     

       110
       110
       +
       

     

       111
       111
       +
               Additional keyword arguments are output as additional arguments to the

     

       112
       112
       +
               SVG node e.g. fill="red", font_size=20, text_anchor="middle". '''

     

       113
       113
       +
           TAG_NAME = 'text'

     

       114
       114
       +
           hasContent = True

     

       115
       115
       +
           def __init__(self, text, fontSize, x, y, center=False, **kwargs):

     

       116
       116
       +
               if center:

     

       117
       117
       +
                   if 'text_anchor' not in kwargs:

     

       118
       118
       +
                       kwargs['text_anchor'] = 'middle'

     

       119
       119
       +
                   try:

     

       120
       120
       +
                       fontSize = float(fontSize)

     

       121
       121
       +
                       translate = 'translate(0,{})'.format(fontSize*0.5*center)

     

       122
       122
       +
                       if 'transform' in kwargs:

     

       123
       123
       +
                           kwargs['transform'] = translate + ' ' + kwargs['transform']

     

       124
       124
       +
                       else:

     

       125
       125
       +
                           kwargs['transform'] = translate

     

       126
       126
       +
                   except TypeError:

     

       127
       127
       +
                       pass

     

       128
       128
       +
               super().__init__(x=x, y=-y, font_size=fontSize, **kwargs)

     

       129
       129
       +
               self.escapedText = xml.escape(text)

     

       130
       130
       +
           def writeContent(self, outputFile):

     

       131
       131
       +
               outputFile.write(self.escapedText)

     

       94
       132
        
       

     

       95
       133
        
       class Rectangle(DrawingBasicElement):

     

       96
       134
        
           ''' A rectangle