class Note(db.Model):
__tablename__ = 'note'
id = db.Column(db.Integer, primary_key=True)
text = db.Column(db.Text(2048), nullable=False)
at = db.Column(db.DateTime, nullable=False)
by_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
by = db.relationship('User')
ticket_id = db.Column(db.Integer, db.ForeignKey('ticket.id'))
ticket = db.relationship('Ticket')
class TicketEvent(Event):
__tablename__ = 'event_ticket'
id = db.Column(db.Integer, db.ForeignKey('event.id'), primary_key=True)
ticket_id = db.Column(db.Integer, db.ForeignKey('ticket.id'))
ticket = db.relationship('Ticket', back_populates='events')
__mapper_args__ = {
'polymorphic_identity': 'event_ticket',
}
def __repr__(self):
return 'TicketEvent'
class TicketCloseEvent(TicketEvent):
__tablename__ = 'event_ticket_close'
id = db.Column(db.Integer,
db.ForeignKey('event_ticket.id'),
primary_key=True)
reason_id = db.Column(db.Integer, db.ForeignKey('ticket_close_reason.id'))
reason = db.relationship('TicketCloseReason')
__mapper_args__ = {
'polymorphic_identity': 'event_ticket_close',
}
def __repr__(self):
return 'TicketCloseEvent'
class TicketNoteAddedEvent(TicketEvent):
__tablename__ = 'event_ticket_note_added'
id = db.Column(db.Integer,
db.ForeignKey('event_ticket.id'),
primary_key=True)
note_id = db.Column(db.Integer, db.ForeignKey('note.id'), nullable=False)
note = db.relationship('Note')
__mapper_args__ = {
'polymorphic_identity': 'event_ticket_note_added',
}
def __repr__(self):
return 'TicketNoteAddedEvent'
class TicketAreaTransferEvent(TicketEvent):
__tablename__ = 'event_ticket_area_transfer'
id = db.Column(db.Integer,
db.ForeignKey('event_ticket.id'),
primary_key=True)
from_area_id = db.Column(db.Integer, db.ForeignKey('area.id'))
from_area = db.relationship('Area', foreign_keys=[from_area_id])
to_area_id = db.Column(db.Integer,
db.ForeignKey('area.id'),
nullable=False)
to_area = db.relationship('Area', foreign_keys=[to_area_id])
__mapper_args__ = {
'polymorphic_identity': 'event_ticket_area_transfer',
}
def __repr__(self):
return 'TicketAreaTransferEvent'
class TicketReopenEvent(TicketEvent):
__tablename__ = 'event_ticket_reopen'
id = db.Column(db.Integer,
db.ForeignKey('event_ticket.id'),
primary_key=True)
__mapper_args__ = {
'polymorphic_identity': 'event_ticket_reopen',
}
def __repr__(self):
return 'TicketReopenEvent'
class Event(db.Model):
__tablename__ = 'event'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
user = db.relationship('User')
at = db.Column(db.DateTime, nullable=False, default=datetime.now())
type = db.Column(db.String(50))
__mapper_args__ = {'polymorphic_identity': 'event', 'polymorphic_on': type}
def __repr__(self):
return 'Event'
class Ticket(db.Model):
__tablename__ = 'ticket'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.Unicode(255), nullable=False)
text = db.Column(db.UnicodeText(2048))
current_area_id = db.Column(db.Integer,
db.ForeignKey('area.id'),
nullable=False)
current_area = db.relationship('Area')
created_at = db.Column(db.DateTime, nullable=False)
created_by_id = db.Column(db.Integer,
db.ForeignKey('user.id'),
nullable=False)
created_by = db.relationship('User', foreign_keys=[created_by_id])
owned_by_id = db.Column(db.Integer, db.ForeignKey('user.id'))
owned_by = db.relationship('User', foreign_keys=[owned_by_id])
is_closed = db.Column(db.Boolean, default=False)
zabbix_trigger_id = db.Column(db.Integer, default=0)
last_change = db.Column(db.DateTime, nullable=False)
events = db.relationship('TicketEvent', back_populates='ticket')
def touch(self):
self.last_change = datetime.now()
def add_note(self, by, text):
self.touch()
note = Note(text=text, at=datetime.now(), by=by, ticket=self)
db.session.add(note)
note_event = TicketNoteAddedEvent(
note=note,
ticket=self,
user=by,
at=datetime.now(),
)
db.session.add(note_event)
db.session.commit()
logger.info(u'%s added note to ticket #%d' % (by.user, self.id))
return note
def reopen(self, by):
self.touch()
note_event = TicketReopenEvent(
ticket=self,
user=by,
at=datetime.now(),
)
db.session.add(note_event)
db.session.commit()
logger.info(u'%s reopened ticket #%d' % (by.user, self.id))
return note
def take(self, by):
self.touch()
self.owned_by = by
event = TicketTakeEvent(user=by, ticket=self, at=datetime.now())
db.session.add(event)
db.session.commit()
logger.info(u'%s took ticket #%d' % (by.user, self.id))
def transfer_to(self, by, to_area):
self.touch()
event = TicketAreaTransferEvent(user=by,
ticket=self,
to_area=to_area,
from_area=self.current_area,
at=datetime.now())
db.session.add(event)
self.current_area = to_area
self.owned_by = None
db.session.commit()
logger.info(u'%s transfered ticket #%d to %s' %
(by.user, self.id, to_area.name))
def close(self, by, reason):
self.touch()
self.is_closed = True
event = TicketCloseEvent(user=by,
ticket=self,
at=datetime.now(),
reason=reason)
db.session.add(event)
db.session.commit()
logger.info(u'%s closed ticket #%d (%s)' %
(by.user, self.id, reason.reason))
from flask_principal import Principal, Identity, AnonymousIdentity, \
identity_changed, Permission, RoleNeed, identity_loaded
from flask_admin.contrib.sqla import ModelView
from datetime import datetime, timedelta
import math
from zts import db, app, principal
import re
import logging
import config
from pyzabbix import ZabbixAPI
logger = logging.getLogger(config.ZTS_LOGGER_NAME)
user_area_assoc_table = db.Table(
'user_area', db.Model.metadata,
db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
db.Column('area_id', db.Integer, db.ForeignKey('area.id')))
ticket_close_permission = Permission(RoleNeed('ticket_close'))
admin_permission = Permission(RoleNeed('admin'))
report_permission = Permission(RoleNeed('report'))
class User(db.Model):
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
user = db.Column(db.String(32), nullable=False)
name = db.Column(db.Unicode(32))
pwhash = db.Column(db.String(128))
access_admin = db.Column(db.Boolean(), default=False, nullable=False)