##// END OF EJS Templates
events: fix bug, 'url' was not in dict
events: fix bug, 'url' was not in dict

File last commit:

r392:b65452d3 default
r392:b65452d3 default
Show More
base.py
71 lines | 2.1 KiB | text/x-python | PythonLexer
dan
events: add serialization .to_dict() to events based on marshmallow
r379 # Copyright (C) 2016-2016 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/
from datetime import datetime
from marshmallow import Schema, fields
from pyramid.threadlocal import get_current_request
from rhodecode.lib.utils2 import AttributeDict
dan
events: add comment on SYSTEM_USER
r391 # this is a user object to be used for events caused by the system (eg. shell)
dan
events: add serialization .to_dict() to events based on marshmallow
r379 SYSTEM_USER = AttributeDict(dict(
username='__SYSTEM__'
))
class UserSchema(Schema):
"""
Marshmallow schema for a user
"""
username = fields.Str()
class RhodecodeEventSchema(Schema):
"""
Marshmallow schema for a rhodecode event
"""
utc_timestamp = fields.DateTime()
dan
events: fix bugs with serialization of repo/pr events and add tests for those cases
r389 actor = fields.Nested(UserSchema)
actor_ip = fields.Str()
dan
events: remove attribute from schema, redundant
r390 name = fields.Str()
dan
events: add serialization .to_dict() to events based on marshmallow
r379
class RhodecodeEvent(object):
"""
Base event class for all Rhodecode events
"""
MarshmallowSchema = RhodecodeEventSchema
def __init__(self):
self.request = get_current_request()
self.utc_timestamp = datetime.utcnow()
@property
dan
events: fix bugs with serialization of repo/pr events and add tests for those cases
r389 def actor(self):
dan
events: add serialization .to_dict() to events based on marshmallow
r379 if self.request:
return self.request.user.get_instance()
return SYSTEM_USER
@property
dan
events: fix bugs with serialization of repo/pr events and add tests for those cases
r389 def actor_ip(self):
dan
events: add serialization .to_dict() to events based on marshmallow
r379 if self.request:
return self.request.user.ip_addr
return '<no ip available>'
def as_dict(self):
dan
events: fix bug, 'url' was not in dict
r392 return self.MarshmallowSchema().dump(self).data