Run black

src/blenderbim/blenderbim/bim/module/debug/operator.py

@@ -112,6 +112,7 @@ class CreateAllShapes(bpy.types.Operator):
     def execute(self, context):
         self.file = IfcStore.get_file()
         elements = self.file.by_type("IfcElement") + self.file.by_type("IfcSpace")
+
         total = len(elements)
         settings = ifcopenshell.geom.settings()
         failures = []

src/blenderbim/blenderbim/bim/module/drawing/prop.py

@@ -392,6 +392,7 @@ class BIMTextProperties(PropertyGroup):
         name="Font Size",
     )
 
+
 class BIMAssignedProductProperties(PropertyGroup):
     is_editing_product: BoolProperty(name="Is Editing Product", default=False)
     relating_product: PointerProperty(name="Relating Product", type=bpy.types.Object)

src/blenderbim/blenderbim/bim/module/drawing/sheeter.py

@@ -118,7 +118,7 @@ class SheetBuilder:
         sheet_tree = ET.parse(sheet_path)
         sheet_root = sheet_tree.getroot()
 
-        for g in sheet_root.findall('{http://www.w3.org/2000/svg}g'):
+        for g in sheet_root.findall("{http://www.w3.org/2000/svg}g"):
             if g.attrib.get("data-id") == str(reference.id()):
                 sheet_root.remove(g)
                 break

src/blenderbim/blenderbim/bim/module/drawing/svgwriter.py

@@ -81,7 +81,9 @@ class SvgWriter:
     def draw_underlay(self, image):
         self.svg.add(
             self.svg.image(
-                os.path.join("..", "diagrams", os.path.basename(image)), width=self.width, height=self.height,
+                os.path.join("..", "diagrams", os.path.basename(image)),
+                width=self.width,
+                height=self.height,
             )
         )
         return self
@@ -252,7 +254,9 @@ class SvgWriter:
             end = Vector(((x_offset + v1.x), (y_offset - v1.y)))
             vector = end - start
             line = self.svg.add(
-                self.svg.line(start=tuple(start * self.svg_scale), end=tuple(end * self.svg_scale), class_=" ".join(classes))
+                self.svg.line(
+                    start=tuple(start * self.svg_scale), end=tuple(end * self.svg_scale), class_=" ".join(classes)
+                )
             )
             line["stroke-dasharray"] = "12.5, 3, 3, 3"
             axis_tag = tool.Ifc.get_entity(obj).Name
@@ -361,7 +365,9 @@ class SvgWriter:
             end = Vector(((x_offset + v1.x), (y_offset - v1.y)))
             vector = end - start
             line = self.svg.add(
-                self.svg.line(start=tuple(start * self.svg_scale), end=tuple(end * self.svg_scale), class_=" ".join(classes))
+                self.svg.line(
+                    start=tuple(start * self.svg_scale), end=tuple(end * self.svg_scale), class_=" ".join(classes)
+                )
             )
 
     def draw_leader_annotation(self, obj):
@@ -427,7 +433,9 @@ class SvgWriter:
             (symbol_position * self.svg_scale)[1],
         )
 
-        self.svg.add(self.svg.use("#elevation-arrow", insert=tuple(symbol_position * self.svg_scale), transform=transform))
+        self.svg.add(
+            self.svg.use("#elevation-arrow", insert=tuple(symbol_position * self.svg_scale), transform=transform)
+        )
         self.svg.add(self.svg.use("#elevation-tag", insert=tuple(symbol_position * self.svg_scale)))
 
         reference_id, sheet_id = self.get_reference_and_sheet_id_from_annotation(tool.Ifc.get_entity(obj))
@@ -680,7 +688,9 @@ class SvgWriter:
         arc_midpoint = center + ((cog - center).normalized() * radius)
 
         text_position = self.project_point_onto_camera(arc_midpoint)
