class Graph:
path = None
model = None
def __init__(self, path=None):
if (path):
self.path = path
else:
self.path = os.path.join('.', 'domegraph.sqlite3')
self._open()
def _open(self):
if (self.model is not None):
return self.model
# Add try-catch here
# try:
if (os.path.isfile(os.path.join('.', 'domegraph.sqlite3'))):
print('Reusing new store')
storage = Storage(storage_name=STORAGE_NAME,
name=GRAPH_NAME,
options_string="new='false'")
else:
print('Creating new store')
storage = Storage(storage_name=STORAGE_NAME,
name=GRAPH_NAME,
options_string="new='true'")
self.model = Model(storage)
return self.model
# except Exception:
print('Storage creation failed')
# return None
def _add(self, statements):
for statement in statements:
self.model.append(statement)
self.model.sync()
def addDevice(self, label, actuates, prop_src, ha_name, ha_type):
subject = DOME_DATA[cleanUri(ha_name)]
statements = [
Statement(subject, rdf.type, DOME.Device),
Statement(subject, rdfs.label, label),
Statement(subject, DOME.ha_name, ha_name),
Statement(subject, DOME.ha_type, ha_type)
]
assert (isinstance(prop_src, Node))
if (actuates):
statements.append(Statement(subject, DOME.actuates, prop_src))
else:
statements.append(Statement(subject, DOME.observes, prop_src))
self._add(statements)
return subject
def addProperty(self, label, value, updated, changed):
subject = DOME_DATA['property/' + label + str(uuid.uuid4())]
statements = [
Statement(subject, rdf.type, DOME.Property),
Statement(subject, rdfs.label, label),
Statement(subject, DOME.value, value),
Statement(subject, DOME.last_updated, updated),
Statement(subject, DOME.last_changed, changed)
]
self._add(statements)
return subject
def addAutomation(self, label, trigger, actions, enabled=True):
subject = DOME_DATA['automation/' + label + str(uuid.uuid4())]
statements = [
Statement(subject, rdf.type, DOME.Automation),
Statement(subject, rdfs.label, label),
Statement(subject, DOME.triggeredby, DOME.Trigger),
Statement(subject, DOME.isenabled, enabled)
]
for action in actions:
statements.append(subject, DOME.performs, actions)
self._add(statements)
return subject
def addTrigger(self, label, triggers, conditions, operator):
subject = DOME_DATA['trigger/' + label + str(uuid.uuid4())]
statements = [
Statement(subject, rdf.type, DOME.Trigger),
Statement(subject, rdfs.label, label),
Statement(subject, DOME.operatortype, operator)
]
for trigger in triggers:
statements.append(subject, DOME.hassubtrigger, trigger)
for condition in conditions:
statements.append(subject, DOME.hascondition, condition)
self._add(statements)
return subject
def addCondition(self, label, prop_src, value, operator):
subject = DOME_DATA['condition/' + label + str(uuid.uiid4())]
statements = [
Statement(subject, rdf.type, DOME.Condition),
Statement(subject, rdfs.label, label),
Statement(subject, DOME.observes, prop_src),
Statement(subject, DOME.target, value),
Statement(subject, DOME.operatortype, operator)
]
self._add(statements)
return subject
def addAction(self, label, prop_src, service):
subject = DOME_DATA['action/' + str(uuid.uuid4())]
statements = [
Statement(subject, rdf.type, DOME.Action),
Statement(subject, rdfs.label, label),
Statement(subject, DOME.actuates, prop_src),
Statement(subject, DOME.callservice, command)
]
self._add(statements)
return subject
def addWebProperty(self,
label,
host,
ext_res_src,
ext_prop_src,
poll,
graph=None):
subject = DOME_DATA['webproperty/' + str(uuid.uuid4())]
statements = [
Statement(subject, rdf.type, DOME.WebProperty),
Statement(subject, rdfs.label, label),
Statement(subject, DOME.hostedby, host),
Statement(subject, DOME.resource, ext_res_src),
Statement(subject, DOME.property, ext_prop_src),
Statement(subject, DOME.poll, str(poll))
]
if (graph):
statements.append(Statement(subject, DOME.graphname, graph))
self._add(statements)
return subject
# Removes all existing statements first!!!
def updateStatement(self, src, pred, obj):
for stmt in self.model.find_statements(Statement(src, pred, None)):
del self.model[stmt]
self.model.append(Statement(src, pred, obj))
# More adds to be added'
def getModel(self):
if (not self.model):
self._open()
return self.model
def __del__(self):
del self.model
