commit 0602a1d8292dc011bc2deb669670e53c24befd44 · cduck.me/drawsvg

__init__.py

···

       41
       41
        
       ```

     

       42
       42
        
       '''

     

       43
       43
        
       

     

       44
       44
       +
       from .raster import Raster

     

       44
       45
        
       from .drawing import Drawing

     

       45
       46
        
       from .elements import *

     

       46
       47

drawing.py

···

       1
       1
        
       

     

       2
       2
        
       from io import StringIO

     

       3
       3
        
       

     

       4
       4
       +
       from . import Raster

     

       4
       5
        
       from . import elements as elementsModule

     

       5
       6
        
       

     

       6
       7
        
       

     
···

       94
       95
        
           def saveSvg(self, fname):

     

       95
       96
        
               with open(fname, 'w') as f:

     

       96
       97
        
                   self.asSvg(outputFile=f)

     

       98
       98
       +
           def savePng(self, fname):

     

       99
       99
       +
               self.rasterize(toFile=fname)

     

       100
       100
       +
           def rasterize(self, toFile=None):

     

       101
       101
       +
               if toFile:

     

       102
       102
       +
                   return Raster.fromSvgToFile(self.asSvg(), toFile)

     

       103
       103
       +
               else:

     

       104
       104
       +
                   return Raster.fromSvg(self.asSvg())

     

       97
       105
        
           def _repr_svg_(self):

     

       98
       106
        
               ''' Display in Jupyter notebook '''

     

       99
       107
        
               return self.asSvg()

missing.py

···

       1
       1
       +
       

     

       2
       2
       +
       class MissingModule:

     

       3
       3
       +
           def __init__(self, errorMsg):

     

       4
       4
       +
               self.errorMsg = errorMsg

     

       5
       5
       +
           def __getattr__(self, name):

     

       6
       6
       +
               raise RuntimeError(self.errorMsg)

     

       7
       7
       +

+43

raster.py

···

       1
       1
       +
       

     

       2
       2
       +
       import io

     

       3
       3
       +
       

     

       4
       4
       +
       try:

     

       5
       5
       +
           import cairosvg

     

       6
       6
       +
       except ImportError:

     

       7
       7
       +
           import warnings

     

       8
       8
       +
           from .missing import MissingModule

     

       9
       9
       +
           msg = 'CairoSVG will need to be install to rasterize images: Install with `pip3 install cairosvg`'

     

       10
       10
       +
           cairosvg = MissingModule(msg)

     

       11
       11
       +
           warnings.warn(msg, RuntimeWarning)

     

       12
       12
       +
       

     

       13
       13
       +
       

     

       14
       14
       +
       class Raster:

     

       15
       15
       +
           def __init__(self, pngData=None, pngFile=None):

     

       16
       16
       +
               self.pngData = pngData

     

       17
       17
       +
               self.pngFile = pngFile

     

       18
       18
       +
           def savePng(self, fname):

     

       19
       19
       +
               with open(fname, 'wb') as f:

     

       20
       20
       +
                   f.write(self.pngData)

     

       21
       21
       +
           @staticmethod

     

       22
       22
       +
           def fromSvg(svgData):

     

       23
       23
       +
               pngData = cairosvg.svg2png(bytestring=svgData)

     

       24
       24
       +
               return Raster(pngData)

     

       25
       25
       +
           @staticmethod

     

       26
       26
       +
           def fromSvgToFile(svgData, outFile):

     

       27
       27
       +
               cairosvg.svg2png(bytestring=svgData, write_to=outFile)

     

       28
       28
       +
               return Raster(None, pngFile=outFile)

     

       29
       29
       +
           def _repr_png_(self):

     

       30
       30
       +
               if self.pngData:

     

       31
       31
       +
                   return self.pngData

     

       32
       32
       +
               elif self.pngFile:

     

       33
       33
       +
                   try:

     

       34
       34
       +
                       with open(self.pngFile, 'rb') as f:

     

       35
       35
       +
                           return f.read()

     

       36
       36
       +
                   except TypeError:

     

       37
       37
       +
                       pass

     

       38
       38
       +
                   try:

     

       39
       39
       +
                       self.pngFile.seek(0)

     

       40
       40
       +
                       return self.pngFile.read()

     

       41
       41
       +
                   except io.UnsupportedOperation:

     

       42
       42
       +
                       pass

     

       43
       43
       +