61 lines
1.9 KiB
Python
Executable File
61 lines
1.9 KiB
Python
Executable File
#!/bin/python3
|
|
import os
|
|
import sys
|
|
import logging
|
|
import requests
|
|
import json
|
|
import argparse
|
|
|
|
sys.path.append(os.path.abspath("../"))
|
|
from subject import main
|
|
|
|
from lib import encryption_functs
|
|
|
|
logging.basicConfig(format='%(levelname)s\t- %(message)s')
|
|
logger = logging.getLogger()
|
|
logger.setLevel(logging.INFO)
|
|
|
|
state = main(sys.argv)
|
|
|
|
# session file - username - name - email - credentials file
|
|
def addSubject(args):
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument("-k", '--key', nargs=1, help="Path to the key file")
|
|
parser.add_argument("-r", '--repo', nargs=1, help="Address:Port of the repository")
|
|
parser.add_argument("-v", '--verbose', help="Increase verbosity", action="store_true")
|
|
|
|
parser.add_argument('session', nargs='?', default=None)
|
|
parser.add_argument('username', nargs='?', default=None)
|
|
parser.add_argument('name', nargs='?', default=None)
|
|
parser.add_argument('email', nargs='?', default=None)
|
|
parser.add_argument('credentials', nargs='?', default=None)
|
|
|
|
args = parser.parse_args()
|
|
|
|
if len(args) != 5:
|
|
logger.error("Need session file, username, name, email and credentials file.")
|
|
sys.exit(-1)
|
|
|
|
if (not os.path.isfile(args.session)):
|
|
logger.error("File '" + args.session + "' not found.")
|
|
sys.exit(-1)
|
|
|
|
if (not os.path.isfile(args.credentials)):
|
|
logger.error("File '" + args.file + "' not found")
|
|
sys.exit(-1)
|
|
|
|
subject = {'username' : args.username, 'name' : args.name, 'email' : args.email, 'credentials' : args.credentials}
|
|
|
|
try:
|
|
req = requests.post(f'http://{state['REP_ADDRESS']}/subject/add', json=json.dumps(subject))
|
|
req.raise_for_status()
|
|
|
|
except requests.exceptions.RequestException as errex:
|
|
logger.error("Failed to obtain response from server.")
|
|
sys.exit(1)
|
|
|
|
sys.exit(0)
|
|
|
|
if __name__ == '__main__':
|
|
addSubject(sys.argv[1:]) |