def step( self ) :
error = None
# print "uuid '" + self.command_line.uuid_property + "'"
properties = {}
broken_properties = []
if self.command_line.uuid_property == "" or "(process." in self.command_line.uuid_property : # If you request a process property from goobi that doesn't exist, it will return the name of what you asked for. e.g. "(process.uuid)" , therefore if it containts "(process." it must not have been set yet.
# Set a property if not set already.
properties['uuid'] = "urn:uuid:" + str( uuid.uuid4() )
goobi_com = GoobiCommunicate( self.config.goobi.host, self.config.goobi.passcode, self.debug )
success = True
for property, value in properties.iteritems() :
if not goobi_com.addProperty( self.command_line.process_id, property, value ) :
broken_properties.append( str(property) + ":" + str(value) )
success = False
if not success:
error = "Failed to create properties:- " + str(broken_properties)
return error
def closeStep( s ): goobi_com = GoobiCommunicate( s.config.goobi.host, s.config.goobi.passcode, s.debug ) if s.command_line.has( s.commandline_process_id ) : goobi_com.closeStepByProcessId( s.command_line.get( s.commandline_process_id ) ) elif s.command_line.has( s.commandline_step_id ) : goobi_com.closeStep( s.command_line.get( s.commandline_step_id ) ) else: s.info( 'Failed to close this step in ' + s.name + ". Neither " + s.commandline_process_id + " or " + s.commandline_step_id + " were passed into commandline." )
def step( s ): """ Get settings and call communicate """ error = None goobi_com = GoobiCommunicate( s.config.goobi.host, s.config.goobi.passcode, s.debug ) if not goobi_com.addProperty( s.command_line.process_id, s.command_line.property_name, s.command_line.property_value ) : error = "Failed to add exportpath property" return error
def reportToStep( s, message ): """ Pass control back to a previous step """ if s.auto_report_problem: if s.command_line.has( s.commandline_step_id ) : s.info( 'Reporting this problem to "' + s.command_line.get(s.commandline_auto_report_problem_step_name) + "\" step." ) goobi_com = GoobiCommunicate( s.config.goobi.host, s.config.goobi.passcode, s.debug ) goobi_com.reportProblem( s.command_line.get( s.commandline_step_id ), s.command_line.get(s.commandline_auto_report_problem_step_name), message ) else : s.info( 'Failed to report this problem to a previous step. In ' + s.name + " " + s.commandline_step_id + " was not passed into commandline." )
def _com( self ):
self.com = GoobiCommunicate( self.host, self.password_token, self.debugging_on )
class GoobiLogger():
"""
Use GoobiCommunicate to log messages to Goobi and to any other python like log you pass in.
"""
log_format = "{date} PID {process_id:0>8} ({level}) - {message}"
goobi_log_format = "PID {process_id:0>8} ({level}) - {message}"
debugging_on = False
type_info = 'info'
type_debug = 'debug'
type_warning = 'warning'
type_error = 'error'
type_critical = 'critical'
type_user = '******'
def __init__( self,
host,
password_token,
process_id,
pyLogger=None,
use_goobi_communication = True,
intervals_between_alive_logs = 60*10):
self.host = host
self.password_token = password_token
self.use_goobi_communication = use_goobi_communication
if self.use_goobi_communication: self._com()
self.process_id = process_id
self.pyLogger = pyLogger
self.last_alive_timestamp = time.time()
# Interval should be configurable
self.intervals_between_alive_logs = intervals_between_alive_logs
def getLogger( self ):
"""
Return initial logger
"""
return self.pyLogger
def debugging( self ) :
self.debugging_on = True
if self.use_goobi_communication:
self._com()
# Use this one if you only want to print with a specific interval,
# e.g. to inform user that process is a alive
def alive(self,message):
if ((time.time() - self.last_alive_timestamp) >=
self.intervals_between_alive_logs):
alive_msg = 'Script is still alive and processing: {0}'.format(message)
self.info(alive_msg)
self.last_alive_timestamp = time.time()
# logger like interface to goobi.
def info( self, message ) :
return self._log( 'info', message )
def debug( self, message ) :
return self._log( 'debug', message )
def exception( self, message ) :
self._pyLog('exception', message)
return self._log( 'error', str(message) )
def warning( self, message ): # Not an actual goobi message but here to complete the set of logging functions.
return self._log( 'warning' , message )
def error( self, message ) :
return self._log( 'error', message )
def critical( self, message ) :
return self._log( 'critical', message )
def user( self, message ): # Not part of the python logging but here for completeness.
return self._log( 'user', message )
def _com( self ):
self.com = GoobiCommunicate( self.host, self.password_token, self.debugging_on )
def _log( self, level, message ):
self._pyLog( level, "(PID" + str(self.process_id).zfill(8) +") " + message )
dt = datetime.datetime.utcnow().isoformat()
formatted_message = self.log_format.format(date=dt,
process_id=self.process_id,
level=level,
message=message )
goobi_message = self.goobi_log_format.format(process_id=self.process_id,
level=level,
message=message )
#if self.debugging_on:
# print(formatted_message)
if level == 'warning':
level = 'info'
elif level == 'critical' :
level = 'error'
elif level == 'exception' :
level = 'error'
# Push the error out to stderr, this will cause Goobi to pause the step if the script is an automatic one.
if level == 'error' :
sys.stderr.write( "stderr: " + formatted_message + "\n" )
if (not (level == 'debug') or
(level == 'debug' and self.debugging_on)):
if self.use_goobi_communication:
return self.com.addToProcessLog(level,goobi_message,
self.process_id )
def _pyLog( self, level, message ):
if self.pyLogger != None:
if level == 'warning' :
self.pyLogger.warning( message )
elif level == 'debug' :
self.pyLogger.debug( message )
elif level == 'error' :
self.pyLogger.error( message )
elif level == 'critical' :
self.pyLogger.critical( message )
elif level == 'info':
self.pyLogger.info( message )
elif level == 'exception':
self.pyLogger.exception( message )
elif level == 'user' :
self.pyLogger.info( "(Goobi User level) " + message )
def __init__( self ) :
# Default names for essential config
self.cli_process_id_arg = "process_id"
self.cli_step_id_arg = "step_id"
self.cli_auto_complete_arg = 'auto_complete'
self.cli_auto_report_problem_arg = "auto_report_problem"
self.cli_detach_arg = "detach"
self.cli_debug_arg = "debug"
self.cli_step_name_arg = 'step_name'
self.config_general_section = 'general'
self.config_goobi_section = 'goobi'
self.name = ""
self.system_config_path = ''
self.config_path = ''
self.config_main_section = "" # Info for this particular step
# General section and goobi section must always be present
# These are placed in a system specific config file
self.essential_system_config_sections = set ([self.config_general_section,
self.config_goobi_section])
# Config is specific for workflow scripts
self.essential_config_sections = set( [] )
# Process id must always be present
self.essential_commandlines = {self.cli_process_id_arg: "number"}
self.command_line = None
self.config = None
self.glogger = None
self.glogger_handlers = {}
self.debug = False
#TODO: add this to config and load it in this init.
self.print_debug = False # Whether to "print" debug msgs
self.auto_complete = False
self.detach = False
# Run setup for specific workflow script
self.setup()
# Update
self.essential_config_sections.update( [self.config_main_section] )
# We need to make sure we have a full path to our config file
#if self.cli_config_path_arg not in self.essential_commandlines.keys() :
# self.essential_commandlines[self.cli_config_path_arg] = "file"
#
# Get command line parameters (want to pass process_id to log if we have it)
self.command_line, error_command_line = \
self.getCommandLine( must_have=self.essential_commandlines )
# Get process id
self.process_id = self.command_line.get(self.cli_process_id_arg)
# Load system configuration information
if self.system_config_path == '':
if not self.command_line.has("system_config_path"):
self.system_config_path = '/opt/digiverso/goobi/scripts/kb/workflows/system/config.ini'
else:
self.system_config_path = self.command_line.system_config_path
self.getConfig(self.system_config_path,
must_have=self.essential_system_config_sections )
# Load config specific for step
if self.command_line.has("config_path"):
# root is the one above the folder "goobi" where goobi_step.py
# is located
cwd = os.path.dirname(os.path.realpath(__file__))
root = os.path.split(cwd)[0]
config_path = self.command_line.get('config_path')
alt_config_path = os.path.join(root,config_path)
if (os.path.exists(config_path) and os.path.isfile(config_path)):
self.config_path = config_path
elif (os.path.exists(alt_config_path) and
os.path.isfile(alt_config_path)):
# config_path is a relative path from current working dir
self.config_path = alt_config_path
else:
error = ('config_path from command line does not exist '
'or is not a valid file. Neither {0} or {1}.')
error = error.format(config_path,alt_config_path)
raise IOError(error)
if not self.config_path == '':
self.info_message('config_path: '+self.config_path)
self.getConfig(self.config_path,
must_have=self.essential_config_sections )
if self.command_line.has(self.cli_step_name_arg) and self.name == '':
self.name = self.command_line.get(self.cli_step_name_arg)
#
# Are we debugging?
self.debug = self.debugging()
if self.command_line and self.command_line.has(self.cli_debug_arg) :
self.debug = not ( self.command_line.debug.lower() == "false" )
# Override config setting at commandline. (if it says anything but false turn it on.)
if self.debug:
print(self.name + ": Debugging ON")
# Create out logger
logger_name = str(self.name).replace(' ','').lower()
self.glogger, error = self.getLoggingSystems(self.config,
self.config_main_section,
self.command_line,
self.debug,
logger_name + "_logger")
if error:
self.exit( error,self.glogger )
self.goobi_com = GoobiCommunicate(self.config.goobi.host,
self.config.goobi.passcode,
self.debug,
process_id = self.process_id
)
#
# Check Commandline parameters
if error_command_line:
self.exit( error_command_line,self.glogger )
#
# Use auto_complete if script needs to output to goobi.
# Goobi won't self complete automatic task if script outputs to goobi.
if self.command_line.has(self.cli_auto_complete_arg):
self.auto_complete = \
(self.command_line.auto_complete.lower() == "true" )
if self.command_line.has( self.cli_detach_arg ) :
self.detach = ( self.command_line.detach.lower() == "true" )
# If self.cli_auto_report_problem_arg is sit in command_line the value
# is the step name that should be reported back to.
if self.command_line.has(self.cli_auto_report_problem_arg):
self.auto_report_problem = self.command_line.get(
self.cli_auto_report_problem_arg )
else:
self.auto_report_problem = None
#
# Pass message back to Goobi to say everything looks fine and start process.
update_message = "Basic checks complete, "
update_message += 'beginning main process of step "' + str(self.name) + '" - '
update_message += 'DETACH:' + ( "ON" if self.detach else "OFF" )
update_message += ', AUTO-COMPLETE:' + ( "ON" if self.auto_complete else "OFF" ) # if successful
update_message += ', REPORT-PROBLEM:' + ( "ON" if self.auto_report_problem else "OFF" ) # if unsuccessful
update_message += ', DEBUG:' + ( "ON" if self.debug else "OFF" ) + "."
self.debug_message( update_message )
class Step( object ):
"""
Base class for all steps.
This :
Checks config
Checks commandline
Creates loggers - goobi, file and email.
Additional commandlines:
detach - if true the step will detach from goobi after the basic
checks are done, the file continues to be run but Goobi
no longer watches it or moves to another step.
auto_complete - if true and the step completes fine, the step will
be closed.
auto_report_problem - auto return to another step if this step
fails. Equal to the stepname (step_id={stepid} is also needed
to report to a previous step)
debug - override config debug value to display additional
information and output more information.
First define the setup(s) function.
give the step a name,
s.name="my step name"
Then specify the default config section for this step,
s.config_main_section='uuid_insert'
Then specify the set of essential config sections this step needs to
run, s.essential_config_sections=set( ["sec1","sec2"] )
Then specify the dict of essential commandlin values with a type
(number, file, folder or string),
s.essential_commandlines = {"process_id" : "number",
"metafile" : "file",
"tiff_path" : "folder",
"step_name" : "string" }
Now put your code in the step() function.
You can acess self.commandline and self.config.
Use error(), warning(), info() and debug() to log to the logger.
You can use other commandline parameters for the process id and step
id by overwriting cli_process_id_arg and cli_step_id_arg in the
child class
"""
# Types for commandlines
type_string = "string"
type_number = "number"
type_file = "file"
type_folder = "folder"
type_ignore = "ignore"
# TODO: Switch types to these: (Use with Step.CLTYPE.STRING
class CLTYPE:
STRING = "string"
NUMBER = "number"
FILE = "file"
FOLDER = "folder"
IGNORE = "ignore"
__metaclass__ = ABCMeta
@abstractmethod
def setup(self,_): pass
@abstractmethod
def step(self,_): pass
def __init__( self ) :
# Default names for essential config
self.cli_process_id_arg = "process_id"
self.cli_step_id_arg = "step_id"
self.cli_auto_complete_arg = 'auto_complete'
self.cli_auto_report_problem_arg = "auto_report_problem"
self.cli_detach_arg = "detach"
self.cli_debug_arg = "debug"
self.cli_step_name_arg = 'step_name'
self.config_general_section = 'general'
self.config_goobi_section = 'goobi'
self.name = ""
self.system_config_path = ''
self.config_path = ''
self.config_main_section = "" # Info for this particular step
# General section and goobi section must always be present
# These are placed in a system specific config file
self.essential_system_config_sections = set ([self.config_general_section,
self.config_goobi_section])
# Config is specific for workflow scripts
self.essential_config_sections = set( [] )
# Process id must always be present
self.essential_commandlines = {self.cli_process_id_arg: "number"}
self.command_line = None
self.config = None
self.glogger = None
self.glogger_handlers = {}
self.debug = False
#TODO: add this to config and load it in this init.
self.print_debug = False # Whether to "print" debug msgs
self.auto_complete = False
self.detach = False
# Run setup for specific workflow script
self.setup()
# Update
self.essential_config_sections.update( [self.config_main_section] )
# We need to make sure we have a full path to our config file
#if self.cli_config_path_arg not in self.essential_commandlines.keys() :
# self.essential_commandlines[self.cli_config_path_arg] = "file"
#
# Get command line parameters (want to pass process_id to log if we have it)
self.command_line, error_command_line = \
self.getCommandLine( must_have=self.essential_commandlines )
# Get process id
self.process_id = self.command_line.get(self.cli_process_id_arg)
# Load system configuration information
if self.system_config_path == '':
if not self.command_line.has("system_config_path"):
self.system_config_path = '/opt/digiverso/goobi/scripts/kb/workflows/system/config.ini'
else:
self.system_config_path = self.command_line.system_config_path
self.getConfig(self.system_config_path,
must_have=self.essential_system_config_sections )
# Load config specific for step
if self.command_line.has("config_path"):
# root is the one above the folder "goobi" where goobi_step.py
# is located
cwd = os.path.dirname(os.path.realpath(__file__))
root = os.path.split(cwd)[0]
config_path = self.command_line.get('config_path')
alt_config_path = os.path.join(root,config_path)
if (os.path.exists(config_path) and os.path.isfile(config_path)):
self.config_path = config_path
elif (os.path.exists(alt_config_path) and
os.path.isfile(alt_config_path)):
# config_path is a relative path from current working dir
self.config_path = alt_config_path
else:
error = ('config_path from command line does not exist '
'or is not a valid file. Neither {0} or {1}.')
error = error.format(config_path,alt_config_path)
raise IOError(error)
if not self.config_path == '':
self.info_message('config_path: '+self.config_path)
self.getConfig(self.config_path,
must_have=self.essential_config_sections )
if self.command_line.has(self.cli_step_name_arg) and self.name == '':
self.name = self.command_line.get(self.cli_step_name_arg)
#
# Are we debugging?
self.debug = self.debugging()
if self.command_line and self.command_line.has(self.cli_debug_arg) :
self.debug = not ( self.command_line.debug.lower() == "false" )
# Override config setting at commandline. (if it says anything but false turn it on.)
if self.debug:
print(self.name + ": Debugging ON")
# Create out logger
logger_name = str(self.name).replace(' ','').lower()
self.glogger, error = self.getLoggingSystems(self.config,
self.config_main_section,
self.command_line,
self.debug,
logger_name + "_logger")
if error:
self.exit( error,self.glogger )
self.goobi_com = GoobiCommunicate(self.config.goobi.host,
self.config.goobi.passcode,
self.debug,
process_id = self.process_id
)
#
# Check Commandline parameters
if error_command_line:
self.exit( error_command_line,self.glogger )
#
# Use auto_complete if script needs to output to goobi.
# Goobi won't self complete automatic task if script outputs to goobi.
if self.command_line.has(self.cli_auto_complete_arg):
self.auto_complete = \
(self.command_line.auto_complete.lower() == "true" )
if self.command_line.has( self.cli_detach_arg ) :
self.detach = ( self.command_line.detach.lower() == "true" )
# If self.cli_auto_report_problem_arg is sit in command_line the value
# is the step name that should be reported back to.
if self.command_line.has(self.cli_auto_report_problem_arg):
self.auto_report_problem = self.command_line.get(
self.cli_auto_report_problem_arg )
else:
self.auto_report_problem = None
#
# Pass message back to Goobi to say everything looks fine and start process.
update_message = "Basic checks complete, "
update_message += 'beginning main process of step "' + str(self.name) + '" - '
update_message += 'DETACH:' + ( "ON" if self.detach else "OFF" )
update_message += ', AUTO-COMPLETE:' + ( "ON" if self.auto_complete else "OFF" ) # if successful
update_message += ', REPORT-PROBLEM:' + ( "ON" if self.auto_report_problem else "OFF" ) # if unsuccessful
update_message += ', DEBUG:' + ( "ON" if self.debug else "OFF" ) + "."
self.debug_message( update_message )
def begin(self) :
if self.detach:
# Detach from goobi.
self.detachSelf()
error = None
try:
error = self.step()
except Exception as e:
try:
emsg = str(e)
trace = traceback.format_exc()
self.error_message('Exception occured in ' + str(self.name) +\
' :- ' + emsg + ". Trace: " + trace )
except TypeError:
self.error_message('Exception occured in ' + str(self.name) +\
' :- ' + str(e) + ". Trace: " + trace )
except:
self.error_message( 'Exception occured in ' + str(self.name) )
raise e
if not error :
self.debug_message(str(self.name) +' afsluttet korrekt.')
if self.auto_complete:
self.closeStep()
else:
if self.auto_report_problem:
error_msg = ('Error occured in "{0}". Sending task back to {1}')
error_msg = error_msg.format(self.name,self.auto_report_problem)
self.error_message(error_msg)
self.error_message(str(error))
self.reportToStep( error )
else:
error_msg = ('"{0}" failed. Error message: "{1}"')
error_msg = error_msg.format(self.name,error)
self.error_message(error_msg)
return (error == None)
def reportToStep( self, message ):
"""
Pass control back to a previous step
"""
if self.auto_report_problem is None:
return
try:
step_id = self.command_line.get(self.cli_step_id_arg)
except KeyError:
msg = 'Failed to report this problem to a previous step.'+\
' In "' + str(self.name) + '" ' + self.cli_step_id_arg +\
' was not passed into command line.'
self.error(msg)
raise KeyError(msg)
prev_step_name = self.auto_report_problem
self.goobi_com.reportToPrevStep(step_id,prev_step_name,message)
def closeStep(self):
'''
TODO: Document method
'''
if self.command_line.has(self.cli_step_id_arg): # Prefer this one
self.goobi_com.closeStep( self.command_line.get( self.cli_step_id_arg ) )
#elif self.command_line.has(self.cli_process_id_arg) :
# self.goobi_com.closeStepByProcessId( self.command_line.get( self.cli_process_id_arg ) )
else:
self.info( 'Failed to close this step in "' + str(self.name) + '". Neither ' + self.cli_process_id_arg + " or " + self.cli_step_id_arg + " were passed into commandline." )
def exit( self, message,log=None ) :
# TODO: make this method a nice exit
''' Nice exit '''
msg = ('Exit being called with args: .\nMessage is {0}.\nLog is {1}')
msg = msg.format(str(message),str(log))
if log.__class__.__name__ == 'Logger':
log.error( message )
else:
self.glogger.error(message)
sys.exit(1)
def getConfig( self, config_file, must_have ) :
# Add self.config so new config config can be added to old.
config = ConfigReader(config_file,
self.config,
overwrite_sections=True)
does_not_have_sections = []
for section in must_have:
if len(section) > 0 and not config.hasSection( section ):
print("section {0} not found".format(section))
does_not_have_sections.append( section )
if does_not_have_sections:
error = "Error: Config file does not contain sections:- " + ", ".join( does_not_have_sections )
raise ValueError(error)
self.config = config
def getConfigSection(self, section, config=None):
"""
Return section as dictionary if it exists, otherwise raise key error
"""
value = None
if config == None:
config = self.config
if section in self.config.config.sections():
value = dict(self.config.config.items(section))
if value is None:
error = 'Section {0} not defined in config file.'
error = error.format(section)
raise KeyError(error)
else:
return value
def getConfigItem( self, key, config=None, section=None ):
'''
If section is given, return value for key
If section is not give, sheck for value in the main section, then
fall back to general section
If key can't be found, raise KeyError
'''
value = None
if config == None:
config = self.config
if not section is None:
if config.hasItem(section, key):
value = config.item(section, key )
else:
if config.hasItem( self.config_main_section, key ) :
value = config.item( self.config_main_section, key )
elif config.hasItem( self.config_general_section, key ) :
value = config.item( self.config_general_section, key )
if value is None:
error = '{0} not defined in section {1} in config file.'
error = error.format(key,section)
raise KeyError(error)
else:
return value
def getSetting(self,var_name,var_type=None,conf=None,conf_sec=None,default=None):
'''
Get and parse variable from either commmandline or configuration
settings. The variable is casted to a type, e.g. a int, string og bool.
:param var_name: name of variable to retrieve
:param var_type: what to cast the variable to. Per default the variable is a string
Use: int,float or bool
:param conf: (optional) a configuration to use (alternative to self.config)
:param conf_sec: (optional) the section in the cofiguration to locate the variable witin
:param default: (optional) default return value if the variable wasn't found, e.g. False
'''
try:
basestring
except NameError: # python3
basestring = str
ret_val = default
if self.command_line.has(var_name):
if var_type and (var_type == int or var_type == float):
if (self.command_line.get(var_name) == ''):
ret_val = default
else:
ret_val = self.command_line.get(var_name)
else:
ret_val = self.command_line.get(var_name)
else:
try:
ret_val = self.getConfigItem(var_name, config=conf,
section=conf_sec)
except KeyError as e:
if default is not None: ret_val = default # default may be 0,False or ''
else:
error = '"{0}" not defined in commandline or in config file.'
error = error.format(var_name)
raise KeyError(error)
if var_type is not None:
if var_type == float:
ret_val = float(ret_val)
elif var_type == int:
ret_val = int(ret_val)
elif var_type == bool:
if isinstance(ret_val,basestring):
ret_val = (ret_val.lower() == 'true')
return ret_val
def debugging( self) :
debug = self.getConfigItem( "debug", self.config )
if not debug:
debug = False
return debug
def getCommandLine( self, must_have ) :
command_line = CommandLine()
error = None
# check for process_id first as we use it to log - if we have problem report that first
if not command_line.has( self.cli_process_id_arg ):
error = "Error: Command line does not contain a " + self.cli_process_id_arg + "=<VALUE>. This is needed for logging."
else :
value = command_line.get( self.cli_process_id_arg )
parameter_error = self.checkParameter( self.cli_process_id_arg, 'number', value )
if parameter_error:
# See if we can recover (so we can report the error )
result = re.search( r'^\d+', value )
if result:
process_id = result.group()
command_line.set( self.cli_process_id_arg, process_id )
parameter_error += " Attempted to *guess* correct process_id, found " + str( process_id ) + ". "
error = parameter_error
# Check other commandlines
must_have_keys = must_have.keys()
does_not_have_parameters = []
for parameter in must_have_keys:
if parameter != self.cli_process_id_arg:
if not command_line.has( parameter ):
does_not_have_parameters.append( parameter )
if does_not_have_parameters:
error = (error + " Also, " ) if error else "Error: "
error += "Command line does not contain - " + " ".join( [s + "=<VALUE>" for s in does_not_have_parameters] )
else:
for parameter in must_have_keys:
if parameter != self.cli_process_id_arg :
var_type = must_have[parameter]
value = command_line.get( parameter )
parameter_error = self.checkParameter( parameter, var_type, value )
if parameter_error:
error = (error + " " ) if error else "Error: "
error += parameter_error + " "
return command_line, error
def checkParameter( self, parameter, var_type, value ) :
parameter_error = None
if not value:
parameter_error = 'Error: Parameter "' + parameter + '" can not be empty.'
else:
if var_type == Step.type_folder :
try:
tools.check_folder( value )
except Exception as e:
parameter_error = str(e)
elif var_type == Step.type_file :
try:
tools.check_file( value )
except Exception as e:
parameter_error = str(e)
elif var_type == Step.type_number :
try:
self.checkNumber( value )
except Exception as e:
parameter_error = str(e)
elif var_type == Step.type_string or var_type == Step.type_ignore:
parameter_error = None # anything goes.
else :
parameter_error = "Command line var_type not recognised. Should be set to folder, file, number, string or ignore."
if parameter_error :
parameter_error = 'Error: Parameter "' + parameter + '" - ' + parameter_error
return parameter_error
def checkNumber( self, number ):
error = None
try:
float( number )
except ValueError:
error = 'Number "' + number + '" is not a number.'
return error
def getLogger( self, config, id, debug ) :
logger = logging.getLogger( id )
if debug:
logger.setLevel( logging.DEBUG )
else:
logger.setLevel( logging.INFO )
return logger
def addRotatingLog( self, config, config_main_section, command_line, logger, debug ):
error = None
try:
log_max_bytes = int( self.getConfigItem( "log_max_bytes", config ) )
except ValueError:
log_max_bytes = 50000000
try:
log_backup_count = int( self.getConfigItem( "log_backup_count", config ) )
except ValueError:
log_backup_count = 4
log_file = None
if self.getConfigItem('log'):
log_file = self.getConfigItem('log')
else:
#TODO: Wrong error message
return 'Error: Unable to locate log file: ' + log_file
# Check and create log folder structure
log_folder = os.path.dirname(log_file)
parent_log_folder = os.path.dirname(log_folder)
if not os.path.exists(log_folder):
if os.path.exists(parent_log_folder):
os.mkdir(log_folder)
else:
err = ('Parent folder ({0}) to log folder {1} does not exist. '
'Cannot create logger for log file {2}.')
err = err.format(parent_log_folder,log_folder,log_file)
raise IOError(err)
try:
rotating_logger_handler = logging.handlers.RotatingFileHandler( log_file, maxBytes=log_max_bytes, backupCount=log_backup_count, encoding='utf-8')
# Add Process ID to the log
pid = "unknown"
if command_line.has( self.cli_process_id_arg ) :
pid = str( command_line.get(self.cli_process_id_arg) )
rotating_logger_handler.setFormatter( logging.Formatter( "[PID " + pid + '] %(asctime)s (%(levelname)s) %(message)s') )
if debug :
rotating_logger_handler.setLevel( logging.DEBUG )
else:
rotating_logger_handler.setLevel( logging.INFO )
self.glogger_handlers["rotating"] = rotating_logger_handler
logger.addHandler( rotating_logger_handler )
except IOError:
error = 'Error: Unable to open log file at : ' + log_file
return error
def addEmailLog( self, config, config_main_section, logger ):
log_email = self.getConfigItem( "log_email", config )
if log_email :
log_email_subject = self.getConfigItem( "log_email_subject", config )
if not log_email_subject:
log_email_subject = "Goobi Error " + str(self.name)
email_logger_handler = logging.handlers.SMTPHandler( "smtp.ox.ac.uk", logger.name + "@goobi.bodleian.ox.ac.uk", log_email, log_email_subject)
email_logger_handler.setLevel( logging.WARNING )
self.glogger_handlers["email"] = email_logger_handler
logger.addHandler( email_logger_handler )
def getGoobiLogger( self, config, command_line, logger, debug ):
if command_line.has( self.cli_process_id_arg ):
glogger = GoobiLogger( config.goobi.host, config.goobi.passcode, command_line.get(self.cli_process_id_arg), logger )
if debug :
glogger.debugging()
else:
glogger = logger
return glogger
def getLoggingSystems( self, config, config_main_section, command_line, debug, id):
"""
Creating file, email and goobi logging.
just_file is so we can recreate the file logger after we detach
the prcess, when we detach we need to close all the open files.
The other loggers are unaffected.
"""
use_email = False
if self.getConfigItem('log_use_email'):
use_email = self.getConfigItem('log_use_email')
use_goobi_gui_log = True
if self.getConfigItem('log_use_gui_msg'):
use_goobi_gui_log = self.getConfigItem('log_use_gui_msg')
#
# Create our base logger
logger = self.getLogger( config, id, debug )
#
# Add e-mail log in configured
if use_email:
self.addEmailLog( config, config_main_section, logger )
#
# Add rotary file log if configured
error = self.addRotatingLog( config, config_main_section, command_line, logger, debug )
if error:
return logger, error # Logger will log the error to the email log as the file handler has failed
#
# Add logging to process msg windows in goobi-gui if configured
if use_goobi_gui_log:
logger = self.getGoobiLogger( config, command_line, logger, debug )
return logger, None
def error_message( self, message ):
self.error( message )
def error( self, message ):
if self.glogger:
self.glogger.error( message )
self.debuggingPrint( "Error: " + str(message) )
def warning_message( self, message ):
self.warning( message )
def warning( self, message ):
if self.glogger:
self.glogger.warning( message )
self.debuggingPrint( "Warning: " + str(message) )
def info_message( self, message ):
self.info( message )
def info( self, message ):
if self.glogger:
self.glogger.info( message )
self.debuggingPrint("Info: " + str(message) )
def debug_message( self, message ):
if self.debug:
self.debuggingPrint(message)
def debuggingPrint( self, message ):
if self.glogger:
self.glogger.debug(message)
if self.print_debug:
message = message.encode('ascii','replace').decode()
print("Debug: " + str(message))
