From c2ba7e31dda6477faca01bb10fdc90494ee11df3 Mon Sep 17 00:00:00 2001 From: Marc Nijdam Date: Fri, 23 Dec 2016 16:18:11 -0800 Subject: [PATCH] Don't have None based relationships --- helium/device_configuration.py | 6 +++--- helium/label.py | 7 ++++--- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/helium/device_configuration.py b/helium/device_configuration.py index 21539a9..24dbcd6 100644 --- a/helium/device_configuration.py +++ b/helium/device_configuration.py @@ -21,7 +21,7 @@ def _resource_type(cls): return 'device-configuration' @classmethod - def create(cls, session, device=None, configuration=None): + def create(cls, session, device=None, configuration=None, **kwargs): """Create a device configuration. Create a device configuration with the given device and @@ -45,7 +45,7 @@ def create(cls, session, device=None, configuration=None): any failure occurs. """ - rels = {} + rels = kwargs.setdefault('relationships', {}) if configuration is not None: rel = build_request_relationship(configuration._resource_type(), configuration.id) @@ -57,7 +57,7 @@ def create(cls, session, device=None, configuration=None): rels['device'] = rel return super(DeviceConfiguration, cls).create(session, - relationships=rels) + **kwargs) def is_loaded(self): """Check is a device configuration is loaded. diff --git a/helium/label.py b/helium/label.py index a814eee..85e3c13 100644 --- a/helium/label.py +++ b/helium/label.py @@ -17,12 +17,13 @@ class Label(Resource): @classmethod def create(cls, session, - attributes=None, sensors=None, relationships=None): + attributes=None, sensors=None, **kwargs): if sensors is not None: - relationships = relationships or {} + relationships = kwargs.setdefault('relationships', {}) sensor_ids = [r.id for r in sensors] relationships['sensor'] = build_request_relationship('sensor', sensor_ids) + return super(Label, cls).create(session, attributes=attributes, - relationships=relationships) + **kwargs)