system-config/tools/delete-gerrit-spam.py
James E. Blair b7550ab637 Commit transaction in delete-gerrit-spam
Also, print an indication when the DB is modified.

And get rid of a silly loop.

Change-Id: Ib0b25d51f389413ab255fd2332fcf211aeabec78
2016-07-13 10:51:48 -07:00

100 lines
3.8 KiB
Python
Executable File

#!/usr/bin/env python
# Copyright 2016 Red Hat, Inc.
#
# 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 MySQLdb
import ConfigParser
from prettytable import PrettyTable
import argparse
parser = argparse.ArgumentParser(description=
'Delete spam comments from Gerrit.')
parser.add_argument('account_id',
help='The numeric account id of the spammer')
parser.add_argument('--delete', action='store_true',
help='actually perform the deletion (the default'
'behavior is a dry run)')
args = parser.parse_args()
config = ConfigParser.RawConfigParser()
config.read('/root/.my.cnf')
host = config.get('client', 'host')
user = config.get('client', 'user')
pw = config.get('client', 'password')
db = MySQLdb.connect(host=host, user=user, passwd=pw, db='reviewdb')
cur = db.cursor()
# First, display / delete patch comments.
t = PrettyTable(['Change', 'Patchset', 'File', 'UUID', 'Date', 'Message'])
t.align = 'l'
cur.execute('select change_id, patch_set_id, file_name, uuid, written_on, '
'message from patch_comments where author_id=%s', args.account_id)
# Any of the patch comments we delete might also be a parent of some
# other comment. Keep track of the UUID of the comments we delete and
# check later to see if they are parents.
potential_parents = set()
for row in cur.fetchall():
t.add_row(row)
potential_parents.add(row[3])
if cur.rowcount:
print "Patch Comments -- To Be Deleted"
print t
if args.delete:
cur.execute('delete from patch_comments where author_id=%s',
args.account_id)
print "Deleted %s rows." % cur.rowcount
# If we are deleting some patch comments above, see if any of them are
# parents of other comments. If so, unparent the child comments so
# that they don't have a 'parent_uuid' field pointing to a nonexistent
# entry.
if potential_parents:
# This query formatting is so that we can let the client library
# substitute the value for each member of the UUID set we created
# above.
placeholders = ','.join(['%s'] * len(potential_parents))
query = ('select change_id, patch_set_id, file_name, uuid, written_on, '
'message from patch_comments where parent_uuid in (%s)' %
placeholders)
cur.execute(query, list(potential_parents))
t = PrettyTable(['Change', 'Patchset', 'File', 'UUID', 'Date', 'Message'])
t.align = 'l'
for row in cur.fetchall():
t.add_row(row)
if cur.rowcount:
print "Patch Comment Children -- To Be Unparented"
print t
if args.delete:
query = ('update patch_comments set parent_uuid=NULL where '
'parent_uuid in (%s)' % placeholders)
cur.execute(query, list(potential_parents))
print "Updated %s rows." % cur.rowcount
# Finally, display / delete any change messages.
t = PrettyTable(['Change', 'UUID', 'Date', 'Message'])
t.align = 'l'
cur.execute('select change_id, uuid, written_on, message from change_messages '
'where author_id=%s', args.account_id)
for row in cur.fetchall():
t.add_row(row)
if cur.rowcount:
print "Change Messages -- To Be Deleted"
print t
if args.delete:
cur.execute('delete from change_messages where author_id=%s',
args.account_id)
print "Deleted %s rows." % cur.rowcount
db.commit()