commit 8d42b20839c8e4706079a4c6e3a94328c63adf1b · cduck.me/drawsvg

+49
elements.py
···

       1
       1
        
       

     

       2
       2
        
       import math

     

       3
       3
       +
       import os.path

     

       4
       4
       +
       import base64

     

       5
       5
       +
       import warnings

     

       3
       6
        
       

     

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

     

       5
       8
        
       

     
···

       24
       27
        
               outputFile.write(self.TAG_NAME)

     

       25
       28
        
               outputFile.write(' ')

     

       26
       29
        
               for k, v in self.args.items():

     

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

     

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

     

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

     

       29
       33
        
               outputFile.write('/>')

     
···

       42
       46
        
               if isinstance(other, type(self)):

     

       43
       47
        
                   return True

     

       44
       48
        
               return False

     

       49
       49
       +
       

     

       50
       50
       +
       class Image(DrawingBasicElement):

     

       51
       51
       +
           ''' A linked or embedded raster image '''

     

       52
       52
       +
           TAG_NAME = 'image'

     

       53
       53
       +
           MIME_MAP = {

     

       54
       54
       +
               '.bm':  'image/bmp',

     

       55
       55
       +
               '.bmp': 'image/bmp',

     

       56
       56
       +
               '.gif': 'image/gif',

     

       57
       57
       +
               '.jpeg':'image/jpeg',

     

       58
       58
       +
               '.jpg': 'image/jpeg',

     

       59
       59
       +
               '.png': 'image/png',

     

       60
       60
       +
               '.tif': 'image/tiff',

     

       61
       61
       +
               '.tiff':'image/tiff',

     

       62
       62
       +
               '.pdf': 'application/pdf',

     

       63
       63
       +
               '.txt': 'text/plain',

     

       64
       64
       +
           }

     

       65
       65
       +
           MIME_DEFAULT = 'image/png'

     

       66
       66
       +
           def __init__(self, x, y, width, height, path=None, data=None, embed=False,

     

       67
       67
       +
                        mimeType=None, **kwargs):

     

       68
       68
       +
               ''' Specify either the path or data argument.  If path is used and

     

       69
       69
       +
                   embed is True, the image file is embedded in a data URI. '''

     

       70
       70
       +
               if path is None and data is None:

     

       71
       71
       +
                   raise ValueError('Either path or data arguments must be given')

     

       72
       72
       +
               if mimeType is None and path is not None:

     

       73
       73
       +
                       ext = os.path.splitext(path)[1].lower()

     

       74
       74
       +
                       if ext in self.MIME_MAP:

     

       75
       75
       +
                           mimeType = self.MIME_MAP[ext]

     

       76
       76
       +
                       else:

     

       77
       77
       +
                           mimeType = self.MIME_DEFAULT

     

       78
       78
       +
                           warnings.warn('Unknown image file type "{}"'.format(ext), Warning)

     

       79
       79
       +
               if mimeType is None:

     

       80
       80
       +
                   mimeType = self.MIME_DEFAULT

     

       81
       81
       +
                   warnings.warn('Unspecified image type; assuming png'.format(ext), Warning)

     

       82
       82
       +
               if data is not None:

     

       83
       83
       +
                   embed = True

     

       84
       84
       +
               if embed and data is None:

     

       85
       85
       +
                   with open(path, 'rb') as f:

     

       86
       86
       +
                       data = f.read()

     

       87
       87
       +
               if not embed:

     

       88
       88
       +
                   uri = path

     

       89
       89
       +
               else:

     

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

     

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

     

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

     

       93
       93
       +
                                href=uri, **kwargs)

     

       45
       94
        
       

     

       46
       95
        
       class Rectangle(DrawingBasicElement):

     

       47
       96
        
           ''' A rectangle