From 8aa41f02c4c56a4e17db4ba2d51bc251e9c97982 Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Fri, 27 Oct 2017 17:27:11 +0200 Subject: add tanken feature --- tanken.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 tanken.py (limited to 'tanken.py') diff --git a/tanken.py b/tanken.py new file mode 100644 index 0000000..4685fee --- /dev/null +++ b/tanken.py @@ -0,0 +1,25 @@ +#!/bin/env python3 + +def tanken(drives, cost): + passagers = {} + distance = 0. + + for d in drives: + try: + distance += int(d[0]) + except: + return None, "Lines have to start with the driven distance!" + for p in d[1:]: + if p not in passagers: + passagers[p] = [int(d[0]),0] + else: + passagers[p][0] += int(d[0]) + + c = cost/distance + + for d in drives: + c_d = c * int(d[0]) / (len(d) - 1) + for p in d[1:]: + passagers[p][1] += c_d + + return passagers, None -- cgit v1.2.3