139 lines
5.3 KiB
Python
139 lines
5.3 KiB
Python
import os
|
|
import textwrap
|
|
|
|
from django.utils import timezone
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from orchestra.apps.orchestration import ServiceController
|
|
from orchestra.apps.resources import ServiceMonitor
|
|
|
|
from . import settings
|
|
|
|
|
|
class SystemUserBackend(ServiceController):
|
|
verbose_name = _("System user")
|
|
model = 'systemusers.SystemUser'
|
|
|
|
def save(self, user):
|
|
context = self.get_context(user)
|
|
groups = ','.join(self.get_groups(user))
|
|
context['groups_arg'] = '--groups %s' % groups if groups else ''
|
|
self.append(textwrap.dedent("""
|
|
if [[ $( id %(username)s ) ]]; then
|
|
usermod %(username)s --password '%(password)s' --shell %(shell)s %(groups_arg)s
|
|
else
|
|
useradd %(username)s --home %(home)s --password '%(password)s' --shell %(shell)s %(groups_arg)s
|
|
usermod -a -G %(username)s %(mainusername)s
|
|
fi
|
|
mkdir -p %(home)s
|
|
chown %(username)s.%(username)s %(home)s""" % context
|
|
))
|
|
|
|
def delete(self, user):
|
|
context = self.get_context(user)
|
|
self.append("{ sleep 2 && killall -u %(username)s -s KILL; } &" % context)
|
|
self.append("killall -u %(username)s || true" % context)
|
|
self.append("userdel %(username)s || true" % context)
|
|
self.append("groupdel %(username)s || true" % context)
|
|
self.delete_home(context, user)
|
|
|
|
def delete_home(self, context, user):
|
|
if user.is_main:
|
|
# TODO delete instead of this shit
|
|
context['deleted'] = context['home'].rstrip('/') + '.deleted'
|
|
self.append("mv %(home)s %(deleted)s" % context)
|
|
|
|
def get_groups(self, user):
|
|
if user.is_main:
|
|
return user.account.systemusers.exclude(username=user.username).values_list('username', flat=True)
|
|
groups = list(user.groups.values_list('username', flat=True))
|
|
groups += list(settings.SYSTEMUSERS_DEFAULT_GROUP_MEMBERS)
|
|
return groups
|
|
|
|
def get_context(self, user):
|
|
context = {
|
|
'username': user.username,
|
|
'password': user.password if user.active else '*%s' % user.password,
|
|
'shell': user.shell,
|
|
'mainusername': user.username if user.is_main else user.account.username,
|
|
'home': user.get_home()
|
|
}
|
|
return context
|
|
|
|
|
|
class SystemUserDisk(ServiceMonitor):
|
|
model = 'systemusers.SystemUser'
|
|
resource = ServiceMonitor.DISK
|
|
verbose_name = _('Main user disk')
|
|
|
|
def monitor(self, user):
|
|
context = self.get_context(user)
|
|
self.append("du -s %(home)s | cut -f1 | xargs echo %(object_id)s" % context)
|
|
|
|
def get_context(self, user):
|
|
context = SystemUserBackend().get_context(user)
|
|
context['object_id'] = user.pk
|
|
return context
|
|
|
|
|
|
class FTPTraffic(ServiceMonitor):
|
|
model = 'systemusers.SystemUser'
|
|
resource = ServiceMonitor.TRAFFIC
|
|
verbose_name = _('Main FTP traffic')
|
|
|
|
def prepare(self):
|
|
current_date = timezone.localtime(self.current_date)
|
|
current_date = current_date.strftime("%Y%m%d%H%M%S")
|
|
self.append(textwrap.dedent("""
|
|
function monitor () {
|
|
OBJECT_ID=$1
|
|
INI_DATE=$2
|
|
USERNAME="$3"
|
|
LOG_FILE="$4"
|
|
grep "UPLOAD\|DOWNLOAD" "${LOG_FILE}" \\
|
|
| grep " \\[${USERNAME}\\] " \\
|
|
| awk -v ini="${INI_DATE}" '
|
|
BEGIN {
|
|
end = "%s"
|
|
sum = 0
|
|
months["Jan"] = "01"
|
|
months["Feb"] = "02"
|
|
months["Mar"] = "03"
|
|
months["Apr"] = "04"
|
|
months["May"] = "05"
|
|
months["Jun"] = "06"
|
|
months["Jul"] = "07"
|
|
months["Aug"] = "08"
|
|
months["Sep"] = "09"
|
|
months["Oct"] = "10"
|
|
months["Nov"] = "11"
|
|
months["Dec"] = "12"
|
|
} {
|
|
# log: Fri Jul 11 13:23:17 2014
|
|
split($4, t, ":")
|
|
# line_date = year month day hour minute second
|
|
line_date = $5 months[$2] $3 t[1] t[2] t[3]
|
|
if ( line_date > ini && line_date < end)
|
|
split($0, l, "\\", ")
|
|
split(l[3], b, " ")
|
|
sum += b[1]
|
|
} END {
|
|
print sum
|
|
}
|
|
' | xargs echo ${OBJECT_ID}
|
|
}""" % current_date))
|
|
|
|
def monitor(self, user):
|
|
context = self.get_context(user)
|
|
self.append(
|
|
'monitor %(object_id)i %(last_date)s "%(username)s" "%(log_file)s"' % context)
|
|
|
|
def get_context(self, user):
|
|
last_date = timezone.localtime(self.get_last_date(user.pk))
|
|
return {
|
|
'log_file': settings.SYSTEMUSERS_FTP_LOG_PATH,
|
|
'last_date': last_date.strftime("%Y%m%d%H%M%S"),
|
|
'object_id': user.pk,
|
|
'username': user.username,
|
|
}
|