1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465 |
- import sys
- import logging
- import PyTango
- class Workspace(object):
- """A workspace."""
- def __init__(self, device, workspace_id):
- self._device = device
- self._id = workspace_id
- self._path = device.GetWorkspacePath(workspace_id)
- logging.getLogger().info("Create {}".format(self))
- def __enter__(self):
- return self
- def __exit__(self, *exc):
- self.close()
- def __repr__(self):
- return '<Workspace:id={} path={}>'.format(self.id, self._path)
- @property
- def id(self):
- """Unique id of this workspace."""
- return self._id
- @property
- def path(self):
- """Path associated with the workspace"""
- return self._path
- def close(self):
- """Close this workspace."""
- self._device.CloseWorkspace(self.id)
- logging.getLogger().info("Closed {}".format(self))
- class WorkspaceCreator(object):
- """Workspace creation facility."""
- def __init__(self, device_name):
- """
- Args:
- device_name (str): TANGO device name
- """
- try:
- self._device = PyTango.DeviceProxy(device_name)
- except PyTango.DevFailed as e:
- raise IOError("PyTango: {}".format(e[0].desc))
- def create(self, name):
- """Create a new workspace.
- Args:
- name (str): Name of the workspace or dataset
- Returns:
- Workspace: A new workspace object
- """
- workspace_id = self._device.CreateWorkspace(name)
- return Workspace(self._device, workspace_id)
|