add tanken feature

1 files changed, 25 insertions, 0 deletions

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