github.com/pachyderm/pachyderm@v1.13.4/etc/testing/pipeline-build/python/main.py (about)

     1  import os
     2  import sys
     3  import argparse
     4  from leftpad import left_pad
     5  
     6  LENGTH = 4
     7  INPUT_DIRECTORY = "/pfs/in"
     8  OUTPUT_DIRECTORY = "/pfs/out"
     9  
    10  def main():
    11      pad_char = "0" if len(sys.argv) <= 1 else sys.argv[1]
    12      
    13      for fname in os.listdir(INPUT_DIRECTORY):
    14          if os.path.isfile(os.path.join(INPUT_DIRECTORY, fname)):
    15              with open(os.path.join(INPUT_DIRECTORY, fname), "r") as f_in:
    16                  with open(os.path.join(OUTPUT_DIRECTORY, fname), "w") as f_out: 
    17                      f_out.write("{}".format(left_pad(f_in.read(), LENGTH, pad_char)))
    18  
    19  if __name__ == "__main__":
    20      main()