update member.py

Author: heetrojivadiya

RFEM/BasicObjects/member.py

@@ -1727,6 +1727,63 @@ def CouplingHingeHinge(
         # Add Member to client model
         model.clientModel.service.set_member(clientObject)
 
+    @staticmethod
+    def Spring(no: int = 1,
+               start_node_no: int = 1,
+               end_node_no: int = 2,
+               line: int = None,
+               spring_type: int = None,
+               comment: str = '',
+               params: dict = None,
+               model = Model):
+        """
+        Args:
+            no (int): Member Tag
+            start_node_no (int): Tag of Start Node
+            end_node_no (int): Tag of End Node
+            line (int, optional): Assigned Line
+            spring_type (int, optional): Assign Member Spring Type
+            comment (str, optional): Comment
+            params (dict, optional): Any WS Parameter relevant to the object and its value in form of a dictionary
+            model (RFEM Class, optional): Model to be edited
+        """
+
+        # Client model | Member
+        clientObject = model.clientModel.factory.create('ns0:member')
+
+        # Clears object atributes | Sets all atributes to None
+        clearAttributes(clientObject)
+
+        # Member No.
+        clientObject.no = no
+
+        # Member Type
+        clientObject.type = MemberType.TYPE_SPRING.name
+
+        # Assigned Line number or Node numbers
+        if line is None:
+            clientObject.node_start = start_node_no
+            clientObject.node_end = end_node_no
+        else:
+            clientObject.line = line
+
+        # Spring Type
+        clientObject.member_type_spring = spring_type
+
+        # Comment
+        clientObject.comment = comment
+
+        # Adding optional parameters via dictionary
+        if params:
+            for key in params:
+                clientObject[key] = params[key]
+
+        # Delete None attributes for improved performance
+        deleteEmptyAttributes(clientObject)
+
+        # Add Member to client model
+        model.clientModel.service.set_member(clientObject)
+
     @staticmethod
     def DeleteMember(members_no: str = '1 2', model = Model):
 

RFEM/enums.py

@@ -6,8 +6,8 @@ class MemberType(Enum):
     Member Type
     '''
     TYPE_BEAM, TYPE_BUCKLING, TYPE_CABLE, TYPE_COMPRESSION, TYPE_COUPLING_HINGE_HINGE,\
-    TYPE_COUPLING_HINGE_RIGID, TYPE_COUPLING_RIGID_HINGE, TYPE_COUPLING_RIGID_RIGID,\
-    TYPE_DEFINABLE_STIFFNESS, TYPE_RIB, TYPE_RIGID, TYPE_TENSION, TYPE_TRUSS, TYPE_TRUSS_ONLY_N, TYPE_RESULT_BEAM = range(15)
+    TYPE_COUPLING_HINGE_RIGID, TYPE_COUPLING_RIGID_HINGE, TYPE_COUPLING_RIGID_RIGID, TYPE_SPRING,\
+    TYPE_DEFINABLE_STIFFNESS, TYPE_RIB, TYPE_RIGID, TYPE_TENSION, TYPE_TRUSS, TYPE_TRUSS_ONLY_N, TYPE_RESULT_BEAM = range(16)
 
 
 class NodalSupportType(Enum):

UnitTests/test_member.py

@@ -338,3 +338,23 @@ def test_member_couplingHingeHinge():
 
     assert round(member.analytical_length, 5) == 5.19615
     assert member.type == "TYPE_COUPLING_HINGE_HINGE"
+
+def test_member_spring():
+
+    Model.clientModel.service.delete_all()
+    Model.clientModel.service.begin_modification()
+
+    Material(1, 'S235')
+    Section(1, 'IPE 300', 1)
+
+    Node(1, 0, 0, 0)
+    Node(2, 3, 3, 3)
+
+    Member.Spring(1, 1, 2)
+
+    Model.clientModel.service.finish_modification()
+
+    member = Model.clientModel.service.get_member(1)
+
+    assert round(member.analytical_length, 5) == 5.19615
+    assert member.type == "TYPE_SPRING"