Add a tool to delete (redact) gerrit comments

This adds a CLI tool that allows a gerrit admin to list and delete
change messages and comments.

Change-Id: I3600a59520c1f13a24f99f04eaf9ceb17af67fff
This commit is contained in:
James E. Blair 2024-02-06 09:45:46 -08:00
parent af14ca1aba
commit 9d62c9abf9

172
tools/gerrit-delete-comment.py Executable file
View File

@ -0,0 +1,172 @@
#!/usr/bin/env python3
#
# Copyright 2024 Acme Gating, LLC
#
# 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
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# 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.
import sys
import json
import requests
import argparse
BREAK = '------------------------------------------------------------'
class Gerrit:
def __init__(self, url, username, password):
if url.endswith('/'):
url = url[:-1]
self.url = url
self.session = requests.Session()
self.session.auth = requests.auth.HTTPBasicAuth(username, password)
def get(self, path):
url = f'{self.url}{path}'
r = self.session.get(url,
headers={'Accept': 'application/json',
'Accept-Encoding': 'gzip'})
if r.status_code == 200:
ret = json.loads(r.text[4:])
return ret
def post(self, path, data):
url = f'{self.url}{path}'
r = self.session.post(
url,
data=json.dumps(data).encode('utf8'),
headers={'Content-Type': 'application/json;charset=UTF-8'})
if r.status_code > 400:
raise Exception("POST to %s failed with http code %s (%s)",
path, r.status_code, r.text)
if r.text and len(r.text) > 4:
return json.loads(r.text[4:])
def list_messages(self, change):
data = self.get(f'/a/changes/{change}/messages')
for msg in data:
name = msg['real_author'].get('name')
username = msg['real_author'].get('username')
email = msg['real_author'].get('email')
date = msg['date']
msgid = msg['id']
print(BREAK)
print(f'Id : {msgid}')
print(f'Author: {name} ({username}) <{email}>')
print(f'Date : {date}')
print(msg['message'])
def delete_message(self, change, message_id, reason):
self.post(
f'/a/changes/{change}/messages/{message_id}/delete',
{'reason': reason})
print("Deleted")
def list_comments(self, change, revision):
data = self.get(f'/a/changes/{change}/revisions/{revision}/comments')
for path, comments in data.items():
for msg in comments:
name = msg['author'].get('name')
username = msg['author'].get('username')
email = msg['author'].get('email')
line = msg.get('line')
msgid = msg['id']
print(BREAK)
print(f'Id : {msgid}')
print(f'Author: {name} ({username}) <{email}>')
print(f'File : {path} line {line}')
print(msg['message'])
def delete_comment(self, change, revision, comment_id, reason):
self.post(
f'/a/changes/{change}/revisions/{revision}/'
f'comments/{comment_id}/delete',
{'reason': reason})
print("Deleted")
class App:
def __init__(self):
self.parser = argparse.ArgumentParser()
p = self.parser
p.add_argument('url',
help='Gerrit HTTP url')
p.add_argument('username',
help='Username of Gerrit administrator')
p.add_argument('password',
help='Password of Gerrit administrator')
subparsers = p.add_subparsers(title='commands',
help='valid commands')
cmd_list_messages = subparsers.add_parser(
'list-messages', help='List change messages')
cmd_list_messages.add_argument('change',
help='Change number')
cmd_list_messages.set_defaults(func=self.list_messages)
cmd_delete_message = subparsers.add_parser(
'delete-message', help='Delete change message')
cmd_delete_message.add_argument('change',
help='Change number')
cmd_delete_message.add_argument('message_id',
help='Message ID')
cmd_delete_message.add_argument(
'reason',
help='Reason for removal (will replace message)')
cmd_delete_message.set_defaults(func=self.delete_message)
cmd_list_comments = subparsers.add_parser(
'list-comments', help='List change comments')
cmd_list_comments.add_argument('change',
help='Change number')
cmd_list_comments.add_argument('revision',
help='Change revision')
cmd_list_comments.set_defaults(func=self.list_comments)
cmd_delete_comment = subparsers.add_parser(
'delete-comment', help='Delete change comment')
cmd_delete_comment.add_argument('change',
help='Change number')
cmd_delete_comment.add_argument('revision',
help='Change revision')
cmd_delete_comment.add_argument('comment_id',
help='Comment ID')
cmd_delete_comment.add_argument(
'reason',
help='Reason for removal (will replace comment)')
cmd_delete_comment.set_defaults(func=self.delete_comment)
args = p.parse_args()
self.args = args
self.gerrit = Gerrit(args.url, args.username, args.password)
if 'func' not in args:
p.print_help()
sys.exit(1)
args.func()
def list_messages(self):
self.gerrit.list_messages(self.args.change)
def delete_message(self):
self.gerrit.delete_message(self.args.change,
self.args.message_id, self.args.reason)
def list_comments(self):
self.gerrit.list_comments(self.args.change, self.args.revision)
def delete_comment(self):
self.gerrit.delete_comment(self.args.change, self.args.revision,
self.args.comment_id, self.args.reason)
if __name__ == '__main__':
App()