pyIdentityManagement/processSheet.py at master · OCLC-Developer-Network/pyIdentityManagement · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
## need to import the relevant files
import argparse
import yaml
from src import handle_files, process_data, make_requests
import sys
import string

with open("config.yml", 'r') as stream:
    config = yaml.safe_load(stream)

def processArgs():
    try:
        parser = argparse.ArgumentParser()
        parser.add_argument('--itemFile', required=True, help='File you want to process')
        parser.add_argument('--operation', required=True, choices=['getUsers', 'createUsers', 'deleteUsers', 'findUsers', 'addCorrelationInfoToUsers', 'findUserCorrelationInfo'], help='Operation to run: getUsers, createUsers, deleteUsers, findUsers, addCorrelationInfoToUsers')
        parser.add_argument('--outputDir', required=True, help='Directory to save output to')

        args = parser.parse_args()
        return args
    except SystemExit:
        raise
def validateOperation(csv_read, operation):
    if operation == "getUsers":
        if 'principalId' not in csv_read.columns.values.tolist():
            sys.exit("CSV does not contain column principalId")
    elif operation == "createUsers":
        requiredFields = ['name']
        if not all(item in csv_read.columns.values.tolist() for item in requiredFields):
            sys.exit("CSV does not contain column name")
    elif operation == "deleteUsers":
        if 'principalId' not in csv_read.columns.values.tolist():
            sys.exit("CSV does not contain column principalId")
    elif operation == "findUsers":
        if 'barcode' not in csv_read.columns.values.tolist():
            sys.exit("CSV does not contain column barcode")
    elif operation == "findUserCorrelationInfo":
        requiredFields = ['barcode', 'sourceSystem']
        if not all(item in csv_read.columns.values.tolist() for item in requiredFields):
            sys.exit("CSV does not contain columns barcode and sourceSystem")
    elif operation == "addCorrelationInfoToUsers":
        requiredFields = ['principalId', 'sourceSystem', 'sourceSystemId']
        if not all(item in csv_read.columns.values.tolist() for item in requiredFields):
            sys.exit("CSV does not contain columns principalId, sourceSystem, sourceSystemId")

def process(args):
    item_file = handle_files.readFileFromLocal(args.itemFile)

    operation = args.operation
    output_dir = args.outputDir

    csv_read = handle_files.loadCSV(item_file)
    try:
        validateOperation(csv_read, operation)

        # get a token
        scope = ['SCIM']
        try:
            oauth_session = make_requests.createOAuthSession(config, scope)

            config.update({"oauth-session": oauth_session})
            processConfig = config
            if operation == "getUsers":
                csv_read = process_data.getUsers(processConfig, csv_read)
            elif operation == "createUsers":
                csv_read = process_data.createUsers(processConfig, csv_read)
            elif operation == "deleteUsers":
                csv_read = process_data.deleteUsers(processConfig, csv_read)
            elif operation == "findUsers":
                csv_read = process_data.findUsers(processConfig, csv_read)
            elif operation == "findUserCorrelationInfo":
                csv_read = process_data.findUserCorrelationInfo(processConfig, csv_read)
            elif operation == "addCorrelationInfoToUsers":
                csv_read = process_data.addUserCorrelationInfo(processConfig, csv_read)
            return handle_files.saveFileLocal(csv_read, output_dir)

        except BaseException as err:
            result = str(err)
            return result
    except SystemExit as err:
        return err

if __name__ == '__processSheet__':
    try:
        args = processArgs()
        print(process(args))
    except SystemExit:
        print("Invalid Operation specified")
else:
    try:
        args = processArgs()
        print(process(args))
    except SystemExit:
        print("Invalid Operation specified")