made image generation exception safe
This commit is contained in:
13
draw/code_to_image_wrapper.py
Normal file
13
draw/code_to_image_wrapper.py
Normal file
@@ -0,0 +1,13 @@
|
||||
import draw.code_to_image as cti
|
||||
|
||||
class NSD_writer(object):
|
||||
def __init__(self, filepath: str, x_sz: int, y_sz: int) -> None:
|
||||
self.filepath = filepath
|
||||
self.x_sz = x_sz
|
||||
self.y_sz = y_sz
|
||||
|
||||
def __enter__(self):
|
||||
cti.NSD_init(self.x_sz, self.y_sz)
|
||||
|
||||
def __exit__(self, _, __, ___):
|
||||
cti.NSD_save(self.filepath)
|
||||
Reference in New Issue
Block a user