SSD_Assignment3/q1.py at master · mansi-k/SSD_Assignment3 · 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
import json


def load_org_data():
    with open('org.json') as f:
        orgdata = json.load(f)
    return orgdata

def take_input():
    empno_inp_list = [x.strip() for x in input().split()]
    if len(empno_inp_list) < 2:
        print("Enter atleast 2 employees")
        exit(0)
    return empno_inp_list

def check_empno(empno):
    if (not empno.isdecimal()) or len(empmo)!=3 or int(empmo)<1 or int(empmo)>999:
        print("Employee name",empno,"is not a number in range 001-999")
        exit(0)

def flatten_orgdata(emplev_inp_list):
    l = 0
    flat_orgdata = []
    for level in orgdata:
        lev_dict = {} # dict per level
        if -1 in emplev_inp_list: # if any input emp is still not parsed
            for emp in orgdata[level]:
                check_empno(emp['name'])
                if 'parent' in emp:
                    lev_dict[emp['name']] = emp['parent']
                else:
                    lev_dict[emp['name']] = -1
                if -1 in emplev_inp_list:
                    for i in range(len(empno_inp_list)):
                        if emplev_inp_list[i]==-1 and empno_inp_list[i]==emp['name']:
                            emplev_inp_list[i] = l
                else:
                    break
            l+=1
            flat_orgdata.append(lev_dict)
        else:
            break
    return flat_orgdata, emplev_inp_list

def get_common_leaders(empno_inp_list,emplev_inp_list,flat_orgdata):
    parents = {}
    for i in range(len(empno_inp_list)):
        p = empno_inp_list[i]
        l = emplev_inp_list[i]
        filter_parents = {}
        while l>0:
            p = flat_orgdata[l][p]
            l -= 1
            if i==0:
                filter_parents[p] = l
            elif p in parents:
                filter_parents[p] = l
        parents = filter_parents
    return parents

def print_lowest_common_leader(empno_inp_list,emplev_inp_list,parents):
    maxlev = -1
    commonboss = ''
    for p in parents:
        l = parents[p]
        if l>maxlev:
            maxlev = l
            commonboss = p

    if maxlev==-1:
        print("No common leader")
    else:
        print(commonboss)
        for i in range(len(empno_inp_list)):
            print(commonboss,"is",emplev_inp_list[i]-maxlev,"levels above",empno_inp_list[i])

orgdata = load_org_data()
empno_inp_list = take_input()
emplev_inp_list = [-1 for x in empno_inp_list]
flat_orgdata, emplev_inp_list = flatten_orgdata(emplev_inp_list) # dict for all levels
parents = get_common_leaders(empno_inp_list,emplev_inp_list,flat_orgdata)
print_lowest_common_leader(empno_inp_list,emplev_inp_list,parents)