示例#1
0
文件: nagent.py 项目: mbseid/nagent
def get_executable_temp_file(command, event_id):
    """
    Saves the given command into a file file so that it can be executed directly.
    """
    try:
        # Replace all new lines with carriage returns.
        path = os_sep_join(AGENT_OUTPUT_DIR, event_id + "_cmd")
        temp_file = open(path, "w+")
        temp_file.write(command)
        temp_file.close()

        # Make the file executable.
        st = os.stat(path)
        os.chmod(path, st.st_mode | S_IEXEC)
        return path
    except Exception:
        _, e, _ = sys.exc_info()
        nlogger.error("Cannot write to a temporary file. Error: %s", repr(e))
        return None
示例#2
0
文件: nagent.py 项目: mbseid/nagent
def get_temp_batch_file_for_windows(command):
    """
    Saves the given command into a batch file on Windows after taking care of new line conversions for
    windows. This method should only be invoked on Windows.
    """
    try:
        # Replace all new lines with carriage returns.
        text = command.replace("\n", "\r\n")
        path = os_sep_join(AGENT_OUTPUT_DIR, "command.bat")
        temp_file = open(path, "w+")
        temp_file.write(text)
        temp_file.close()

        # Make the file executable.
        st = os.stat(path)
        os.chmod(path, st.st_mode | S_IEXEC)
        return path
    except Exception:
        _, e, _ = sys.exc_info()
        nlogger.error("Cannot write to a temporary file. Error: %s", repr(e))
        return None
示例#3
0
文件: nagent.py 项目: mbseid/nagent
from boto.sqs.message import RawMessage
from logging.handlers import RotatingFileHandler
from os.path import join as os_sep_join
from subprocess import Popen
from stat import S_IEXEC

# Force use of pyopenssl
try:
    import urllib3.contrib.pyopenssl
    urllib3.contrib.pyopenssl.inject_into_urllib3()
except ImportError:
    pass

AGENT_VERSION = "1.0.0"
USER_HOME = os.path.expanduser("~")
AGENT_HOME = os_sep_join(USER_HOME, "neptuneio")
DEFAULT_CONFIG_FILE = os_sep_join(AGENT_HOME, "nagent.cfg")
LOG_FILE_PATH = os_sep_join(AGENT_HOME, "nagent.log")
AGENT_STATUS_FILE_PATH = os_sep_join(AGENT_HOME, ".status")
AGENT_OUTPUT_DIR = os_sep_join(AGENT_HOME, "output")
AGENT_EVENT_LIST_FILE = os_sep_join(AGENT_OUTPUT_DIR, ".nagent_events")
PLATFORM_INFO = platform.platform()
WINDOWS_OS_NAME = 'nt'

# Agent operation settings
LOGROTATE_MAX_SIZE = 10000000  # 10MB
LOGROTATE_NUM_FILES = 5
AGENT_HEALTH_CHECK_PERIOD = 3600  # 1 hour
AGENT_REGISTRATION_RETRY_INTERVAL = 5 * 60  # 5 min
NUM_RETRIES_FOR_SQS_CONNECT = 5
NUM_MSGS_TO_FETCH_FROM_SQS = 10