-        text_position = Vector(((x_offset + text_position.x) * self.svg_scale, (y_offset - text_position.y) * self.svg_scale))
+        text_position = Vector(
+            ((x_offset + text_position.x) * self.svg_scale, (y_offset - text_position.y) * self.svg_scale)
+        )
 
         center_projected = self.project_point_onto_camera(center)
         center_position = Vector(
@@ -704,8 +714,12 @@ class SvgWriter:
 
         # Draw SVG arc, see for details: http://xahlee.info/js/svg_circle_arc.html
         arc_proj_end_pts = [self.project_point_onto_camera(v) for v in arc_end_pts]
-        p1 = Vector(((x_offset + arc_proj_end_pts[0].x) * self.svg_scale, (y_offset - arc_proj_end_pts[0].y) * self.svg_scale))
-        p2 = Vector(((x_offset + arc_proj_end_pts[1].x) * self.svg_scale, (y_offset - arc_proj_end_pts[1].y) * self.svg_scale))
+        p1 = Vector(
+            ((x_offset + arc_proj_end_pts[0].x) * self.svg_scale, (y_offset - arc_proj_end_pts[0].y) * self.svg_scale)
+        )
+        p2 = Vector(
+            ((x_offset + arc_proj_end_pts[1].x) * self.svg_scale, (y_offset - arc_proj_end_pts[1].y) * self.svg_scale)
+        )
         r = radius * self.svg_scale
         # reflex = 1 if angle > math.pi else 0
         reflex = is_reflex
@@ -735,10 +749,16 @@ class SvgWriter:
             path = self.svg.add(self.svg.path(d=d, class_=" ".join(classes)))
 
             p0 = Vector(
-                ((x_offset + projected_points[0].x) * self.svg_scale, (y_offset - projected_points[0].y) * self.svg_scale)
+                (
+                    (x_offset + projected_points[0].x) * self.svg_scale,
+                    (y_offset - projected_points[0].y) * self.svg_scale,
+                )
             )
             p1 = Vector(
-                ((x_offset + projected_points[1].x) * self.svg_scale, (y_offset - projected_points[1].y) * self.svg_scale)
+                (
+                    (x_offset + projected_points[1].x) * self.svg_scale,
+                    (y_offset - projected_points[1].y) * self.svg_scale,
+                )
             )
             text_offset = (p0 - p1).xy.normalized() * 5
             text_position = projected_points[0]
@@ -823,7 +843,9 @@ class SvgWriter:
             text_position = (mid * self.svg_scale) + perpendicular
         rotation = math.degrees(vector.angle_signed(Vector((1, 0))))
         line = self.svg.add(
-            self.svg.line(start=tuple(start * self.svg_scale), end=tuple(end * self.svg_scale), class_=" ".join(classes))
+            self.svg.line(
+                start=tuple(start * self.svg_scale), end=tuple(end * self.svg_scale), class_=" ".join(classes)
+            )
         )
         if text_override is not None:
             text = text_override

src/blenderbim/blenderbim/bim/module/geometry/operator.py

@@ -207,7 +207,10 @@ class UpdateRepresentation(bpy.types.Operator):
         new_representation = ifcopenshell.api.run("geometry.add_representation", self.file, **representation_data)
 
         if tool.Geometry.is_body_representation(new_representation):
-            [tool.Geometry.run_style_add_style(obj=mat) for mat in tool.Geometry.get_object_materials_without_styles(obj)]
+            [
+                tool.Geometry.run_style_add_style(obj=mat)
+                for mat in tool.Geometry.get_object_materials_without_styles(obj)
+            ]
             ifcopenshell.api.run(
                 "style.assign_representation_styles",
                 self.file,

src/blenderbim/blenderbim/bim/module/material/data.py

@@ -84,7 +84,9 @@ class ObjectMaterialData:
 
     @classmethod
     def materials(cls):
-        return sorted([(str(m.id()), m.Name or "Unnamed", "") for m in tool.Ifc.get().by_type("IfcMaterial")], key=lambda x: x[1])
+        return sorted(
+            [(str(m.id()), m.Name or "Unnamed", "") for m in tool.Ifc.get().by_type("IfcMaterial")], key=lambda x: x[1]
+        )
 
     @classmethod
     def type_material(cls):

src/blenderbim/blenderbim/bim/module/material/prop.py

@@ -138,6 +138,4 @@ class BIMObjectMaterialProperties(PropertyGroup):
         items=getParameterizedProfileClasses, name="Parameterized Profile Classes"
     )
 
-    getter_enum = {
-        "material": get_materials
-    }
+    getter_enum = {"material": get_materials}

src/blenderbim/blenderbim/bim/module/model/product.py

@@ -120,9 +120,7 @@ class AddTypeInstance(bpy.types.Operator):
         collection_obj = bpy.data.objects.get(collection.name)
         bpy.ops.bim.assign_class(obj=obj.name, ifc_class=instance_class)
         element = tool.Ifc.get_entity(obj)
-        blenderbim.core.type.assign_type(
-            tool.Ifc, tool.Type, element=element, type=relating_type
-        )
+        blenderbim.core.type.assign_type(tool.Ifc, tool.Type, element=element, type=relating_type)
 
         if building_obj:
             if instance_class in ["IfcWindow", "IfcDoor"]:

src/blenderbim/blenderbim/bim/module/owner/data.py

@@ -253,10 +253,7 @@ class ObjectActorData:
     @classmethod
     def load(cls):
         cls.is_loaded = True
-        cls.data = {
-            "actor": cls.actor(),
-            "actors": cls.actors()
-        }
+        cls.data = {"actor": cls.actor(), "actors": cls.actors()}
 
     @classmethod
     def actor(cls):
@@ -279,9 +276,9 @@ class ObjectActorData:
                     roles = cls.get_roles(actor.TheActor.ThePerson)
                     roles.extend(cls.get_roles(actor.TheActor.TheOrganization))
                 role = ", ".join(roles)
-                results.append({
-                    "id": actor.id(), "name": actor.Name or "Unnamed", "role": role, "ifc_class": actor.is_a()
-                })
+                results.append(
+                    {"id": actor.id(), "name": actor.Name or "Unnamed", "role": role, "ifc_class": actor.is_a()}
+                )
         return results
 
     @classmethod

src/blenderbim/blenderbim/bim/module/pset_template/operator.py

@@ -68,7 +68,7 @@ class AddPsetFile(bpy.types.Operator):
                 "Name": "Name",
                 "Description": "Description",
                 "TemplateType": "PSET_TYPEDRIVENONLY",
-                "ApplicableEntity": "IfcTypeObject"
+                "ApplicableEntity": "IfcTypeObject",
             }
         )
         template.write(filepath)
