import sys
import os
import glob
from paraview.simple import *

if hasattr(sys, 'argv') and len(sys.argv) > 1:
   namepattern = sys.argv[1]
   files = glob.glob(namepattern)
   cleanup = (len(sys.argv) > 2 and sys.argv[2] == "cleanup")
   for f in files:
      r = LegacyVTKReader( FileNames=[f] )
      w = XMLUnstructuredGridWriter()
      w.DataMode = "Appended"
      w.CompressorType = "ZLib"
      w.FileName = f[:-3]+"vtu"
      w.UpdatePipeline()
      if cleanup:
         os.remove(f)
else:
   print('Usage: pvpython convert_vtk_to_vtu.py "./results/filename_*.vtk" "cleanup"')