Ross
465 lines
15 KiB
Python
465 lines
15 KiB
Python
from pyomo.environ import *
|
|
from pyomo.opt import SolverFactory
|
|
from shifts import SingleShift, RotaBuilder, RotaResults, days
|
|
from workers import Worker
|
|
|
|
from copy import deepcopy
|
|
|
|
import math
|
|
|
|
import scipy.stats as stats
|
|
|
|
import statistics
|
|
|
|
import datetime
|
|
import itertools
|
|
import operator
|
|
|
|
use_neos = False
|
|
|
|
weeks_to_rota = 52
|
|
|
|
sites = ("truro", "exeter", "torquay", "barnstaple", "plymouth")
|
|
|
|
Rota = RotaBuilder(
|
|
datetime.date(2020, 9, 7),
|
|
weeks_to_rota=weeks_to_rota,
|
|
balance_offset_modifier=1,
|
|
max_weekend_frequency=2,
|
|
max_night_frequency=2,
|
|
)
|
|
|
|
rota_collections = {
|
|
"current": deepcopy(Rota),
|
|
# "extended proc nights only": deepcopy(Rota),
|
|
# "no weekends": deepcopy(Rota),
|
|
# "no twighlights": deepcopy(Rota),
|
|
# "no nights": deepcopy(Rota),
|
|
# "nights + proc twighlights3 + normal weekends": deepcopy(Rota),
|
|
# "nights + proc twighlights4 + normal weekends": deepcopy(Rota),
|
|
# "nights + proc twighlights + proc weekends": deepcopy(Rota),
|
|
}
|
|
|
|
rota_collections["current"].add_shifts(
|
|
SingleShift(("exeter",), "exeter_twilight", 12.5, days[:5]),
|
|
SingleShift(("truro",), "truro_twilight", 12.5, days[:5]),
|
|
SingleShift(("torquay",), "torquay_twilight", 12.5, days[:5]),
|
|
SingleShift(("plymouth",), "plymouth_twilight", 12.5, days[:5]),
|
|
SingleShift(("exeter",), "weekend_exeter", 12.5, days[5:], assign_as_block=True),
|
|
SingleShift(("truro",), "weekend_truro", 12.5, days[5:], assign_as_block=True),
|
|
SingleShift(("torquay",), "weekend_torquay", 12.5, days[5:], assign_as_block=True),
|
|
SingleShift(
|
|
("plymouth",),
|
|
"weekend_plymouth",
|
|
8,
|
|
days[5:],
|
|
workers_required=2,
|
|
assign_as_block=True,
|
|
),
|
|
SingleShift(
|
|
(sites),
|
|
"night_weekday",
|
|
12.25,
|
|
days[:4],
|
|
balance_offset=5,
|
|
balance_weighting=1,
|
|
workers_required=3,
|
|
force_as_block=False,
|
|
rota_on_nwds=True,
|
|
constraints=["night"],
|
|
),
|
|
SingleShift(
|
|
(sites),
|
|
"night_weekend",
|
|
12.25,
|
|
days[4:],
|
|
balance_offset=4,
|
|
balance_weighting=1,
|
|
workers_required=3,
|
|
force_as_block=False,
|
|
rota_on_nwds=True,
|
|
constraints=["night"],
|
|
),
|
|
)
|
|
|
|
# rota_collections['extended proc nights only'].add_shifts(
|
|
# #SingleShift((sites), "night", 12.5, days, balance_weighting=1, workers_required=3, rota_on_nwds=True),
|
|
# SingleShift((sites),
|
|
# "night_weekday",
|
|
# 13,
|
|
# days[:4],
|
|
# balance_offset=5,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# SingleShift((sites),
|
|
# "night_weekend",
|
|
# 13,
|
|
# days[4:],
|
|
# balance_offset=4,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# )
|
|
|
|
# rota_collections['no weekends'].add_shifts(
|
|
# SingleShift(("exeter", ), "exeter_twilight", 12.5, days[:5]),
|
|
# SingleShift(("truro", ), "truro_twilight", 12.5, days[:5]),
|
|
# SingleShift(("torquay", ), "torquay_twilight", 12.5, days[:5]),
|
|
# SingleShift(("plymouth", ), "plymouth_twilight", 12.5, days[:5]),
|
|
# #SingleShift((sites), "night", 12.5, days, balance_weighting=1, workers_required=3, rota_on_nwds=True),
|
|
# SingleShift((sites),
|
|
# "night_weekday",
|
|
# 13,
|
|
# days[:4],
|
|
# balance_offset=5,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# SingleShift((sites),
|
|
# "night_weekend",
|
|
# 13,
|
|
# days[4:],
|
|
# balance_offset=4,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# )
|
|
|
|
# rota_collections['no twighlights'].add_shifts(
|
|
# SingleShift(("exeter", ),
|
|
# "weekend_exeter",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("truro", ),
|
|
# "weekend_truro",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("torquay", ),
|
|
# "weekend_torquay",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("plymouth", ),
|
|
# "weekend_plymouth",
|
|
# 8,
|
|
# days[5:],
|
|
# workers_required=2,
|
|
# assign_as_block=True),
|
|
# #SingleShift((sites), "night", 12.5, days, balance_weighting=1, workers_required=3, rota_on_nwds=True),
|
|
# SingleShift((sites),
|
|
# "night_weekday",
|
|
# 13,
|
|
# days[:4],
|
|
# balance_offset=5,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# SingleShift((sites),
|
|
# "night_weekend",
|
|
# 13,
|
|
# days[4:],
|
|
# balance_offset=4,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# )
|
|
|
|
# rota_collections['no nights'].add_shifts(
|
|
# SingleShift(("exeter", ), "exeter_twilight", 12.5, days[:5]),
|
|
# SingleShift(("truro", ), "truro_twilight", 12.5, days[:5]),
|
|
# SingleShift(("torquay", ), "torquay_twilight", 12.5, days[:5]),
|
|
# SingleShift(("plymouth", ), "plymouth_twilight", 12.5, days[:5]),
|
|
# SingleShift(("exeter", ),
|
|
# "weekend_exeter",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("truro", ),
|
|
# "weekend_truro",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("torquay", ),
|
|
# "weekend_torquay",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("plymouth", ),
|
|
# "weekend_plymouth",
|
|
# 8,
|
|
# days[5:],
|
|
# workers_required=2,
|
|
# assign_as_block=True),
|
|
# #SingleShift((sites), "night", 12.5, days, balance_weighting=1, workers_required=3, rota_on_nwds=True),
|
|
# )
|
|
|
|
# rota_collections['nights + proc twighlights4 + normal weekends'].add_shifts(
|
|
# SingleShift((sites), "proc_twilight", 12.5, days[:5], workers_required=4),
|
|
# SingleShift(("exeter", ),
|
|
# "weekend_exeter",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("truro", ),
|
|
# "weekend_truro",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("torquay", ),
|
|
# "weekend_torquay",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("plymouth", ),
|
|
# "weekend_plymouth",
|
|
# 8,
|
|
# days[5:],
|
|
# workers_required=2,
|
|
# assign_as_block=True),
|
|
# #SingleShift((sites), "night", 12.5, days, balance_weighting=1, workers_required=3, rota_on_nwds=True),
|
|
# SingleShift((sites),
|
|
# "night_weekday",
|
|
# 12.25,
|
|
# days[:4],
|
|
# balance_offset=5,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# SingleShift((sites),
|
|
# "night_weekend",
|
|
# 12.25,
|
|
# days[4:],
|
|
# balance_offset=4,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# )
|
|
|
|
# rota_collections['nights + proc twighlights3 + normal weekends'].add_shifts(
|
|
# SingleShift((sites), "proc_twilight", 12.5, days[:5], workers_required=3),
|
|
# SingleShift(("exeter", ),
|
|
# "weekend_exeter",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("truro", ),
|
|
# "weekend_truro",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("torquay", ),
|
|
# "weekend_torquay",
|
|
# 12.5,
|
|
# days[5:],
|
|
# assign_as_block=True),
|
|
# SingleShift(("plymouth", ),
|
|
# "weekend_plymouth",
|
|
# 8,
|
|
# days[5:],
|
|
# workers_required=2,
|
|
# assign_as_block=True),
|
|
# #SingleShift((sites), "night", 12.5, days, balance_weighting=1, workers_required=3, rota_on_nwds=True),
|
|
# SingleShift((sites),
|
|
# "night_weekday",
|
|
# 12.25,
|
|
# days[:4],
|
|
# balance_offset=5,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# SingleShift((sites),
|
|
# "night_weekend",
|
|
# 12.25,
|
|
# days[4:],
|
|
# balance_offset=4,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# )
|
|
|
|
# rota_collections['nights + proc twighlights + proc weekends'].add_shifts(
|
|
# SingleShift((sites), "proc_twilight", 12.5, days[:5], workers_required=3),
|
|
# SingleShift((sites), "proc_weekends", 12.5, days[5:], workers_required=3),
|
|
# #SingleShift((sites), "night", 12.5, days, balance_weighting=1, workers_required=3, rota_on_nwds=True),
|
|
# SingleShift((sites),
|
|
# "night_weekday",
|
|
# 12.25,
|
|
# days[:4],
|
|
# balance_offset=5,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# SingleShift((sites),
|
|
# "night_weekend",
|
|
# 12.25,
|
|
# days[4:],
|
|
# balance_offset=4,
|
|
# balance_weighting=1,
|
|
# workers_required=3,
|
|
# force_as_block=False,
|
|
# rota_on_nwds=True,
|
|
# constraints=["night"]),
|
|
# )
|
|
|
|
|
|
load_leave = True
|
|
|
|
if load_leave:
|
|
import leave
|
|
|
|
leave = leave.load_leave()
|
|
|
|
# Import trainee data
|
|
import csv
|
|
|
|
with open("trainees.csv", newline="") as f:
|
|
reader = csv.reader(f, delimiter=",", quotechar="|")
|
|
next(reader) # skip header
|
|
|
|
n = 0
|
|
for row in reader:
|
|
n = n + 1
|
|
# print(row)
|
|
name, site, grade, fte, nwd, end_date, oop, extra = row
|
|
|
|
# Ignore trainees if fte == 0 (what are they doing here anyway)
|
|
if fte == "0":
|
|
continue
|
|
|
|
nwds = nwd.split("/") if nwd else None
|
|
end_date = end_date if end_date else None
|
|
oop = oop.split("-") if oop else None
|
|
|
|
if load_leave:
|
|
if name not in leave:
|
|
raise ValueError("{} not found it leave requests".format(name))
|
|
|
|
leave_requests = [i[0] for i in leave[name] if i[0] != ""]
|
|
|
|
# print(nwds, end_date, oop)
|
|
for rot in rota_collections:
|
|
|
|
Rota = rota_collections[rot]
|
|
Rota.add_worker(
|
|
Worker(
|
|
Rota,
|
|
n,
|
|
name,
|
|
site.lower(),
|
|
int(grade[2]),
|
|
int(fte),
|
|
nwds,
|
|
end_date,
|
|
oop,
|
|
not_available_to_work=leave_requests,
|
|
)
|
|
)
|
|
|
|
with open("rotas.html", "w") as f:
|
|
f.write(
|
|
"""<html>
|
|
<head>
|
|
<link rel="stylesheet" type="text/css" href="timetable.css">
|
|
<script src="https://ajax.googleapis.com/ajax/libs/jquery/3.5.1/jquery.min.js"></script>
|
|
<script src="timetable.js" defer></script>
|
|
</head>
|
|
<body>"""
|
|
)
|
|
|
|
for rot in rota_collections:
|
|
print("Running rota: ", rot)
|
|
Rota = rota_collections[rot]
|
|
|
|
Rota.build_workers()
|
|
|
|
print("-Building model")
|
|
Rota.build_model()
|
|
print("-Building model: complete")
|
|
|
|
opt = SolverFactory("cbc") # choose a solver
|
|
# opt = SolverFactory('ipopt') # choose a solver
|
|
|
|
if use_neos:
|
|
print("-Solving with neos")
|
|
solver_manager = SolverManagerFactory("neos") # Solve in neos server
|
|
results = solver_manager.solve(
|
|
Rota.model, opt=opt, logfile="{}.log".format(rot)
|
|
)
|
|
else:
|
|
print("-Solving with local cbc")
|
|
results = opt.solve(
|
|
Rota.model,
|
|
tee=True,
|
|
options={
|
|
"seconds": 10200,
|
|
"allow": 2000,
|
|
},
|
|
logfile="{}.log".format(rot),
|
|
) # solve the model with the, options="seconds=60" selected solver
|
|
|
|
results.solver.status
|
|
|
|
ResultsHolder = RotaResults(Rota, results)
|
|
|
|
week_table = ResultsHolder.get_work_table() # list with the required workers
|
|
# worker_timetable = ResultsHolder.get_worker_timetable()
|
|
worker_timetable_brief = ResultsHolder.get_worker_timetable_brief(show_prefs=False)
|
|
|
|
worker_shift_summary = ResultsHolder.get_shift_summary()
|
|
|
|
ResultsHolder.export_rota_to_csv("rotas/" + rot)
|
|
with open("rotas.html", "a") as f:
|
|
f.write("<hr>")
|
|
f.write("<h2>{}</h2>".format(rot))
|
|
# f.write("----------------------------")
|
|
f.write("\n\n")
|
|
|
|
for s in Rota.shifts:
|
|
f.write(
|
|
"<div id='{}-{}-shift-summary' class='shift-summary'>{}</div>".format(
|
|
rot, s.name, s.get_shift_summary()
|
|
)
|
|
)
|
|
f.write("\n")
|
|
f.write("\n\n")
|
|
|
|
f.write(ResultsHolder.get_worker_timetable_html(table_name=rot))
|
|
f.write("\n\n")
|
|
f.write(
|
|
"<pre class='worker-shift-summary'>{}</pre>".format(worker_shift_summary)
|
|
)
|
|
f.write("\n\n\n")
|
|
# workers_no_pref = ResultsHolder.get_no_preference(Rota.model.no_pref) # list with the non-satisfied workers (work on Sat but not on Sun)
|
|
# worker_night_block = ResultsHolder.get_night_blocks()
|
|
# multi = ResultsHolder.get_multinight()
|
|
|
|
# break
|
|
|
|
with open("rotas.html", "a") as f:
|
|
f.write("</body></html>")
|