user.py
133 lines
| 5.7 KiB
| text/x-python
|
PythonLexer
r0 | # -*- coding: utf-8 -*- | |||
r112 | # Copyright 2010 - 2017 RhodeCode GmbH and the AppEnlight project authors | |||
r0 | # | |||
r112 | # Licensed under the Apache License, Version 2.0 (the "License"); | |||
# you may not use this file except in compliance with the License. | ||||
# You may obtain a copy of the License at | ||||
r0 | # | |||
r112 | # http://www.apache.org/licenses/LICENSE-2.0 | |||
r0 | # | |||
r112 | # Unless required by applicable law or agreed to in writing, software | |||
# distributed under the License is distributed on an "AS IS" BASIS, | ||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||||
# See the License for the specific language governing permissions and | ||||
# limitations under the License. | ||||
r0 | ||||
import logging | ||||
import sqlalchemy as sa | ||||
from datetime import datetime | ||||
from appenlight.models import Base, get_db_session | ||||
from appenlight.models.services.event import EventService | ||||
from appenlight.models.integrations import IntegrationException | ||||
from pyramid.threadlocal import get_current_request | ||||
from ziggurat_foundations.models.user import UserMixin | ||||
r135 | from ziggurat_foundations.models.services.user import UserService | |||
r0 | ||||
log = logging.getLogger(__name__) | ||||
class User(UserMixin, Base): | ||||
__possible_permissions__ = [] | ||||
first_name = sa.Column(sa.Unicode(25)) | ||||
last_name = sa.Column(sa.Unicode(25)) | ||||
company_name = sa.Column(sa.Unicode(255), default='') | ||||
company_address = sa.Column(sa.Unicode(255), default='') | ||||
zip_code = sa.Column(sa.Unicode(25), default='') | ||||
city = sa.Column(sa.Unicode(50), default='') | ||||
default_report_sort = sa.Column(sa.Unicode(25), default='newest') | ||||
notes = sa.Column(sa.UnicodeText, default='') | ||||
notifications = sa.Column(sa.Boolean(), default=True) | ||||
registration_ip = sa.Column(sa.UnicodeText(), default='') | ||||
alert_channels = sa.orm.relationship('AlertChannel', | ||||
cascade="all,delete-orphan", | ||||
passive_deletes=True, | ||||
passive_updates=True, | ||||
backref='owner', | ||||
order_by='AlertChannel.channel_name, ' | ||||
'AlertChannel.channel_value') | ||||
alert_actions = sa.orm.relationship('AlertChannelAction', | ||||
cascade="all,delete-orphan", | ||||
passive_deletes=True, | ||||
passive_updates=True, | ||||
backref='owner', | ||||
order_by='AlertChannelAction.pkey') | ||||
auth_tokens = sa.orm.relationship('AuthToken', | ||||
cascade="all,delete-orphan", | ||||
passive_deletes=True, | ||||
passive_updates=True, | ||||
backref='owner', | ||||
order_by='AuthToken.creation_date') | ||||
def get_dict(self, exclude_keys=None, include_keys=None, | ||||
extended_info=False): | ||||
result = super(User, self).get_dict(exclude_keys, include_keys) | ||||
if extended_info: | ||||
result['groups'] = [g.group_name for g in self.groups] | ||||
r135 | result['permissions'] = [p.perm_name for p in UserService.permissions(self)] | |||
r0 | request = get_current_request() | |||
r135 | apps = UserService.resources_with_perms(self, | |||
r0 | ['view'], resource_types=['application']) | |||
result['applications'] = sorted( | ||||
[{'resource_id': a.resource_id, | ||||
'resource_name': a.resource_name} | ||||
for a in apps.all()], | ||||
key=lambda x: x['resource_name'].lower()) | ||||
result['assigned_reports'] = [r.get_dict(request) for r | ||||
in self.assigned_report_groups] | ||||
result['latest_events'] = [ev.get_dict(request) for ev | ||||
in self.latest_events()] | ||||
exclude_keys_list = exclude_keys or [] | ||||
include_keys_list = include_keys or [] | ||||
d = {} | ||||
for k in result.keys(): | ||||
if (k not in exclude_keys_list and | ||||
(k in include_keys_list or not include_keys)): | ||||
d[k] = result[k] | ||||
return d | ||||
def __repr__(self): | ||||
return '<User: %s, id: %s>' % (self.user_name, self.id) | ||||
@property | ||||
def assigned_report_groups(self): | ||||
from appenlight.models.report_group import ReportGroup | ||||
r135 | resources = UserService.resources_with_perms(self, ['view'], resource_types=['application']) | |||
r0 | query = self.assigned_reports_relation | |||
rid_list = [r.resource_id for r in resources] | ||||
query = query.filter(ReportGroup.resource_id.in_(rid_list)) | ||||
query = query.limit(50) | ||||
return query | ||||
def feed_report(self, report): | ||||
""" """ | ||||
if not hasattr(self, 'current_reports'): | ||||
self.current_reports = [] | ||||
self.current_reports.append(report) | ||||
def send_digest(self, request, application, reports, since_when=None, | ||||
db_session=None): | ||||
db_session = get_db_session(db_session) | ||||
if not reports: | ||||
return True | ||||
if not since_when: | ||||
since_when = datetime.utcnow() | ||||
for channel in self.alert_channels: | ||||
if not channel.channel_validated or not channel.daily_digest: | ||||
continue | ||||
try: | ||||
channel.send_digest(resource=application, | ||||
user=self, | ||||
request=request, | ||||
since_when=since_when, | ||||
reports=reports) | ||||
except IntegrationException as e: | ||||
log.warning('%s' % e) | ||||
def latest_events(self): | ||||
return EventService.latest_for_user(self) | ||||