from mdx import syntax as s
from mdx import MDXSetConstructor, writerContext
from mdx import loadJson, loadMDX

toyCodeBuilder = MDXSetConstructor(memberPath="[Products].[ProductName]", comment="Product Codes")
colorCodeBuilder = MDXSetConstructor(memberPath="[Colors]", comment="Color Codes")

toys = {
  'dolls' : toyCodeBuilder(codeList=[1,2,3], listName="toy dolls")
}
# This creates the mdx set
# {
#   // Product Codes toy dolls
#   [Products].[ProductName].&[1],
#   [Products].[ProductName].&[2],
#   [Products].[ProductName].&[3]
# }

colors = {
  'blue' : colorCodeBuilder(codeList=['blue', 'cyan'], listName="blues")
}
# This creates the mdx set
# {
#   // Color Codes blues
#   [Colors].&[blue],
#   [Colors].&[cyan]
# }


# Loads .mdx files in path
# Each file becomes available as a tuple member with the file name as key
toys = loadMDX(
  "path/to/toys",
  extraMembers = toys, # Add extra members defined by hand
  toyCodeBuilder
)

# Loads .json files in path
# Each file becomes available as a member of a named tuple with the file name as field
colors = loadJSON(
  "path/to/colors",
  extraMembers = colors,  # Add extra members defined by hand
  colorCodeBuilder
)

# The first two arguments are passed automatically by the context manager
@measure
def volume(writer, BI, name, *expressions, **kwargs):

  (toys, colors) = BI
  # Perform some common .mdx operations here...
  writer.write(
    s.createMeasure(s.measureMember(name),
      s.sumExisting('[Measures].[PurchaseCount]', s.intersect(*expressions))
    )
  )

writer = MDXWriter(name='writer', path='./interestingToys') # by default the path is set on config["OUTPUT_FOLDER"]

# Putting it all together
with writerContext(BI=(toys, colors), writer=writer) as (measures, config):

  measures(volume, someOtherMeasure, etc)(
    # the writer and set variables are passed to the measure after the arguments
    'volume of interesting toy purchases',
    s.tuple(toys.dolls, colors.blue),
    s.exclude(
      s.tuple(toys.cars, colors.any),
      s.tuple(toys.cars, colors.red),
    )
  )

# The output will be available at ./interestingToys.mdx

from mdx import config
config["OUTPUT_FOLDER"] = "./path/to/folder"
config["LIST_BASE_PATH"] = "./path/to/folder"
config["TAB_SIZE"] = 2