1
0
Fork 0
mirror of https://gitlab.wikimedia.org/ladsgroup/Phabricator-maintenance-bot synced 2024-07-22 17:16:12 +02:00
Phabricator-maintenance-bot/gerrit.py
Amir Sarabadani 260ec45d9d gerrit.py changes
Avoid using ssh and the mess around it
2022-08-13 16:06:50 +02:00

139 lines
4.7 KiB
Python

"""
Copyright (C) 2019 Kunal Mehta <legoktm@member.fsf.org>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
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 Affero 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 <https://www.gnu.org/licenses/>.
"""
import json
import os
import subprocess
import tempfile
import urllib
from contextlib import contextmanager
dir_path = os.path.dirname(os.path.realpath(__file__))
with open(os.path.join(dir_path, 'gerrit-creds.json'), 'r') as f:
creds = json.loads(f.read())
@contextmanager
def cd(dirname):
cwd = os.getcwd()
os.chdir(dirname)
try:
yield dirname
finally:
os.chdir(cwd)
def gerrit_url(repo: str, user=None, ssh=False, password=None) -> str:
if user is not None:
prefix = user
if password is not None:
prefix += ':' + password
prefix += '@'
else:
prefix = ''
if ssh:
return 'ssh://{}gerrit.wikimedia.org:29418/{}'.format(prefix, repo)
else:
return 'https://{}gerrit.wikimedia.org/r/{}.git'.format(prefix, repo)
class ShellMixin:
def check_call(self, args: list, stdin='', env=None,
ignore_returncode=False) -> str:
debug = self.log if hasattr(self, 'log') else print
#debug('$ ' + ' '.join(args) + ' env: ' + str(env))
env = None
res = subprocess.run(
args,
input=stdin.encode(),
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
env=env
)
debug(res.stdout.decode())
if not ignore_returncode:
res.check_returncode()
return res.stdout.decode()
def clone(self, repo):
url = gerrit_url(repo, user=creds['name'])
self.check_call(['git', 'clone', url, 'repo', '--depth=1'])
os.chdir('repo')
self.check_call(['git', 'config', 'user.name', creds['name']])
self.check_call(['git', 'config', 'user.email', creds['email']])
self.check_call(['git', 'submodule', 'update', '--init'])
self.check_call(['mkdir', '-p', '.git/hooks'])
self.check_call(['curl', '-Lo', '.git/hooks/commit-msg', 'https://gerrit.wikimedia.org/r/tools/hooks/commit-msg'])
self.check_call(['chmod', '+x', '.git/hooks/commit-msg'])
def build_push_command(self, options: dict, ticket='') -> list:
per = '%topic=lsc'
if ticket:
per += '-' + ticket
for hashtag in options.get('hashtags', []):
per += ',t=' + hashtag
if options.get('vote'):
per += ',l=' + options['vote']
# If we're not automerging, vote V+1 to trigger jenkins (T254070)
if options.get('message'):
per += ',m=' + urllib.parse.quote_plus(options['message'])
branch = options.get('branch', 'master')
return ['git', 'push',
gerrit_url(options['repo'], creds['name'], password=creds['password']),
'HEAD:refs/for/' + branch + per]
class GerritBot(ShellMixin):
def __init__(self, name, commit_message):
self.name = name
self.commit_message = commit_message
def run(self):
with tempfile.TemporaryDirectory() as tmpdirname:
with cd(tmpdirname):
self.clone(self.name)
self.changes()
self.commit()
def changes(self):
files = [
'i18n/wikimediaprojectnames/en.json',
'i18n/wikimediaprojectnames/qqq.json'
]
for file_ in files:
with open(file_, 'r') as f:
result = json.load(f)
with open(file_, 'w') as f:
f.write(json.dumps(result, ensure_ascii=False,
indent='\t', sort_keys=True))
def commit(self):
self.check_call(['git', 'add', '.'])
with open('.git/COMMIT_EDITMSG', 'w') as f:
f.write(self.commit_message)
self.check_call(['git', 'commit', '-F', '.git/COMMIT_EDITMSG'])
self.check_call(
self.build_push_command({'repo': self.name})
)
if __name__ == "__main__":
gerritbot = GerritBot(
'mediawiki/extensions/WikimediaMessages',
"Order entries by alphabetical order\n\nThis would make creating automated patches easier")
gerritbot.run()