updated hierarchy drilldown with level

mdxpy/mdx.py

@@ -462,10 +462,10 @@ def ancestor(member: Union[str, Member], ancestor: int) -> 'MdxHierarchySet':
         return AncestorHierarchySet(member, ancestor)
 
     @staticmethod
-    def drill_down_level(member: Union[str, Member]) -> 'MdxHierarchySet':
+    def drill_down_level(member: Union[str, Member], level) -> 'MdxHierarchySet':
         if isinstance(member, str):
             member = Member.of(member)
-        return DrillDownLevelHierarchySet(member)
+        return DrillDownLevelHierarchySet(member, level)
 
     @staticmethod
     def descendants(member: Union[str, Member], level_or_depth: Union[MdxLevelExpression, int] = None,
@@ -700,12 +700,19 @@ def to_mdx(self) -> str:
 
 class DrillDownLevelHierarchySet(MdxHierarchySet):
 
-    def __init__(self, member: Member):
+    def __init__(self, member: Member, level):
         super(DrillDownLevelHierarchySet, self).__init__(member.dimension, member.hierarchy)
+        self.level = level
         self.member = member
 
     def to_mdx(self) -> str:
-        return f"{{DRILLDOWNLEVEL({{{self.member.unique_name}}})}}"
+        startstring = ''
+        endstring = ''
+        for _ in range(self.level):
+            startstring += 'DRILLDOWNLEVEL('
+            endstring += ')'
+
+        return f"{{{startstring}{{{self.member.unique_name}}}{endstring}}}"
 
 
 class DescendantsHierarchySet(MdxHierarchySet):

test.py

@@ -273,7 +273,7 @@ def test_mdx_hierarchy_set_ancestor(self):
             hierarchy_set.to_mdx())
 
     def test_mdx_hierarchy_set_drill_down_level(self):
-        hierarchy_set = MdxHierarchySet.drill_down_level(Member.of("Dimension", "Element"))
+        hierarchy_set = MdxHierarchySet.drill_down_level(Member.of("Dimension", "Element"), level=1)
 
         self.assertEqual(
             "{DRILLDOWNLEVEL({[dimension].[dimension].[element]})}",