#!/usr/bin/env python
import os
import sys
import netCDF4
import numpy as np

if __name__ == '__main__':
    if len(sys.argv) != 5:
        print("usage: python forcing_ana_att.py input_file initial_time output_time tothr", file=sys.stderr)
        sys.exit(1)

    if os.path.exists(sys.argv[1]):
        fileName = sys.argv[1]
        initial_time = sys.argv[2]
        output_time = sys.argv[3]
        tothr = np.int32(sys.argv[4])
        try: 
           ioF = netCDF4.Dataset(fileName,"r+", format="NETCDF4")
        except IOError as e:
           print("Could not open file {}: {}".format(fileName, e.args[1]), file=sys.stderr)
           exit(2)
         
        ioF.model_initialization_time = initial_time
        ioF.model_output_valid_time = output_time
        ioF.model_total_valid_times = tothr
        ioF.close()
    else:
        print("Input file not found: {}".format(sys.argv[1]), file=sys.stderr)
        sys.exit(1)