diff --git a/ereuse_devicehub/migrations/versions/93daff872771_add_hash_hid_to_device.py b/ereuse_devicehub/migrations/versions/93daff872771_add_hash_hid_to_device.py index ea1d2037..e274496e 100644 --- a/ereuse_devicehub/migrations/versions/93daff872771_add_hash_hid_to_device.py +++ b/ereuse_devicehub/migrations/versions/93daff872771_add_hash_hid_to_device.py @@ -36,7 +36,11 @@ def upgrade_data(): continue dev_id = d.id chid = hashlib.sha3_256(d.hid.encode('utf-8')).hexdigest() - sql = f"update {get_inv()}.device set chid={chid} where id={dev_id};" + sql = f"update {get_inv()}.device set chid='{chid}' where id={dev_id};" + con.execute(sql) + + sql = f"update {get_inv()}.snapshot set active=true;" + con.execute(sql) def upgrade(): @@ -52,11 +56,19 @@ def upgrade(): schema=f'{get_inv()}', ) + op.add_column( + 'snapshot', + sa.Column('active', sa.Boolean(), default=True, nullable=True), + schema=f'{get_inv()}', + ) + upgrade_data() op.alter_column('computer', 'user_trusts', nullable=False, schema=f'{get_inv()}') + op.alter_column('snapshot', 'active', nullable=False, schema=f'{get_inv()}') def downgrade(): op.drop_column('computer', 'user_trusts', schema=f'{get_inv()}') op.drop_column('device', 'chid', schema=f'{get_inv()}') + op.drop_column('snapshot', 'active', schema=f'{get_inv()}') diff --git a/ereuse_devicehub/resources/action/models.py b/ereuse_devicehub/resources/action/models.py index 7454cac4..5f141570 100644 --- a/ereuse_devicehub/resources/action/models.py +++ b/ereuse_devicehub/resources/action/models.py @@ -678,6 +678,7 @@ class Snapshot(JoinedWithOneDeviceMixin, ActionWithOneDevice): sid = Column(CIText(), nullable=True) settings_version = Column(CIText(), nullable=True) is_server_erase = Column(Boolean(), nullable=True) + active = Column(Boolean(), nullable=True) def get_last_lifetimes(self): """We get the lifetime and serial_number of the first disk"""