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')
