diff options
Diffstat (limited to 'scripts/jira/jira-bug-closer/gerrit/streamparser.py')
-rw-r--r-- | scripts/jira/jira-bug-closer/gerrit/streamparser.py | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/scripts/jira/jira-bug-closer/gerrit/streamparser.py b/scripts/jira/jira-bug-closer/gerrit/streamparser.py new file mode 100644 index 00000000..e1900169 --- /dev/null +++ b/scripts/jira/jira-bug-closer/gerrit/streamparser.py @@ -0,0 +1,64 @@ +#!/usr/bin/env python3 +############################################################################# +## +## Copyright (C) 2019 The Qt Company Ltd. +## Contact: https://www.qt.io/licensing/ +## +## This file is part of the Quality Assurance module of the Qt Toolkit. +## +## $QT_BEGIN_LICENSE:GPL-EXCEPT$ +## Commercial License Usage +## Licensees holding valid commercial Qt licenses may use this file in +## accordance with the commercial license agreement provided with the +## Software or, alternatively, in accordance with the terms contained in +## a written agreement between you and The Qt Company. For licensing terms +## and conditions see https://www.qt.io/terms-conditions. For further +## information use the contact form at https://www.qt.io/contact-us. +## +## GNU General Public License Usage +## Alternatively, this file may be used under the terms of the GNU +## General Public License version 3 as published by the Free Software +## Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT +## included in the packaging of this file. Please review the following +## information to ensure the GNU General Public License requirements will +## be met: https://www.gnu.org/licenses/gpl-3.0.html. +## +## $QT_END_LICENSE$ +## +############################################################################# + +import json +from typing import Optional +from logger import logger + +log = logger(name="gerrit_stream_parser") + + +class GerritEvent: + def __init__(self, type: str, project: str, branch: str) -> None: + self.type = type + self.project = project + self.branch = branch + + def __eq__(self, other: object) -> bool: + return self.__dict__ == other.__dict__ + + def __repr__(self) -> str: + return "<GerritEvent '%s': '%s' '%s'>" % (self.type, self.project, self.branch) + + +class GerritStreamParser: + def parse(self, data: str) -> Optional[GerritEvent]: + try: + event = json.loads(data) + except json.decoder.JSONDecodeError: + log.exception('Invalid JSON: "%s"', data) + return None + eventType = event.get('type') + if eventType in ('comment-added', 'change-abandoned', 'change-deferred', 'change-merged', 'change-restored', + 'draft-published', 'merge-failed', 'patchset-created', 'reviewer-added'): + return GerritEvent(type=eventType, project=event['change']['project'], branch=event['change']['branch']) + if eventType in ('ref-updated',): + return GerritEvent(type=eventType, project=event['refUpdate']['project'], branch=event['refUpdate']['refName']) + log.warning('unhandled event type in gerrit ssh stream: "%s" data: "%s"', eventType, data) + return None |