add and remove placeholders for snaposhot devices
This commit is contained in:
parent
aed23450df
commit
3a3b403433
|
@ -0,0 +1,221 @@
|
|||
"""Create placeholders
|
||||
|
||||
Revision ID: 2b90b41a556a
|
||||
Revises: 3e3a67f62972
|
||||
Create Date: 2022-07-19 12:17:16.690865
|
||||
|
||||
"""
|
||||
import copy
|
||||
|
||||
from ereuse_devicehub.config import DevicehubConfig
|
||||
from ereuse_devicehub.db import db
|
||||
from ereuse_devicehub.devicehub import Devicehub
|
||||
from ereuse_devicehub.inventory.models import Transfer
|
||||
from ereuse_devicehub.parser.models import PlaceholdersLog
|
||||
from ereuse_devicehub.resources.action.models import (
|
||||
ActionDevice,
|
||||
Allocate,
|
||||
DataWipe,
|
||||
Deallocate,
|
||||
Management,
|
||||
Prepare,
|
||||
Ready,
|
||||
Recycling,
|
||||
Refurbish,
|
||||
ToPrepare,
|
||||
ToRepair,
|
||||
Use,
|
||||
)
|
||||
from ereuse_devicehub.resources.device.models import Computer, Device, Placeholder
|
||||
from ereuse_devicehub.resources.lot.models import LotDevice
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = '2b90b41a556a'
|
||||
down_revision = '3e3a67f62972'
|
||||
branch_labels = None
|
||||
depends_on = None
|
||||
|
||||
|
||||
def init_app():
|
||||
app = Devicehub(inventory=DevicehubConfig.DB_SCHEMA)
|
||||
app.app_context().push()
|
||||
|
||||
|
||||
def clone_computers():
|
||||
for computer in Computer.query.all():
|
||||
clone_device(computer)
|
||||
|
||||
|
||||
def clone_device(device):
|
||||
if device.binding:
|
||||
return
|
||||
|
||||
dict_device = copy.copy(device.__dict__)
|
||||
dict_device.pop('_sa_instance_state')
|
||||
dict_device.pop('id', None)
|
||||
dict_device.pop('devicehub_id', None)
|
||||
dict_device.pop('actions_multiple', None)
|
||||
dict_device.pop('actions_one', None)
|
||||
dict_device.pop('components', None)
|
||||
dict_device.pop('tags', None)
|
||||
dict_device.pop('system_uuid', None)
|
||||
new_device = device.__class__(**dict_device)
|
||||
db.session.add(new_device)
|
||||
|
||||
if hasattr(device, 'components'):
|
||||
for c in device.components:
|
||||
new_c = clone_device(c)
|
||||
new_c.parent = new_device
|
||||
|
||||
placeholder = Placeholder(device=new_device, binding=device)
|
||||
db.session.add(placeholder)
|
||||
|
||||
tags = [x for x in device.tags]
|
||||
for tag in tags:
|
||||
tag.device = new_device
|
||||
|
||||
lots = [x for x in device.lots]
|
||||
for lot in lots:
|
||||
for rel_lot in LotDevice.query.filter_by(lot_id=lot.id, device=device):
|
||||
rel_lot.device = new_device
|
||||
return new_device
|
||||
|
||||
|
||||
def manual_actions():
|
||||
MANUAL_ACTIONS = (
|
||||
Recycling,
|
||||
Use,
|
||||
Refurbish,
|
||||
Management,
|
||||
Allocate,
|
||||
Deallocate,
|
||||
ToPrepare,
|
||||
Prepare,
|
||||
DataWipe,
|
||||
ToRepair,
|
||||
Ready,
|
||||
Transfer,
|
||||
)
|
||||
|
||||
for action in MANUAL_ACTIONS:
|
||||
change_device(action)
|
||||
|
||||
|
||||
def change_device(action):
|
||||
for ac in action.query.all():
|
||||
if hasattr(ac, 'device'):
|
||||
if not ac.device.binding:
|
||||
continue
|
||||
ac.device = ac.device.binding.device
|
||||
|
||||
if hasattr(ac, 'devices'):
|
||||
for act in ActionDevice.query.filter_by(action_id=ac.id):
|
||||
if not act.device.binding:
|
||||
continue
|
||||
act.device = act.device.binding.device
|
||||
|
||||
|
||||
def change_lot():
|
||||
for placeholder in Placeholder.query.all():
|
||||
device = placeholder.device
|
||||
binding = placeholder.binding
|
||||
if not device or not binding:
|
||||
continue
|
||||
lots = [x for x in device.lots]
|
||||
for lot in lots:
|
||||
for rel_lot in LotDevice.query.filter_by(
|
||||
lot_id=lot.id, device_id=device.id
|
||||
):
|
||||
if binding:
|
||||
rel_lot.device_id = binding.id
|
||||
db.session.commit()
|
||||
|
||||
|
||||
def change_tags():
|
||||
for placeholder in Placeholder.query.all():
|
||||
device = placeholder.device
|
||||
binding = placeholder.binding
|
||||
if not device or not binding:
|
||||
continue
|
||||
tags = [x for x in device.tags]
|
||||
for tag in tags:
|
||||
tag.device = binding
|
||||
db.session.commit()
|
||||
|
||||
|
||||
def remove_manual_actions():
|
||||
MANUAL_ACTIONS = (
|
||||
Recycling,
|
||||
Use,
|
||||
Refurbish,
|
||||
Management,
|
||||
Allocate,
|
||||
Deallocate,
|
||||
ToPrepare,
|
||||
Prepare,
|
||||
DataWipe,
|
||||
ToRepair,
|
||||
Ready,
|
||||
Transfer,
|
||||
)
|
||||
|
||||
for action in MANUAL_ACTIONS:
|
||||
remove_change_device(action)
|
||||
|
||||
|
||||
def remove_change_device(action):
|
||||
for ac in action.query.all():
|
||||
if hasattr(ac, 'device'):
|
||||
if not ac.device.placeholder:
|
||||
continue
|
||||
if not ac.device.placeholder.binding:
|
||||
continue
|
||||
ac.device = ac.device.placeholder.binding
|
||||
|
||||
if hasattr(ac, 'devices'):
|
||||
for act in ActionDevice.query.filter_by(action_id=ac.id):
|
||||
if not act.device.placeholder:
|
||||
continue
|
||||
if not act.device.placeholder.binding:
|
||||
continue
|
||||
act.device = act.device.placeholder.binding
|
||||
db.session.commit()
|
||||
|
||||
|
||||
def remove_placeholders():
|
||||
devices = []
|
||||
for placeholder in Placeholder.query.all():
|
||||
device = placeholder.device
|
||||
binding = placeholder.binding
|
||||
if not device or not binding:
|
||||
continue
|
||||
devices.append(placeholder.device.id)
|
||||
|
||||
for dev in Device.query.filter(Device.id.in_(devices)):
|
||||
db.session.delete(dev)
|
||||
|
||||
for placeholder in Placeholder.query.all():
|
||||
device = placeholder.device
|
||||
binding = placeholder.binding
|
||||
if not device or not binding:
|
||||
continue
|
||||
for plog in PlaceholdersLog.query.filter_by(placeholder=placeholder).all():
|
||||
db.session.delete(plog)
|
||||
|
||||
db.session.delete(placeholder)
|
||||
db.session.commit()
|
||||
|
||||
|
||||
def upgrade():
|
||||
init_app()
|
||||
clone_computers()
|
||||
manual_actions()
|
||||
db.session.commit()
|
||||
|
||||
|
||||
def downgrade():
|
||||
init_app()
|
||||
remove_manual_actions()
|
||||
change_lot()
|
||||
change_tags()
|
||||
remove_placeholders()
|
Reference in New Issue