Skip to content

Commit 7a45964

Browse files
committed
Added static asset
1 parent 8d405fd commit 7a45964

1 file changed

Lines changed: 5 additions & 9 deletions

File tree

enginecore/enginecore/model/system_modeler.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -530,7 +530,7 @@ def create_pdu(key, attr, preset_file=os.path.join(os.path.dirname(__file__), 'p
530530
# Outlet-specific OIDs
531531
for k, v in data["outletOIDs"].items():
532532

533-
# For outlet state, outlet asset will need to be created
533+
# For outlet state, Outlet asset will need to be created
534534
if k == "OutletState":
535535
if 'oidDesc' in v:
536536
oid_desc = dict((y,x) for x,y in v["oidDesc"].items())
@@ -561,7 +561,7 @@ def create_pdu(key, attr, preset_file=os.path.join(os.path.dirname(__file__), 'p
561561
props_stm = _get_props_stm({'OID': oid, 'OIDName': k, 'dataType': v['dataType'], 'defaultValue': v['defaultValue']})
562562
query.append("CREATE ({oid_node_name}:OID {{ {props_stm} }})".format(oid_node_name=oid_node_name, props_stm=props_stm))
563563

564-
# set the relationships
564+
# set the relationships (outlet powerd by state oid etc..)
565565
query.append("CREATE (out{})-[:POWERED_BY]->({})".format(out_key, oid_node_name))
566566
query.append("CREATE ({})-[:HAS_STATE_DETAILS]->(oidDesc)".format(oid_node_name))
567567
query.append("CREATE (pdu)-[:HAS_OID]->({})".format(oid_node_name))
@@ -582,13 +582,9 @@ def create_static(key, attr):
582582
raise KeyError('Static asset requires power_consumption')
583583

584584
with graph_ref.get_session() as session:
585-
session.run("\
586-
CREATE (:Asset:StaticAsset { \
587-
name: $name, \
588-
type: 'staticasset', \
589-
key: $key})",
590-
name=attr['name'], key=key)
591-
set_properties(key, attr)
585+
s_attr = ["name", "img_url", "type", "key", "off_delay", "on_delay", "power_consumption", "power_source"]
586+
props_stm = _get_props_stm({**attr, **{'type': 'staticasset', 'key': key}}, supported_attr=s_attr)
587+
session.run("CREATE (:Asset:StaticAsset {{ {} }})".format(props_stm))
592588

593589

594590
def drop_model():

0 commit comments

Comments
 (0)