@@ -200,6 +200,7 @@ class EnableEditingPropTemplate(bpy.types.Operator):
 
         return {"FINISHED"}
 
+
 class DeletePropEnum(bpy.types.Operator):
     bl_idname = "bim.delete_prop_enum"
     bl_label = "delete property enumeration"
@@ -211,6 +212,7 @@ class DeletePropEnum(bpy.types.Operator):
         active_prop.enum_values.remove(self.index)
         return {"FINISHED"}
 
+
 class AddPropEnum(bpy.types.Operator):
     bl_idname = "bim.add_prop_enum"
     bl_label = "add property enumeration"
@@ -380,7 +382,7 @@ class EditPropTemplate(bpy.types.Operator):
                     "Description": props.active_prop_template.description,
                     "PrimaryMeasureType": props.active_prop_template.primary_measure_type,
                     "TemplateType": props.active_prop_template.template_type,
-                    "Enumerators": enumerator
+                    "Enumerators": enumerator,
                 },
             }
         )
@@ -397,8 +399,9 @@ class EditPropTemplate(bpy.types.Operator):
         prop_enum = self.file.create_entity(
             "IFCPROPERTYENUMERATION",
             Name=prop.name,
-                        EnumerationValues=tuple(self.file.create_entity(
-                            prop.primary_measure_type, ev[data_type]) for ev in prop.enum_values)
+            EnumerationValues=tuple(
+                self.file.create_entity(prop.primary_measure_type, ev[data_type]) for ev in prop.enum_values
+            ),
         )
         return prop_enum
 

