def test_is_event_id(self): """Test to check if a event is in server.""" self.assertTrue(is_event_id('20130110041009'), 'should be event id') self.assertFalse( is_event_id('20130110041090'), 'should not be event id') self.assertFalse(is_event_id('2013'), 'should not be event id') self.assertFalse(is_event_id('AAA'), 'should not be event id')
def test_is_event_id(self): """Test to check if a event is in server.""" self.assertTrue(is_event_id('20130110041009'), 'should be event id') self.assertFalse(is_event_id('20130110041090'), 'should not be event id') self.assertFalse(is_event_id('2013'), 'should not be event id') self.assertFalse(is_event_id('AAA'), 'should not be event id')
def get_list_event_ids(self): """Get all event id indicated by folder in working dir.""" if os.path.exists(self.working_dir): directories = os.listdir(self.working_dir) else: LOGGER.debug( 'Directory %s does not exist, return None' % self.working_dir) return None # Filter the dirs to only contain valid event dirs valid_dirs = [] for directory in directories: if is_event_id(directory): valid_dirs.append(directory) if len(valid_dirs) == 0: raise EmptyShakeDirectoryError( 'The directory %s does not contain any shakemaps.' % self.working_dir) return valid_dirs
def get_list_event_ids(self): """Get all event id indicated by folder in working dir.""" if os.path.exists(self.working_dir): directories = os.listdir(self.working_dir) else: LOGGER.debug('Directory %s does not exist, return None' % self.working_dir) return None # Filter the dirs to only contain valid event dirs valid_dirs = [] for directory in directories: if is_event_id(directory): valid_dirs.append(directory) if len(valid_dirs) == 0: raise EmptyShakeDirectoryError( 'The directory %s does not contain any shakemaps.' % self.working_dir) return valid_dirs
'Usage:\n%s [working_dir] \nor\n%s [working_dir] --list\nor%s ' '[working_dir] --run-all\nor%s ' '[working_dir] [event_id]' % (sys.argv[0], sys.argv[0], sys.argv[0], sys.argv[0])) elif len(sys.argv) == 3: print('The events in the working dir:') working_directory = sys.argv[1] event_option = sys.argv[2] if event_option in '--list': dir_listing = os.listdir(working_directory) for event in dir_listing: print event sys.exit(0) else: print('Processing shakemap %s' % event_option) if is_event_id(event_option): process_event(working_dir=working_directory, event_id=event_option, locale=locale_option) else: print('%s is not a valid event ID' % event_option) else: working_directory = sys.argv[1] event_option = None print('Processing latest shakemap') # noinspection PyBroadException try: process_event(working_dir=working_directory, locale=locale_option) except: # pylint: disable=W0702 LOGGER.exception('Process event failed')
'Usage:\n%s [working_dir] \nor\n%s [working_dir] --list\nor%s ' '[working_dir] --run-all\nor%s ' '[working_dir] [event_id]' % ( sys.argv[0], sys.argv[0], sys.argv[0], sys.argv[0])) elif len(sys.argv) == 3: print('The events in the working dir:') working_directory = sys.argv[1] event_option = sys.argv[2] if event_option in '--list': dir_listing = os.listdir(working_directory) for event in dir_listing: print event sys.exit(0) else: print('Processing shakemap %s' % event_option) if is_event_id(event_option): process_event( working_dir=working_directory, event_id=event_option, locale=locale_option) else: print('%s is not a valid event ID' % event_option) else: working_directory = sys.argv[1] event_option = None print('Processing latest shakemap') # noinspection PyBroadException try: process_event(working_dir=working_directory, locale=locale_option) except: # pylint: disable=W0702 LOGGER.exception('Process event failed')