105 lines
4.2 KiB
Python
Executable File
105 lines
4.2 KiB
Python
Executable File
# 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.
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import argparse
|
|
import glob
|
|
import hashlib
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import time
|
|
|
|
from six.moves.urllib.request import urlopen
|
|
|
|
from openstack_election import owners
|
|
from openstack_election import utils
|
|
|
|
# Exclude the system / bot accounts
|
|
# OpenStack Release Bot:
|
|
# curl https://review.openstack.org/accounts/22816
|
|
# OpenStack Proposal Bot
|
|
# curl https://review.openstack.org/accounts/11131
|
|
MINUS_BOT_ACCOUNTS = ["-i", "11131", "-i", "22816"]
|
|
|
|
|
|
def main():
|
|
start = utils.conf['timeframe']['start']
|
|
end = utils.conf['timeframe']['end']
|
|
description = ('Generate electorate rolls')
|
|
parser = argparse.ArgumentParser(description)
|
|
parser.add_argument('--tag', dest='tag', default=utils.conf['tag'],
|
|
help=('The governance tag to validate against. '
|
|
'Default: %(default)s'))
|
|
parser.add_argument('--rolls_dir', default="./rolls/%s" % (
|
|
utils.conf['tag']), help=('The output directory. '
|
|
'Default: %(default)s'))
|
|
parser.add_argument('--before', default=end.strftime("%Y-%m-%d %H-%M-%S"),
|
|
help=('The start date. Default: %(default)s'))
|
|
parser.add_argument('--after', default=start.strftime("%Y-%m-%d"),
|
|
help=('The end date. Default: %(default)s'))
|
|
parser.add_argument('--without-stable', dest='with_stable', default=True,
|
|
action='store_false', help=('Do not include stables'))
|
|
parser.add_argument('--cached-owners-script', default=False,
|
|
action='store_true', help=('Do not fetch the script'))
|
|
|
|
args = parser.parse_args()
|
|
|
|
print("Run information:")
|
|
print(" - TAG : %s" % args.tag)
|
|
print(" - ROLLS : %s" % args.rolls_dir)
|
|
print(" - STABLE : %s" % args.with_stable)
|
|
print(" - BEFORE : %s" % args.before)
|
|
print(" - AFTER : %s" % args.after)
|
|
|
|
if os.path.isdir(args.rolls_dir):
|
|
shutil.rmtree(args.rolls_dir)
|
|
os.makedirs(args.rolls_dir, 0o700)
|
|
|
|
os.chdir(os.path.dirname(args.rolls_dir))
|
|
print("Starting roll generation @%s" % time.ctime())
|
|
owners.main(["owners.py", "-a", args.after, "-b", args.before,
|
|
"-o", args.tag, "-r", args.tag] + MINUS_BOT_ACCOUNTS)
|
|
print("Finished roll generation @%s" % time.ctime())
|
|
|
|
if args.with_stable:
|
|
tmp_dir = tempfile.mkdtemp(prefix='election.')
|
|
print("Starting (Stable) roll generation @%s" % time.ctime())
|
|
owners.main(["owners.py", "-a", args.after, "-b", args.before,
|
|
"-o", tmp_dir, "-r", args.tag, "-n",
|
|
"-s", "branch:^stable/.*"] + MINUS_BOT_ACCOUNTS)
|
|
print("Finished (Stable) roll generation @%s" % time.ctime())
|
|
shutil.copy("%s/_electorate.txt" % tmp_dir,
|
|
"./%s/stable_branch_maintenance.txt" % args.tag)
|
|
shutil.copy("%s/_all_owners.yaml" % tmp_dir,
|
|
"./%s/stable_branch_maintenance.yaml" % args.tag)
|
|
shutil.rmtree(tmp_dir)
|
|
|
|
print("Electoral rolls SHA256sum")
|
|
with open("%s.shasums" % args.tag, "wb") as ofile:
|
|
for fname in sorted(glob.glob("./%s/*.txt" % args.tag)):
|
|
# Sort the list
|
|
with open(fname) as list_file:
|
|
roll = sorted(list_file.readlines())
|
|
content = "".join(roll).encode('utf-8')
|
|
with open(fname, "wb") as list_file:
|
|
list_file.write(content)
|
|
# Generate the hash
|
|
h = hashlib.sha256(content).hexdigest()
|
|
s = "%s %s" % (h, fname)
|
|
print(s)
|
|
ofile.write(("%s\n" % s).encode('utf-8'))
|