mirror of https://github.com/veops/cmdb.git
clear
This commit is contained in:
pycook
pycook
parent
6489a2eb48
commit
7f647887c8
|
|
@ -19,3 +19,4 @@ config.cfg
|
|||
core/special.py
|
||||
lib/special
|
||||
lib/audit*
|
||||
templates/*audit*
|
||||
|
|
|
|||
168
lib/audit.py
168
lib/audit.py
|
|
@ -1,168 +0,0 @@
|
|||
# -*- coding:utf-8 -*-
|
||||
|
||||
__author__ = 'pycook'
|
||||
|
||||
import datetime
|
||||
|
||||
from flask import current_app
|
||||
|
||||
from models.cmdb import CIAudit
|
||||
from models.cmdb import CIAttributeAudit
|
||||
from models.cmdb import CIAttributeCache
|
||||
from models.cmdb import CITypeCache
|
||||
from models.cmdb import CI
|
||||
from models import row2dict
|
||||
from extensions import db
|
||||
from lib.const import TableMap
|
||||
from tasks.cmdb import ci_cache
|
||||
|
||||
|
||||
class CIAuditManager(object):
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def get_cis_for_audits(self, page, type_ids, per_page=25):
|
||||
audit_cis = db.session.query(CIAudit)
|
||||
if type_ids:
|
||||
audit_cis = audit_cis.join(CI, CI.ci_id == CIAudit.ci_id).filter(
|
||||
CI.type_id.in_(type_ids))
|
||||
|
||||
audit_cis = audit_cis.filter(CIAudit.is_audit == 0).order_by(
|
||||
CIAudit.created_at)
|
||||
numfound = audit_cis.count()
|
||||
audit_cis = audit_cis.offset((page - 1) * per_page).limit(per_page)
|
||||
total = audit_cis.count()
|
||||
audit_cis = audit_cis.all()
|
||||
result = list()
|
||||
for audit_ci in audit_cis:
|
||||
audit_dict = row2dict(audit_ci)
|
||||
audit_attrs = db.session.query(CIAttributeAudit).filter(
|
||||
CIAttributeAudit.audit_id == audit_ci.audit_id).all()
|
||||
audit_dict["values"] = list()
|
||||
for audit_attr in audit_attrs:
|
||||
audit_attr_dict = row2dict(audit_attr)
|
||||
audit_attr_dict["attr_name"] = CIAttributeCache.get(
|
||||
audit_attr.attr_id).attr_name
|
||||
audit_dict['values'].append(audit_attr_dict)
|
||||
result.append(audit_dict)
|
||||
return numfound, total, result
|
||||
|
||||
def create_ci_audits(self, type_name, attr_pairs):
|
||||
ci_type = CITypeCache.get(type_name)
|
||||
uniq_key = CIAttributeCache.get(ci_type.uniq_id)
|
||||
table = TableMap(attr_name=uniq_key.attr_name).table
|
||||
value = db.session.query(table.ci_id).filter(
|
||||
table.attr_id == uniq_key.attr_id).filter(
|
||||
table.value == attr_pairs.get(uniq_key.attr_name)).first()
|
||||
del attr_pairs[uniq_key.attr_name]
|
||||
if value and attr_pairs:
|
||||
ci_audit = db.session.query(CIAudit).filter(
|
||||
CIAudit.ci_id == value.ci_id).filter(
|
||||
CIAudit.is_audit == 0).first()
|
||||
if ci_audit is None:
|
||||
ci_audit = CIAudit()
|
||||
ci_audit.is_notified = False
|
||||
ci_audit.created_at = datetime.datetime.now()
|
||||
ci_audit.ci_id = value.ci_id
|
||||
ci_audit.updated_at = datetime.datetime.now()
|
||||
ci_audit.origin = 1 # TODO
|
||||
db.session.add(ci_audit)
|
||||
db.session.commit()
|
||||
for attr, attr_value in attr_pairs.items():
|
||||
attr_id = CIAttributeCache.get(attr).attr_id
|
||||
ci_attr_audit = CIAttributeAudit()
|
||||
ci_attr_audit.attr_id = attr_id
|
||||
all_values = attr_value.strip().split("###")
|
||||
ci_attr_audit.cur_value = all_values[0]
|
||||
ci_attr_audit.new_value = all_values[1]
|
||||
ci_attr_audit.audit_id = ci_audit.audit_id
|
||||
db.session.add(ci_attr_audit)
|
||||
db.session.flush()
|
||||
try:
|
||||
db.session.commit()
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
current_app.logger.error("create ci audits error, %s" % str(e))
|
||||
return False, "create ci audits error, %s" % str(e)
|
||||
return True, None
|
||||
|
||||
def _update_cmdb(self, ci_id, attr_id, value):
|
||||
try:
|
||||
attr_name = CIAttributeCache.get(attr_id).attr_name
|
||||
table = TableMap(attr_name=attr_name).table
|
||||
attr_value = db.session.query(table).filter(
|
||||
table.attr_id == attr_id).filter(
|
||||
table.ci_id == ci_id).first()
|
||||
attr_value.value = value
|
||||
db.session.add(attr_value)
|
||||
|
||||
except Exception as e:
|
||||
return False, "audit failed, %s" % str(e)
|
||||
return True, ci_id
|
||||
|
||||
def audit_by_attr(self, audit_id, attr_ids, value=None):
|
||||
ci_audit = CIAudit.query.get(audit_id)
|
||||
ci_id = ci_audit.ci_id
|
||||
for attr_id in attr_ids:
|
||||
attr_audit = db.session.query(CIAttributeAudit).filter(
|
||||
CIAttributeAudit.audit_id == audit_id).filter(
|
||||
CIAttributeAudit.attr_id == attr_id).first()
|
||||
if attr_audit:
|
||||
attr_audit.is_audit = True
|
||||
attr_audit.auditor = 1 # TODO
|
||||
attr_audit.audited_at = datetime.datetime.now()
|
||||
if value is not None:
|
||||
attr_audit.audit_value = value
|
||||
else:
|
||||
attr_audit.audit_value = attr_audit.new_value
|
||||
if attr_audit.cur_value != value: # update cmdb
|
||||
ret, res = self._update_cmdb(ci_id, attr_id, value)
|
||||
if not ret:
|
||||
return False, res
|
||||
attr_audit.is_updated = True
|
||||
db.session.add(attr_audit)
|
||||
db.session.flush()
|
||||
if db.session.query(CIAttributeAudit).filter_by(
|
||||
audit_id=audit_id).filter_by(is_audit=0).first() is None:
|
||||
ci_audit.is_audit = True
|
||||
ci_audit.updated_at = datetime.datetime.now()
|
||||
db.session.add(ci_audit)
|
||||
ci_cache.apply_async([ci_id], queue="cmdb_async")
|
||||
try:
|
||||
db.session.commit()
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
current_app.logger.error(
|
||||
"audit by attribute is error, {0}".format(str(e)))
|
||||
return False, "audit by attribute is error, %s" % str(e)
|
||||
return True, None
|
||||
|
||||
def audit_by_cis(self, ci_ids):
|
||||
for ci_id in ci_ids:
|
||||
ci_audit = db.session.query(CIAudit).filter_by(ci_id=ci_id).first()
|
||||
attr_audits = db.session.query(CIAttributeAudit).filter_by(
|
||||
audit_id=ci_audit.audit_id).all()
|
||||
for attr_audit in attr_audits:
|
||||
attr_audit.is_audit = True
|
||||
attr_audit.auditor = 1 # TODO
|
||||
attr_audit.audited_at = datetime.datetime.now()
|
||||
attr_audit.audit_value = attr_audit.new_value
|
||||
ret, res = self._update_cmdb(
|
||||
ci_id, attr_audit.attr_id,
|
||||
attr_audit.new_value)
|
||||
if not ret:
|
||||
return False, res
|
||||
attr_audit.is_updated = True
|
||||
db.session.add(attr_audit)
|
||||
ci_audit.is_audit = True
|
||||
ci_audit.updated_at = datetime.datetime.now()
|
||||
db.session.add(ci_audit)
|
||||
ci_cache.apply_async([ci_id], queue="cmdb_async")
|
||||
try:
|
||||
db.session.commit()
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
current_app.logger.error(
|
||||
"audit by attribute error, {0}".format(str(e)))
|
||||
return False, "audit by cis is error, %s" % str(e)
|
||||
return True, None
|
||||
86
lib/mail.py
86
lib/mail.py
|
|
@ -1,86 +0,0 @@
|
|||
# -*- coding:utf-8 -*-
|
||||
|
||||
|
||||
import requests
|
||||
|
||||
from flask import current_app
|
||||
from flask.ext.mail import Message
|
||||
|
||||
from extensions import mail
|
||||
from models.account import User
|
||||
|
||||
|
||||
def sendmail(users, subject, message, html=False, app=None):
|
||||
if app:
|
||||
mail.app = app
|
||||
else:
|
||||
app = current_app
|
||||
recipients = [x.email for x in users if isinstance(x, User)]
|
||||
recipients.extend(
|
||||
[x for x in users if isinstance(x, basestring) and '@' in x])
|
||||
sender = app.config.get('DEFAULT_MAIL_SENDER')
|
||||
if html:
|
||||
msg = Message(recipients=recipients,
|
||||
html=message,
|
||||
subject=subject,
|
||||
sender=sender)
|
||||
else:
|
||||
msg = Message(recipients=recipients,
|
||||
body=message,
|
||||
subject=subject,
|
||||
sender=sender)
|
||||
mail.send(msg)
|
||||
|
||||
|
||||
from email.mime.multipart import MIMEMultipart
|
||||
from email.mime.text import MIMEText
|
||||
from email.header import Header
|
||||
from email.mime.image import MIMEImage
|
||||
import smtplib
|
||||
import time
|
||||
from email import Utils
|
||||
|
||||
|
||||
def send_mail(sender, receiver, subject, content, ctype="html", pics=(),
|
||||
smtpserver='mail.51ping.com',
|
||||
username="networkbench@51ping.com", password="12qwaszx"):
|
||||
"""subject and body are unicode objects"""
|
||||
if ctype == "html":
|
||||
msg = MIMEText(content, 'html', 'utf-8')
|
||||
else:
|
||||
msg = MIMEText(content, 'plain', 'utf-8')
|
||||
|
||||
if len(pics) != 0:
|
||||
msgRoot = MIMEMultipart('related')
|
||||
msgText = MIMEText(content, 'html', 'utf-8')
|
||||
msgRoot.attach(msgText)
|
||||
i = 1
|
||||
for pic in pics:
|
||||
fp = open(pic, "rb")
|
||||
image = MIMEImage(fp.read())
|
||||
fp.close()
|
||||
image.add_header('Content-ID', '<img%02d>' % i)
|
||||
msgRoot.attach(image)
|
||||
i += 1
|
||||
msg = msgRoot
|
||||
|
||||
msg['Subject'] = Header(subject, 'utf-8')
|
||||
msg['From'] = sender
|
||||
msg['To'] = ';'.join(receiver)
|
||||
msg['Message-ID'] = Utils.make_msgid()
|
||||
msg['date'] = time.strftime('%a, %d %b %Y %H:%M:%S %z')
|
||||
|
||||
smtp = smtplib.SMTP()
|
||||
smtp.connect(smtpserver, 25)
|
||||
smtp.login(username, password)
|
||||
smtp.sendmail(sender, receiver, msg.as_string())
|
||||
smtp.quit()
|
||||
|
||||
|
||||
def send_sms(mobile, content):
|
||||
sms_uri = current_app.config.get("SMS_URI") % (mobile, content)
|
||||
try:
|
||||
current_app.logger.info(sms_uri)
|
||||
requests.get(sms_uri)
|
||||
except Exception as e:
|
||||
current_app.logger.error("send sms error, %s" % str(e))
|
||||
|
|
@ -1,51 +0,0 @@
|
|||
<body xmlns="http://www.w3.org/1999/html">
|
||||
<br>
|
||||
|
||||
<table dir="ltr" border="1" cellpadding="0"
|
||||
style="font-family:arial,sans-serif;font-size:13px; width:0;border-collapse:collapse;">
|
||||
<thead>
|
||||
<tr>
|
||||
<th>变更设备</th>
|
||||
<!--<th>来源</th>-->
|
||||
<th>变更字段</th>
|
||||
<th>原值</th>
|
||||
<th>变更值</th>
|
||||
<th>创建时间</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{%- for res in result %}
|
||||
{%- for v in res.get("values") %}
|
||||
<tr style="height: 20px">
|
||||
{%- if res.get("values").index(v) == 0 %}
|
||||
<td dir="ltr" rowspan="{{res.get('values') | length}}" style="min-width: 100px">
|
||||
{{res.get("ci_id")}}
|
||||
</td>
|
||||
{%- endif %}
|
||||
<td nowrap="" align="center" style="min-width:80px; padding-left:10px; text-align:left">
|
||||
{{ v.get('attr_name') }}
|
||||
</td>
|
||||
<td nowrap="" align="center" style="padding-left:10px; text-align:left">
|
||||
{{ v.get('cur_value') }}
|
||||
</td>
|
||||
<td nowrap="" align="center" style="padding-left:10px; text-align:left">
|
||||
{{ v.get('new_value') }}
|
||||
</td>
|
||||
<td nowrap="" align="center" style="padding-left:10px; text-align:left">
|
||||
{{ v.get('created_at') }}
|
||||
</td>
|
||||
|
||||
</tr>
|
||||
{%- endfor %}
|
||||
{%- endfor %}
|
||||
</tbody>
|
||||
</table>
|
||||
|
||||
<div>
|
||||
<p class="MsoNormal"><span style="font-family:宋体; background-color: yellowgreen"></span>
|
||||
<span lang="EN-US"><u></u><u></u></span>
|
||||
<a href="http://web.cmdb.dp/audit/">查看详情</a>
|
||||
</p>
|
||||
<br>
|
||||
</div>
|
||||
</body>
|
||||
Loading…
Reference in New Issue