fix #2605

.IntegerValue in lieu get value function that does not apply to WorksetId Class

Author: jmcouffin

extensions/pyRevitTools.extension/pyRevit.tab/Project.panel/ptools.stack/Links.pulldown/Create Workset For Linked Element.pushbutton/script.py

@@ -1,34 +1,39 @@
-from pyrevit import revit, DB, UI
+from pyrevit import revit, DB
 from pyrevit import script
 from pyrevit import forms
-from pyrevit.compat import get_elementid_value_func
 
 logger = script.get_logger()
 
 selection = revit.get_selection()
 
-get_elementid_value = get_elementid_value_func()
-
-linkedModelName = ''
+linked_model_name = ""
 
 if len(selection) > 0:
-    for el in selection:
-        if isinstance(el, DB.RevitLinkInstance):
-            linkedModelName = el.Name.split(':')[0]
-        elif isinstance(el, DB.ImportInstance):
-            linkedModelName = \
-                el.Parameter[DB.BuiltInParameter.IMPORT_SYMBOL_NAME].AsString()
-        if linkedModelName:
-            if not revit.doc.IsWorkshared and revit.doc.CanEnableWorksharing:
-                revit.doc.EnableWorksharing('Shared Levels and Grids',
-                                            'Workset1')
-            with revit.Transaction('Create Workset for linked model'):
+    with revit.Transaction("Create Workset for linked model"):
+        for el in selection:
+            if isinstance(el, DB.RevitLinkInstance):
+                linked_model_name = "ZL_RVT_" + el.Name.split(":")[0].split(".rvt")[0]
+            elif isinstance(el, DB.ImportInstance):
+                linked_model_name = (
+                    "ZL_DWG_"
+                    + el.Parameter[DB.BuiltInParameter.IMPORT_SYMBOL_NAME]
+                    .AsString()
+                    .split(".dwg")[0]
+                )
+            if linked_model_name:
+                if not revit.doc.IsWorkshared and revit.doc.CanEnableWorksharing:
+                    revit.doc.EnableWorksharing("Shared Levels and Grids", "Workset1")
                 try:
-                    newWs = DB.Workset.Create(revit.doc, linkedModelName)
-                    worksetParam = \
-                        el.Parameter[DB.BuiltInParameter.ELEM_PARTITION_PARAM]
-                    worksetParam.Set(get_elementid_value(newWs.Id))
+                    newWs = DB.Workset.Create(revit.doc, linked_model_name)
+                    worksetParam = el.Parameter[
+                        DB.BuiltInParameter.ELEM_PARTITION_PARAM
+                    ]
+                    worksetParam.Set(newWs.Id.IntegerValue)
                 except Exception as e:
-                    print('Workset: {} already exists\nError: {}'.format(linkedModelName,e))
+                    print(
+                        "Workset: {} already exists\nError: {}".format(
+                            linked_model_name, e
+                        )
+                    )
 else:
-    forms.alert('At least one linked element must be selected.')
+    forms.alert("At least one linked element must be selected.")