mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-11-16 01:22:15 +03:00
91 lines
4.3 KiB
Python
91 lines
4.3 KiB
Python
#!/usr/bin/python3
|
||
# -*- coding: utf-8 -*-
|
||
# Author:Mine DOGAN <mine.dogan@agem.com.tr>
|
||
|
||
import json
|
||
import os
|
||
import sys
|
||
|
||
from base.plugin.abstract_plugin import AbstractPlugin
|
||
|
||
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__))))
|
||
|
||
|
||
|
||
class DiskQuota(AbstractPlugin):
|
||
def __init__(self, data, context):
|
||
super(AbstractPlugin, self).__init__()
|
||
self.data = data
|
||
self.context = context
|
||
self.logger = self.get_logger()
|
||
self.message_code = self.get_message_code()
|
||
|
||
self.username = self.context.get('username')
|
||
|
||
self.mount = 'mount -o remount {}'
|
||
self.quotaon_all = 'quotaon --all'
|
||
self.quotaon_avug = 'quotaon -avug'
|
||
#self.set_quota = 'setquota --always-resolve -u {0} {1} {2} 0 0 --all'
|
||
self.set_quota = 'quotatool -u {0} -b -q {1} -l {2} /home'
|
||
#self.get_quota = 'quota -u {0} | awk \'{{print $4}}\' | tail -1'
|
||
self.get_quota = 'repquota /home | grep {0} | awk \'{{print $5}}\''
|
||
|
||
self.parameters = json.loads(self.data)
|
||
|
||
self.soft_quota = str(int(self.parameters['soft-quota']) * 1024)
|
||
self.hard_quota = str(int(self.parameters['hard-quota']) * 1024)
|
||
self.default_quota = str(int(self.parameters['default-quota']) * 1024)
|
||
|
||
self.old_quota = None
|
||
|
||
self.logger.debug('Parameters were initialized.')
|
||
|
||
def handle_policy(self):
|
||
self.logger.debug('Policy handling...')
|
||
try:
|
||
self.username = self.context.get('username')
|
||
|
||
|
||
self.old_quota = self.execute(self.get_quota.format(self.username))[1]
|
||
|
||
self.execute(self.set_quota.format(self.username, self.soft_quota, self.hard_quota))
|
||
self.logger.debug(
|
||
'Set soft and hard quota. Username: {0}, Soft Quota: {1}, Hard Quota: {2}'.format(self.username,
|
||
self.soft_quota,
|
||
self.hard_quota))
|
||
|
||
self.create_default_quota_file()
|
||
|
||
result = dict()
|
||
if self.context.is_mail_send():
|
||
mail_content = self.context.get_mail_content()
|
||
if mail_content.__contains__('{ahenk-ip}'):
|
||
mail_content = str(mail_content).replace('{ahenk-ip}', ' {0} IP\'li Ahenk\'teki yeni'.format(
|
||
str(self.Hardware.ip_addresses())))
|
||
if mail_content.__contains__('{old-quota}'):
|
||
mail_content = str(mail_content).replace('{old-quota}',
|
||
' Mevcut kota değeri {0} MB olan'.format(str(int(self.old_quota)/1024)))
|
||
if mail_content.__contains__('{soft-quota}'):
|
||
mail_content = str(mail_content).replace('{soft-quota}',str(int(self.soft_quota)/1024)+' MB')
|
||
if mail_content.__contains__('{hard-quota}'):
|
||
mail_content = str(mail_content).replace('{hard-quota}', str(int(self.hard_quota)/1024)+' MB')
|
||
if mail_content.__contains__('{default-quota}'):
|
||
mail_content = str(mail_content).replace('{default-quota}', str(int(self.default_quota)/1024)+' MB')
|
||
|
||
self.context.set_mail_content(mail_content)
|
||
result['mail_content'] = str(self.context.get_mail_content())
|
||
result['mail_subject'] = str(self.context.get_mail_subject())
|
||
result['mail_send'] = self.context.is_mail_send()
|
||
|
||
self.context.create_response(code=self.get_message_code().POLICY_PROCESSED.value,
|
||
data=json.dumps(result),
|
||
message='Kotalar başarıyla güncellendi.',
|
||
content_type=self.get_content_type().APPLICATION_JSON.value)
|
||
|
||
except Exception as e:
|
||
self.logger.error('[DiskQuota] A problem occurred while handling browser profile: {0}'.format(str(e)))
|
||
self.context.create_response(code=self.get_message_code().POLICY_ERROR.value,
|
||
message='Disk Quota profili uygulanırken bir hata oluştu.')
|
||
|
||
def create_default_quota_file(self):
|
||
self.write_file('default_quota', self.default_quota)
|