src/blenderbim/blenderbim/bim/module/pset_template/prop.py

@@ -164,11 +164,8 @@ class PropTemplate(PropertyGroup):
     description: StringProperty(name="Description")
     primary_measure_type: EnumProperty(items=get_primary_measure_type, name="Primary Measure Type")
     template_type: EnumProperty(
-        items=[
-            ("P_SINGLEVALUE","P_SINGLEVALUE",""),
-            ("P_ENUMERATEDVALUE","P_ENUMERATEDVALUE","")
-        ],
-        name="Template Type"
+        items=[("P_SINGLEVALUE", "P_SINGLEVALUE", ""), ("P_ENUMERATEDVALUE", "P_ENUMERATEDVALUE", "")],
+        name="Template Type",
     )
     enum_values: CollectionProperty(type=EnumerationValues)
     getter_enum = {

src/blenderbim/blenderbim/bim/module/pset_template/ui.py

@@ -105,7 +105,9 @@ class BIM_PT_pset_template(Panel):
                     row = layout.row(align=True)
                     split = row.split(factor=0.2)
                     c = split.column()
-                    c.label(text="Enumerations:",)
+                    c.label(
+                        text="Enumerations:",
+                    )
 
                     c = split.column()
                     box = c.box()
@@ -119,7 +121,6 @@ class BIM_PT_pset_template(Panel):
                         op = r.operator("bim.delete_prop_enum", icon="X", text="")
                         op.index = k
 
-                    
             else:
                 row.label(text=prop_template["Name"])
                 row.label(text=prop_template["Description"])

src/blenderbim/blenderbim/bim/module/sequence/operator.py

@@ -624,6 +624,7 @@ class EditTaskTime(bpy.types.Operator):
             if prop.is_null:
                 attributes[prop.name] = None
                 return True
+            # TODO make this parse PT32 as P4D
             attributes[prop.name] = helper.parse_duration(prop.string_value)
             return True
 

src/blenderbim/blenderbim/bim/module/unit/data.py

@@ -37,7 +37,7 @@ class UnitsData:
             "unit_classes": cls.unit_classes(),
             "named_unit_types": cls.named_unit_types(),
             "conversion_unit_types": cls.conversion_unit_types(),
-            "total_units": cls.get_total_units()
+            "total_units": cls.get_total_units(),
         }
         cls.is_loaded = True
 

src/blenderbim/blenderbim/bim/prop.py

@@ -195,6 +195,7 @@ class Attribute(PropertyGroup):
     is_null: BoolProperty(name="Is Null")
     is_optional: BoolProperty(name="Is Optional")
     is_uri: BoolProperty(name="Is Uri", default=False)
+    is_selected: BoolProperty(name="Is Selected", default=False)
 
     def get_value(self):
         if self.is_null:

src/blenderbim/blenderbim/tool/drawing.py

@@ -375,15 +375,11 @@ class Drawing(blenderbim.core.tool.Drawing):
 
     @classmethod
     def open_spreadsheet(cls, uri):
-        cls.open_with_user_command(
-            bpy.context.preferences.addons["blenderbim"].preferences.spreadsheet_command, uri
-        )
+        cls.open_with_user_command(bpy.context.preferences.addons["blenderbim"].preferences.spreadsheet_command, uri)
 
     @classmethod
     def open_svg(cls, uri):
