1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
#!/usr/bin/env python3
# Copyright (C) 2019 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
import json
from logger import get_logger
log = get_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)
def is_branch_update(self) -> bool:
if self.type != 'ref-updated':
return False
if 'staging' in self.branch:
return False
if self.branch.startswith('refs/changes/'):
return False
return True
class GerritStreamParser:
def parse(self, data: str) -> GerritEvent:
try:
event = json.loads(data)
except json.decoder.JSONDecodeError:
log.exception('Invalid JSON: "%s"', data)
return GerritEvent(type='invalid', project='', branch='')
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', 'reviewer-deleted'):
return GerritEvent(type=eventType, project=event['change']['project'], branch=event['change']['branch'])
if eventType in ('ref-replication-scheduled', 'ref-replicated', 'ref-replication-done'):
return GerritEvent(type=eventType, project=event['project'], branch=event['ref'])
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 GerritEvent(type='invalid', project='', branch='')
|
