14 lines
424 B
Python
14 lines
424 B
Python
import draw.code_to_image as cti
|
|
|
|
class NSD_writer(object):
|
|
def __init__(self, filepath: str, filename: str, x_sz: int, y_sz: int) -> None:
|
|
self.filepath = filepath
|
|
self.filename = filename
|
|
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, self.filename) |