pysteve/cli/mkelection.py

#!/usr/bin/env/python
#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You 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 os, sys, random, time

path = os.path.abspath(os.getcwd() + "/..")
sys.path.append(path)

version = 2
if sys.hexversion < 0x03000000:
    import ConfigParser as configparser
else:
    import configparser
    version = 3
    
config = configparser.RawConfigParser()
config.read('../steve.cfg')

homedir = config.get("general", "homedir")

from lib import election
import argparse

parser = argparse.ArgumentParser(description='Command line options.')
parser.add_argument('--id', dest='id', type=str, nargs=1,
                   help='Election ID: If defined, attempt to create an election using this as the election ID (OPTIONAL)')
parser.add_argument('--owner', required=True, dest='owner', nargs=1,
                   help='Sets the owner of this election, as according to steve.cfg [REQUIRED]')
parser.add_argument('--title', required=True, dest='title', nargs=1,
                   help='Sets the title (name) of the election [REQUIRED]')
parser.add_argument('--monitors', dest='monitors', nargs=1,
                   help='Comma-separated list of email addresses to use for monitoring (OPTIONAL)')
parser.add_argument('--public', dest='public', action='store_true',
                   help='If set, create the election as a public (open) election where anyone can vote (OPTIONAL)')

args = parser.parse_args()
eid = args.id[0] if args.id and len(args.id) > 0 else None
if not eid:
    eid = ("%08x" % int(time.time() * random.randint(1,999999999999)))[0:8]
print("Creating new election with ID %s" % eid)
monitors = []
if args.monitors:
    monitors = args.monitors[0].split(",")
if not config.has_option("karma", args.owner[0]):
    print("Sorry, I could not find '%s' in the karma list in steve.cfg!" % args.owner[0])
    sys.exit(-1)
else:
    election.createElection(eid, args.title[0], args.owner[0], monitors, 0, 0, args.public)
    
    print("Election created!")
    print("Election ID: %s" % eid)
    print("Election Admin URL: %s/edit_election.html?%s" % (config.get("general", "rooturl"), eid))