-        cls.open_with_user_command(
-            bpy.context.preferences.addons["blenderbim"].preferences.svg_command, uri
-        )
+        cls.open_with_user_command(bpy.context.preferences.addons["blenderbim"].preferences.svg_command, uri)
 
     @classmethod
     def run_root_assign_class(

src/blenderbim/blenderbim/tool/material.py

@@ -75,7 +75,9 @@ class Material(blenderbim.core.tool.Material):
                     new = props.materials.add()
                     new.ifc_definition_id = material.id()
                     new.name = get_name(material)
-                    new.total_elements = len(ifcopenshell.util.element.get_elements_by_material(tool.Ifc.get(), material))
+                    new.total_elements = len(
+                        ifcopenshell.util.element.get_elements_by_material(tool.Ifc.get(), material)
+                    )
             return
         for material in materials:
             new = props.materials.add()

src/blenderbim/test/core/test_system.py

@@ -125,7 +125,6 @@ class TestHidePorts:
         system.delete_element_objects(["port"]).should_be_called()
         subject.hide_ports(ifc, system, element="element")
 
-
     def test_syncing_locations_if_objects_moved_prior_to_hiding_ports(self, ifc, system):
         ifc.get_object("element").should_be_called().will_return("obj")
         ifc.is_moved("obj").should_be_called().will_return(True)

src/blenderbim/test/tool/test_georeference.py

@@ -163,6 +163,7 @@ class TestGetMapConversionAttributes(NewFile):
             "Scale": 6.0,
         }
 
+
 class TestGetTrueNorthAttributes(NewFile):
     def test_run(self):
         TestImportTrueNorth().test_run()
@@ -185,18 +186,18 @@ class TestDisableEditing(NewFile):
 
 class TestSetCoordinates(NewFile):
     def test_run(self):
-        subject.set_coordinates("input", [1., 2., 3.])
+        subject.set_coordinates("input", [1.0, 2.0, 3.0])
         assert bpy.context.scene.BIMGeoreferenceProperties.coordinate_input == "1.0,2.0,3.0"
-        subject.set_coordinates("output", [4., 5., 6.])
+        subject.set_coordinates("output", [4.0, 5.0, 6.0])
         assert bpy.context.scene.BIMGeoreferenceProperties.coordinate_output == "4.0,5.0,6.0"
 
 
 class TestGetCoordinates(NewFile):
     def test_run(self):
         bpy.context.scene.BIMGeoreferenceProperties.coordinate_input = "1.0,2.0,3.0"
-        assert subject.get_coordinates("input") == [1., 2., 3.]
+        assert subject.get_coordinates("input") == [1.0, 2.0, 3.0]
         bpy.context.scene.BIMGeoreferenceProperties.coordinate_output = "4.0,5.0,6.0"
-        assert subject.get_coordinates("output") == [4., 5., 6.]
+        assert subject.get_coordinates("output") == [4.0, 5.0, 6.0]
 
 
 class TestGetCursorLocation(NewFile):
@@ -207,8 +208,8 @@ class TestGetCursorLocation(NewFile):
         ifcopenshell.api.run("context.add_context", ifc, context_type="Model")
         unit = ifcopenshell.api.run("unit.add_si_unit", ifc, unit_type="LENGTHUNIT", prefix="MILLI", name="METRE")
         ifcopenshell.api.run("unit.assign_unit", ifc, units=[unit])
-        bpy.context.scene.cursor.location = (1., 2., 3.)
-        assert subject.get_cursor_location() == [1000., 2000., 3000.]
+        bpy.context.scene.cursor.location = (1.0, 2.0, 3.0)
+        assert subject.get_cursor_location() == [1000.0, 2000.0, 3000.0]
 
 
 class TestSetCursorLocation(NewFile):
@@ -219,8 +220,8 @@ class TestSetCursorLocation(NewFile):
         ifcopenshell.api.run("context.add_context", ifc, context_type="Model")
         unit = ifcopenshell.api.run("unit.add_si_unit", ifc, unit_type="LENGTHUNIT", prefix="MILLI", name="METRE")
         ifcopenshell.api.run("unit.assign_unit", ifc, units=[unit])
-        subject.set_cursor_location([1000., 2000., 3000.])
-        assert list(bpy.context.scene.cursor.location) == [1., 2., 3.]
+        subject.set_cursor_location([1000.0, 2000.0, 3000.0])
+        assert list(bpy.context.scene.cursor.location) == [1.0, 2.0, 3.0]
 
 
 class TestSetIfcTrueNorth(NewFile):
