1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
|
#
# Copyright 2013, Martin Owens <doctormo@gmail.com>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 3.0 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library.
#
"""
Access logs in known locations to find information about them.
"""
import os
import re
import codecs
from datetime import datetime
MATCHER = r'(?P<date>\w+ +\d+ +\d\d:\d\d:\d\d) (?P<host>\w+) ' + \
r'CRON\[(?P<pid>\d+)\]: \((?P<user>\w+)\) CMD \((?P<cmd>.*)\)'
class LogReader(object):
"""Opens a Log file, reading backwards and watching for changes"""
def __init__(self, filename, mass=4096):
self.filename = filename
self.mass = mass
self.size = -1
self.read = -1
self.pipe = None
def __enter__(self):
self.size = os.stat(self.filename)[6]
self.pipe = codecs.open(self.filename, 'r', encoding='utf-8')
return self
def __exit__(self, error_type, value, traceback):
self.pipe.close()
def __iter__(self):
if self.pipe is None:
with self as reader:
for (offset, line) in reader.readlines():
yield line
else:
for (offset, line) in self.readlines():
yield line
def readlines(self, until=0):
"""Iterator for reading lines from a file backwards"""
if not self.pipe or self.pipe.closed:
raise IOError("Can't readline, no opened file.")
# Always seek to the end of the file, this accounts for file updates
# that happen during our running process.
location = self.size
halfline = ''
while location > until:
location -= self.mass
mass = self.mass
if location < 0:
mass = self.mass + location
location = 0
self.pipe.seek(location)
line = self.pipe.read(mass) + halfline
data = line.split('\n')
if location != 0:
halfline = data.pop(0)
loc = location + mass
data.reverse()
for line in data:
if line.strip() == '':
continue
yield (loc, line)
loc -= len(line)
def cron_date_to_datetime(cron_str):
with_year = f"{cron_str} {datetime.now().year}"
return datetime.strptime(with_year, "%b %d %H:%M:%S %Y")
class CronLog(LogReader):
"""Use the LogReader to make a Cron specific log reader"""
def __init__(self, filename='/var/log/syslog', user=None):
LogReader.__init__(self, filename)
self.user = user
def for_program(self, command):
"""Return log entries for this specific command name"""
return ProgramLog(self, command)
def __iter__(self):
for line in super(CronLog, self).__iter__():
match = re.match(MATCHER, str(line))
datum = match and match.groupdict()
if datum and (not self.user or datum['user'] == self.user):
datum['date'] = cron_date_to_datetime(datum['date'])
yield datum
class ProgramLog(object):
"""Specific log control for a single command/program"""
def __init__(self, log, command):
self.log = log
self.command = command
def __iter__(self):
for entry in self.log:
if entry['cmd'] == str(self.command):
yield entry
|