from PLC.Sites import Site, Sites
can_update = lambda (field, value): field in \
- ['hostname', 'protocol',
+ ['site_id',
+ 'ip', 'hostname', 'protocol',
'username', 'password',
'model', 'notes']
roles = ['admin', 'pi', 'tech']
- update_fields = dict(filter(can_update, PCU.fields.items()))
+ pcu_fields = dict(filter(can_update, PCU.fields.items()))
accepts = [
PasswordAuth(),
- Mixed(Site.fields['site_id'],
- Site.fields['login_base']),
- PCU.fields['ip'],
- update_fields
+ pcu_fields
]
returns = Parameter(int, 'New pcu_id (> 0) if successful')
object_type = 'PCU'
object_ids = []
- def call(self, auth, site_id_or_login_base, ip, pcu_fields = {}):
+ def call(self, auth, pcu_fields = {}):
pcu_fields = dict(filter(can_update, pcu_fields.items()))
# Get associated site details
- sites = Sites(self.api, [site_id_or_login_base]).values()
+ sites = Sites(self.api, [pcu_fields['site_id']]).values()
if not sites:
raise PLCInvalidArgument, "No such site"
site = sites[0]
raise PLCPermissionDenied, "Not allowed to add a PCU to that site"
pcu = PCU(self.api, pcu_fields)
- pcu['site_id'] = site['site_id']
- pcu['ip'] = ip
pcu.sync()
- self.object_ids = [pcu['pcu_id']]
+
+ self.object_ids = [site['site_id'], pcu['pcu_id']]
return pcu['pcu_id']