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)