Files
speckle-blender/bpy_speckle/operators/commit.py
T
2022-01-07 09:09:29 -08:00

64 lines
1.6 KiB
Python

"""
Commit operators
"""
import bpy
from bpy.props import BoolProperty
from bpy_speckle.functions import _check_speckle_client_user_stream
from bpy_speckle.clients import speckle_clients
class DeleteCommit(bpy.types.Operator):
"""
Delete stream
"""
bl_idname = "speckle.delete_commit"
bl_label = "Delete commit"
bl_options = {"REGISTER", "UNDO"}
bl_description = "Delete active commit permanently"
are_you_sure: BoolProperty(
name="Confirm",
default=False,
)
def draw(self, context):
layout = self.layout
col = layout.column()
col.prop(self, "are_you_sure")
def invoke(self, context, event):
wm = context.window_manager
if len(context.scene.speckle.users) > 0:
return wm.invoke_props_dialog(self)
return {"CANCELLED"}
def execute(self, context):
if not self.are_you_sure:
return {"CANCELLED"}
self.are_you_sure = False
speckle = context.scene.speckle
check = _check_speckle_client_user_stream(context.scene)
if check is None:
return {"CANCELLED"}
user, stream = check
client = speckle_clients[int(context.scene.speckle.active_user)]
stream = user.streams[user.active_stream]
if len(stream.branches) < 1:
return {"CANCELLED"}
branch = stream.branches[int(stream.branch)]
if len(branch.commits) < 1:
return {"CANCELLED"}
commit = branch.commits[int(branch.commit)]
deleted = client.commit.delete(stream_id=stream.id, commit_id=commit.id)
return {"FINISHED"}