diff --git a/ereuse_devicehub/migrations/versions/d7ea9a3b2da1_create_placeholders.py b/ereuse_devicehub/migrations/versions/d7ea9a3b2da1_create_placeholders.py index 9e4a9dbe..15aed6d9 100644 --- a/ereuse_devicehub/migrations/versions/d7ea9a3b2da1_create_placeholders.py +++ b/ereuse_devicehub/migrations/versions/d7ea9a3b2da1_create_placeholders.py @@ -59,6 +59,8 @@ def clone_device(device): if device.binding: return + if device.type == "Battery": + device.size dict_device = copy.copy(device.__dict__) dict_device.pop('_sa_instance_state') dict_device.pop('id', None) @@ -76,7 +78,9 @@ def clone_device(device): new_c = clone_device(c) new_c.parent = new_device - placeholder = Placeholder(device=new_device, binding=device, is_abstract=True, owner_id=device.owner_id) + placeholder = Placeholder( + device=new_device, binding=device, is_abstract=True, owner_id=device.owner_id + ) db.session.add(placeholder) tags = [x for x in device.tags]