addDevice.py

#!/usr/bin/env python3
import os
import configparser
import sys
from argparse import ArgumentParser

from HardwareCheckout.models import DeviceQueue, Role, DeviceType
from HardwareCheckout.config import db_path
from HardwareCheckout.auth import PasswordHasher

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker



parser = ArgumentParser()
parser.add_argument("-u", "--username", help="Device user name", required=False)
parser.add_argument("-p", "--password", help="Device user password", required=False)
parser.add_argument("-t", "--type", help="Device type", required=False)
parser.add_argument(
    "-i", "--ini", help="Ini file containing list of device users", required=False
)
args = parser.parse_args()

session = sessionmaker(bind=create_engine(db_path))
s = session()


def deviceAdd(username, password, devtype):
    device = s.query(Role).filter_by(name="Device").first()
    typeID = s.query(DeviceType).filter_by(name=devtype).first()

    if not typeID:
        print("Invalid type")
        exit(1)

    s.add(
        DeviceQueue(
            name=username,
            password=PasswordHasher.hash(password),
            state="want-provision",
            type=typeID.id,
        )
    )
    s.commit()


def iniParse(confPath, devType):
    config = configparser.ConfigParser()
    config.read(confPath)
    result = []

    for item in config.sections():
        innerList = []
        innerList.append(config[item]["username"])
        innerList.append(config[item]["password"])
        innerList.append(devType)
        result.append(innerList)

    return result


def csvParse(csvPath):
    with open(csvPath, "r") as ulist:
        try:
            users = list(filter(bool, ulist.read().split("\n")))
            result = []
            for user in users:
                components = user.split(",")
                if len(components) < 2:
                    print("Parameter missing in line {}".format(user))
                    parser.print_help(sys.stderr)
                    exit(1)
                else:
                    result.append(components)
        except:
            print("Couldn't read {}, {}".format(args.ini))
            parser.print_help(sys.stderr)
            exit(1)

    return result


def printHelp():
    print("Adding multiple devices:")
    print("python3 addDevice.py -i <path/to/inifile> -t <devicetype>")
    print()
    print("Add a single device:")
    print("python3 addDevice.py -u <devicename> -p <password> -t <devicetype>")
    exit(1)


def main():
    if args.ini and (args.username or args.password) and not args.type:
        print(
            "You cannot define username, password, but must define device type when you define a ini file!!"
        )
        parser.print_help(sys.stderr)
    elif args.username and args.password and args.type:
        deviceAdd(args.username, args.password, args.type)
    else:
        if not args.ini or not args.type:
            parser.print_help(sys.stderr)
            print("")
            printHelp()
        if not os.path.isfile(args.ini):
            print("Ini file {} doesn't exist!".format(args.ini))
            parser.print_help(sys.stderr)
            exit(1)
        # for parsing csv files replace this with
        # csvParse(csvPath) --> Note that csvParse expects device type in the file
        users = iniParse(args.ini, args.type)

        for user in users:
            deviceAdd(user[0], user[1], user[2])


if __name__ == "__main__":
    main()