IfcOpenShell/src/ifcopenshell-python/ifcopenshell/api/owner/update_owner_history.py

import time
import ifcopenshell
import ifcopenshell.api.owner.create_owner_history as create_owner_history


class Usecase:
    def __init__(self, file, settings={}):
        self.file = file
        self.settings = ifcopenshell.api.owner.settings.settings
        for key, value in settings.items():
            self.settings[key] = value

    def execute(self):
        if not self.settings["element"].OwnerHistory:
            self.settings["element"].OwnerHistory = create_owner_history.Usecase(self.file, self.settings).execute()
            return
        if len(self.file.get_inverse(self.settings["element"].OwnerHistory)) > 1:
            old_history = self.settings["element"].OwnerHistory
            self.settings["element"].OwnerHistory = self.file.create_entity("IfcOwnerHistory")
            for i, attribute in enumerate(old_history):
                self.settings["element"].OwnerHistory[i] = attribute
        user = self.get_user()
        application = self.get_application()
        self.settings["element"].OwnerHistory.ChangeAction = self.settings["ChangeAction"]
        self.settings["element"].OwnerHistory.LastModifiedDate = int(time.time())
        self.settings["element"].OwnerHistory.LastModifyingUser = user
        self.settings["element"].OwnerHistory.LastModifyingApplication = application
        return self.settings["OwnerHistory"]

    def get_user(self):
        for element in self.file.by_type("IfcPersonAndOrganization"):
            if (
                element.ThePerson == self.settings["person"]
                and element.TheOrganization == self.settings["organisation"]
            ):
                return element
        return self.file.create_entity(
            "IfcPersonAndOrganization",
            **{"ThePerson": self.settings["person"], "TheOrganization": self.settings["organisation"]},
        )

    def get_application(self):
        for element in self.file.by_type("IfcApplication"):
            if element.ApplicationIdentifier == self.settings["ApplicationIdentifier"]:
                return element
        return self.file.create_entity(
            "IfcApplication",
            **{
                "ApplicationDeveloper": self.get_application_organisation(),
                "Version": self.settings["Version"],
                "ApplicationFullName": self.settings["ApplicationFullName"],
                "ApplicationIdentifier": self.settings["ApplicationIdentifier"],
            },
        )

    def get_application_organisation(self):
        return self.file.create_entity(
            "IfcOrganization",
            **{
                "Name": "IfcOpenShell",
                "Description": "IfcOpenShell is an open source (LGPL) software library that helps users and software developers to work with the IFC file format.",
                "Roles": [
                    self.file.create_entity("IfcActorRole", **{"Role": "USERDEFINED", "UserDefinedRole": "CONTRIBUTOR"})
                ],
                "Addresses": [
                    self.file.create_entity(
                        "IfcTelecomAddress",
                        **{
                            "Purpose": "USERDEFINED",
                            "UserDefinedPurpose": "WEBPAGE",
                            "Description": "The main webpage of the software collection.",
                            "WWWHomePageURL": "https://ifcopenshell.org",
                        },
                    ),
                    self.file.create_entity(
                        "IfcTelecomAddress",
                        **{
                            "Purpose": "USERDEFINED",
                            "UserDefinedPurpose": "WEBPAGE",
                            "Description": "The BlenderBIM Add-on webpage of the software collection.",
                            "WWWHomePageURL": "https://blenderbim.org",
                        },
                    ),
                    self.file.create_entity(
                        "IfcTelecomAddress",
                        **{
                            "Purpose": "USERDEFINED",
                            "UserDefinedPurpose": "REPOSITORY",
                            "Description": "The source code repository of the software collection.",
                            "WWWHomePageURL": "https://github.com/IfcOpenShell/IfcOpenShell.git",
                        },
                    ),
                ],
            },
        )