@@ -258,13 +259,13 @@ class TestGetMapConversion(NewFile):
 
 class TestXyz2Enh(NewFile):
     def test_run(self):
-        assert subject.xyz2enh([0., 0., 0.], None) == [0., 0., 0.]
+        assert subject.xyz2enh([0.0, 0.0, 0.0], None) == [0.0, 0.0, 0.0]
 
     def test_using_the_blender_offset(self):
         props = bpy.context.scene.BIMGeoreferenceProperties
         props.has_blender_offset = True
         props.blender_eastings = "1.0"
-        assert subject.xyz2enh([0., 0., 0.], None) == (1., 0., 0.)
+        assert subject.xyz2enh([0.0, 0.0, 0.0], None) == (1.0, 0.0, 0.0)
 
     def test_using_the_map_conversion(self):
         ifc = ifcopenshell.file()
@@ -274,7 +275,7 @@ class TestXyz2Enh(NewFile):
         ifcopenshell.api.run("georeference.add_georeferencing", ifc)
         map_conversion = ifc.by_type("IfcMapConversion")[0]
         map_conversion.Eastings = 1.0
-        assert subject.xyz2enh([0., 0., 0.], map_conversion) == (1., 0., 0.)
+        assert subject.xyz2enh([0.0, 0.0, 0.0], map_conversion) == (1.0, 0.0, 0.0)
 
     def test_applying_both_blender_offset_and_map_conversion(self):
         props = bpy.context.scene.BIMGeoreferenceProperties
@@ -287,18 +288,18 @@ class TestXyz2Enh(NewFile):
         ifcopenshell.api.run("georeference.add_georeferencing", ifc)
         map_conversion = ifc.by_type("IfcMapConversion")[0]
         map_conversion.Northings = 1.0
-        assert subject.xyz2enh([0., 0., 0.], map_conversion) == (1., 1., 0.)
+        assert subject.xyz2enh([0.0, 0.0, 0.0], map_conversion) == (1.0, 1.0, 0.0)
 
 
 class TestEnh2Xyz(NewFile):
     def test_run(self):
-        assert subject.enh2xyz([0., 0., 0.], None) == [0., 0., 0.]
+        assert subject.enh2xyz([0.0, 0.0, 0.0], None) == [0.0, 0.0, 0.0]
 
     def test_using_the_blender_offset(self):
         props = bpy.context.scene.BIMGeoreferenceProperties
         props.has_blender_offset = True
         props.blender_eastings = "1.0"
-        assert subject.enh2xyz([0., 0., 0.], None) == (-1., 0., 0.)
+        assert subject.enh2xyz([0.0, 0.0, 0.0], None) == (-1.0, 0.0, 0.0)
 
     def test_using_the_map_conversion(self):
         ifc = ifcopenshell.file()
@@ -308,7 +309,7 @@ class TestEnh2Xyz(NewFile):
         ifcopenshell.api.run("georeference.add_georeferencing", ifc)
         map_conversion = ifc.by_type("IfcMapConversion")[0]
         map_conversion.Eastings = 1.0
-        assert subject.enh2xyz([0., 0., 0.], map_conversion) == (-1., 0., 0.)
+        assert subject.enh2xyz([0.0, 0.0, 0.0], map_conversion) == (-1.0, 0.0, 0.0)
 
     def test_applying_both_blender_offset_and_map_conversion(self):
         props = bpy.context.scene.BIMGeoreferenceProperties
@@ -321,7 +322,7 @@ class TestEnh2Xyz(NewFile):
         ifcopenshell.api.run("georeference.add_georeferencing", ifc)
         map_conversion = ifc.by_type("IfcMapConversion")[0]
         map_conversion.Northings = 1.0
-        assert subject.enh2xyz([0., 0., 0.], map_conversion) == (-1., -1., 0.)
+        assert subject.enh2xyz([0.0, 0.0, 0.0], map_conversion) == (-1.0, -1.0, 0.0)
 
 
 class TestSetIfcGridNorth(NewFile):