hashcode_2022/projectDistributions.py at master · StefPler/hashcode_2022 · 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
95
96
97
98
from dataStructures import Skill, Contributor, Project, ProjectAssignment

def produceOutput(projects):
    f = open("out.txt", "w")
    f.write(str(len(projects)) + "\n")
    for project in projects:
        f.write(f'{project.name}\n')
        for name in project.namesOfContributors:
            f.write(f'{name} ')
        f.write("\n")
    f.close()

fileA = 'a_an_example.in.txt'
fileB = 'b_better_start_small.in.txt'
fileC = 'c_collaboration.in.txt'
fileD = 'd_dense_schedule.in.txt'
fileE = 'e_exceptional_skills.in.txt'
fileF = 'f_find_great_mentors.in.txt'

with open(f'./tests/{fileB}', 'r') as f:
    lines = f.readlines()
line0 = lines[0].split(" ")
numberOfContributors = int(line0[0])
projects = int(line0[1])
print("Number of contributors is :", numberOfContributors,
      " and number of projects is: ", projects)

i = 0
linecount = 1
projlist = []
contributors = []
while (i < numberOfContributors):
    temp = lines[linecount].split()

    j = int(temp[1])
    linecount += 1
    skills = []
    for k in range(0, j):
        temp2 = lines[linecount].split()
        linecount += 1
        skill = Skill(temp2[0], temp2[1])
        skills.append(skill)
    i = i+1
    contributors.append(Contributor(temp[0], skills))

i = 0
while (i < projects):
    temp = lines[linecount].split()
    linecount += 1
    numberOfRequirements = temp[4]
    skills = []
    for j in range(0, int(numberOfRequirements)):
        temp2 = lines[linecount].split()
        skills.append(Skill(temp2[0], temp2[1]))
        linecount += 1
    projlist.append(
        Project(temp[0], temp[1], temp[2], temp[3], temp[4], skills))
    i += 1

for con in contributors:
    for sk in con.skills:
        print(
            f"contributor name: {con.name} skill name: {sk.name} skill level: {sk.level}")

for proj in projlist:
    print(f"Project name: {proj.name} days to complete: {proj.daysToComplete} score: {proj.score} Best Before: {proj.bestBefore} number of requirements {len(proj.requirements)}")
    for sk in proj.requirements:
        print(f"skill name: {sk.name} skill level: {sk.level}")

projectAssignments = []
contributorAvailability = list(contributors)
for proj in projlist:
    names = []
    projectBound = []
    for sk in proj.requirements:
        for con in contributors:
            if(con.findSkillLevel(sk.name, sk.level) and (con in contributorAvailability) and (con not in projectBound)):
                projectBound.append(con)
                names.append(con.name)
                break
    if(len(names) == len(proj.requirements)):
        projectAssignments.append(ProjectAssignment(proj.name, list(names)))
        for contributor in projectBound:
            try:
                contributorAvailability.remove(contributor)
            except ValueError:
                print(f'contributor {contributor.name} not available')
    projectBound.clear()
    names.clear()

print("----------------------------------------------------")

# for project in projectAssignments:
#     print(f"\n* {project.name} *")
#     for name in project.namesOfContributors:
#         print(f"- {name}")

produceOutput(projectAssignments)