add support for OCCURS on a field level

copybook/field.py

@@ -6,6 +6,7 @@ def __init__(self,s,lok,toks):
         self.toks = toks
         self.list_of_values = {}
         self.redefine_target = toks.get("redefine_target")
+        self.repeat = int(toks.get("occurs","1"))
         self.length = 0
         self.precision = 0
         self.signed = False

copybook/field_group.py

@@ -44,7 +44,7 @@ def _normalize_reoccurs(self):
         for i in range(childcount,0,-1):
             # we reverse loop since we are changing the list within the loop
             child = self.children[i-1]
-            if type(child)==FieldGroup and child.repeat>1:
+            if child.repeat>1:
                 # deep copy
                 for clone_number in range(child.repeat,1,-1):
                     newchild = copy.deepcopy(child)

copybook/parser.py

@@ -74,6 +74,7 @@
     + Optional(
         "OCCURS"
         + Word(nums)("occurs")
+        + Optional("TIMES")
     )
     + "."
 )
@@ -88,6 +89,11 @@
         Suppress("REDEFINES")
         + Word(alphanums+"-")("redefine_target")
     )
+    + Optional(
+        "OCCURS"
+        + Word(nums)("occurs")
+        + Optional("TIMES")
+    )
     + pic_expr
     + "."
     + ZeroOrMore(

tests/test_parse_string.py

@@ -110,3 +110,15 @@ def test_redefines_field():
     assert result[0].get_total_length()==15
     assert result[2].get_total_length()==10 and result[3].get_total_length()==10
     assert [i.start_pos for i in result]==[0,0,5,5]
+
+def test_occurs_field():
+    test_str = """
+       01 POLICY.
+         05 POLICY-KEY         PIC 9(10).
+         05 ROW-EFFECTIVE-DT   OCCURS 3 TIMES PIC 9(8).
+     """
+    result = copybook.parse_string(test_str).flatten()
+    assert len(result)==5
+    assert [type(i) for i in result]==[copybook.FieldGroup,copybook.Field,copybook.Field,copybook.Field,copybook.Field]
+    assert result[0].get_total_length()==10+8*3
+    assert [i.start_pos for i in result]==[0,0,10,18,26]