def test_add_time_information(self):
path = os.path.join(os.getcwd(), "test", "MER_RR__2PRACR20071224_131046_000026212064_00296_30410_0000.nc")
try:
dataset = Dataset(path, 'r')
self.assertFalse(hasattr(dataset, 'start_time'))
self.assertFalse(hasattr(dataset, 'stop_time'))
dataset.close()
timeInformationAdder = TimeInformationAdder()
timeInformationAdder.addTimeInformation(path)
dataset = Dataset(path, 'r')
self.assertTrue(hasattr(dataset, 'start_time'))
self.assertTrue(hasattr(dataset, 'stop_time'))
start_time = datetime.datetime.fromtimestamp(mktime(strptime(dataset.start_time, '%Y-%m-%d %H:%M:%S')))
stop_time = datetime.datetime.fromtimestamp(mktime(strptime(dataset.stop_time, '%Y-%m-%d %H:%M:%S')))
expectedStartDate = datetime.datetime(2007, 12, 24, 13, 10, 46)
expectedEndDate = datetime.datetime(2007, 12, 24, 13, 54, 27)
self.assertEquals(expectedStartDate, start_time)
self.assertEquals(expectedEndDate, stop_time)
dataset.close()
finally:
dataset = Dataset(path, mode='a')
del dataset.start_time
del dataset.stop_time
dataset.close()
def main():
tia = TimeInformationAdder()
if not len(sys.argv) == 2:
print('Usage:\n python add_time_information <directory_path>')
exit(-1)
print('starting process...')
directory = sys.argv[1]
for file in os.listdir(directory):
current_file = os.path.join(directory, file)
print("adding time information to '" + current_file + "'.")
tia.addTimeInformation(current_file)
print('...done.')
exit(0)
