def _run_rt_command(self, command, uid=None): """ This function will run a RT CLI command as given user. If no user is specified the default RT user will be used, if configured. Should any of the configuration parameters be missing this function will return None. Otherwise it will return the standard output from the CLI command. @param command: RT CLI command to execute @type command: string @param uid: the Invenio user id to submit on behalf of. Optional. @type uid: int @return: standard output from the command given. None, if any errors. @rtype: string """ if not CFG_BIBCATALOG_SYSTEM_RT_URL: return None if uid: username, passwd = get_bibcat_from_prefs(uid) else: username = CFG_BIBCATALOG_SYSTEM_RT_DEFAULT_USER passwd = CFG_BIBCATALOG_SYSTEM_RT_DEFAULT_PWD httppart, siteandpath = CFG_BIBCATALOG_SYSTEM_RT_URL.split("//") bibcatalog_rt_server = httppart + "//" + username + ":" + passwd + "@" + siteandpath # set as env var os.environ["RTUSER"] = username os.environ["RTSERVER"] = bibcatalog_rt_server passwd = escape_shell_arg(passwd) error_code, myout, error_output = run_shell_command("echo " + passwd + " | " + command) if error_code > 0: raise ValueError('Problem running "%s": %d - %s' % (command, error_code, error_output)) return myout
def check_system(self, uid=None): """return an error string if there are problems""" if uid: rtuid, rtpw = get_bibcat_from_prefs(uid) else: # Assume default RT user rtuid = CFG_BIBCATALOG_SYSTEM_RT_DEFAULT_USER rtpw = CFG_BIBCATALOG_SYSTEM_RT_DEFAULT_PWD if not rtuid and not rtpw: return "No valid RT user login specified" if not CFG_BIBCATALOG_SYSTEM == "RT": return "CFG_BIBCATALOG_SYSTEM is not RT though this is an RT module" if not CFG_BIBCATALOG_SYSTEM_RT_CLI: return "CFG_BIBCATALOG_SYSTEM_RT_CLI not defined or empty" if not os.path.exists(CFG_BIBCATALOG_SYSTEM_RT_CLI): return "CFG_BIBCATALOG_SYSTEM_RT_CLI " + CFG_BIBCATALOG_SYSTEM_RT_CLI + " file does not exists" # Check that you can execute the binary.. this is a safe call unless someone can fake CFG_BIBCATALOG_SYSTEM_RT_CLI (unlikely) dummy, myout, myerr = run_shell_command(CFG_BIBCATALOG_SYSTEM_RT_CLI + " help") helpfound = False if myerr.count("help") > 0: helpfound = True if not helpfound: return ( "Execution of CFG_BIBCATALOG_SYSTEM_RT_CLI " + CFG_BIBCATALOG_SYSTEM_RT_CLI + " help did not produce output 'help'" ) if not CFG_BIBCATALOG_SYSTEM_RT_URL: return "CFG_BIBCATALOG_SYSTEM_RT_URL not defined or empty" # Construct URL, split RT_URL at // if not CFG_BIBCATALOG_SYSTEM_RT_URL.startswith("http://") and not CFG_BIBCATALOG_SYSTEM_RT_URL.startswith( "https://" ): return "CFG_BIBCATALOG__SYSTEM_RT_URL does not start with 'http://' or 'https://'" httppart, siteandpath = CFG_BIBCATALOG_SYSTEM_RT_URL.split("//") # Assemble by http://user:password@RT_URL bibcatalog_rt_server = httppart + "//" + rtuid + ":" + rtpw + "@" + siteandpath # set as env var os.environ["RTUSER"] = rtuid os.environ["RTSERVER"] = bibcatalog_rt_server # try to talk to RT server # this is a safe call since rtpw is the only variable in it, and it is escaped rtpw = escape_shell_arg(rtpw) dummy, myout, myerr = run_shell_command( "echo " + rtpw + " | " + CFG_BIBCATALOG_SYSTEM_RT_CLI + " ls \"Subject like 'F00'\"" ) if len(myerr) > 0: return "could not connect to " + bibcatalog_rt_server + " " + myerr # finally, check that there is some sane output like tickets or 'No matching result' saneoutput = (myout.count("matching") > 0) or (myout.count("1") > 0) if not saneoutput: return CFG_BIBCATALOG_SYSTEM_RT_CLI + " returned " + myout + " instead of 'matching' or '1'" return ""
def check_system(self, uid=None): """return an error string if there are problems""" if uid: rtuid, rtpw = get_bibcat_from_prefs(uid) else: # Assume default RT user rtuid = CFG_BIBCATALOG_SYSTEM_RT_DEFAULT_USER rtpw = CFG_BIBCATALOG_SYSTEM_RT_DEFAULT_PWD if not rtuid and not rtpw: return "No valid RT user login specified" if not CFG_BIBCATALOG_SYSTEM == 'RT': return "CFG_BIBCATALOG_SYSTEM is not RT though this is an RT module" if not CFG_BIBCATALOG_SYSTEM_RT_CLI: return "CFG_BIBCATALOG_SYSTEM_RT_CLI not defined or empty" if not os.path.exists(CFG_BIBCATALOG_SYSTEM_RT_CLI): return "CFG_BIBCATALOG_SYSTEM_RT_CLI " + CFG_BIBCATALOG_SYSTEM_RT_CLI + " file does not exists" # Check that you can execute the binary.. this is a safe call unless someone can fake CFG_BIBCATALOG_SYSTEM_RT_CLI (unlikely) dummy, myout, myerr = run_shell_command(CFG_BIBCATALOG_SYSTEM_RT_CLI + " help") helpfound = False if myerr.count("help") > 0: helpfound = True if not helpfound: return "Execution of CFG_BIBCATALOG_SYSTEM_RT_CLI " + CFG_BIBCATALOG_SYSTEM_RT_CLI + " help did not produce output 'help'" if not CFG_BIBCATALOG_SYSTEM_RT_URL: return "CFG_BIBCATALOG_SYSTEM_RT_URL not defined or empty" # Construct URL, split RT_URL at // if not CFG_BIBCATALOG_SYSTEM_RT_URL.startswith('http://') and \ not CFG_BIBCATALOG_SYSTEM_RT_URL.startswith('https://'): return "CFG_BIBCATALOG__SYSTEM_RT_URL does not start with 'http://' or 'https://'" httppart, siteandpath = CFG_BIBCATALOG_SYSTEM_RT_URL.split("//") # Assemble by http://user:password@RT_URL bibcatalog_rt_server = httppart + "//" + rtuid + ":" + rtpw + "@" + siteandpath #set as env var os.environ["RTUSER"] = rtuid os.environ["RTSERVER"] = bibcatalog_rt_server #try to talk to RT server #this is a safe call since rtpw is the only variable in it, and it is escaped rtpw = escape_shell_arg(rtpw) dummy, myout, myerr = run_shell_command("echo " + rtpw + " | " + CFG_BIBCATALOG_SYSTEM_RT_CLI + " ls \"Subject like 'F00'\"") if len(myerr) > 0: return "could not connect to " + bibcatalog_rt_server + " " + myerr #finally, check that there is some sane output like tickets or 'No matching result' saneoutput = (myout.count('matching') > 0) or (myout.count('1') > 0) if not saneoutput: return CFG_BIBCATALOG_SYSTEM_RT_CLI + " returned " + myout + " instead of 'matching' or '1'" return ""
def _run_rt_command(self, command, uid=None): """ This function will run a RT CLI command as given user. If no user is specified the default RT user will be used, if configured. Should any of the configuration parameters be missing this function will return None. Otherwise it will return the standard output from the CLI command. @param command: RT CLI command to execute @type command: string @param uid: the Invenio user id to submit on behalf of. Optional. @type uid: int @return: standard output from the command given. None, if any errors. @rtype: string """ if not CFG_BIBCATALOG_SYSTEM_RT_URL: return None if uid: username, passwd = get_bibcat_from_prefs(uid) else: username = CFG_BIBCATALOG_SYSTEM_RT_DEFAULT_USER passwd = CFG_BIBCATALOG_SYSTEM_RT_DEFAULT_PWD httppart, siteandpath = CFG_BIBCATALOG_SYSTEM_RT_URL.split("//") bibcatalog_rt_server = httppart + "//" + username + ":" + passwd + "@" + siteandpath #set as env var os.environ["RTUSER"] = username os.environ["RTSERVER"] = bibcatalog_rt_server passwd = escape_shell_arg(passwd) error_code, myout, error_output = run_shell_command("echo " + passwd + " | " + command) if error_code > 0: raise ValueError('Problem running "%s": %d - %s' % (command, error_code, error_output)) return myout