python-csanal/old/analyse.py
2018-01-14 03:23:42 +01:00

131 lines
4.9 KiB
Python
Raw Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import datetime
import geopy.distance
import json
import math
import os
import glob
import re
import sys
class C2GAnalyse:
def __init__(self):
self.state = {}
self.in_use = []
self.total_minutes = 0.0
self.total_money = 0.0
self.total_distance = 0.0
self.run()
def find_car_type(self, vin):
# 1..3 = manufacturer code
# 4..6 = general type
# 7 = body form
# 8..9 = engine type
# 10 = steering (1=left, 2=right)
# 11 = manufacturing facilty
seg = vin[0:9]
typelist = {
"WDD117342": ["CLA 180", 0.34],
"WDC156912": ["GLA 180 CDI", 0.34],
"WDC156942": ["GLA 180", 0.34],
"WDC156943": ["GLA 200", 0.34],
"WDD176012": ["A 180 CDI", 0.31],
"WDD176042": ["A 180", 0.31],
"WDD176043": ["A 200", 0.31],
"WDD246242": ["B 180", 0.34],
"WME451334": ["smart fortwo 451", 0.26],
"WME451390": ["smart fortwo 451 ed", 0.29],
"WME451391": ["smart fortwo 451 ed", 0.29],
"WME453342": ["smart fortwo 453", 0.26],
}
if not seg in typelist:
print("Missing vehicle type for {} (VIN {})!".format(seg, vin))
return typelist[seg]
def run(self):
for f in sorted(glob.glob(os.path.join("../data/", "*.json"))):
self.scanfile(f)
print("{} cars seen in total.".format(len(self.state)))
print("Total kilometres driven: {}".format(self.total_distance))
print("Total minutes driven: {}".format(self.total_minutes))
print("Total money made: {}".format(self.total_money))
def scanfile(self, filename):
stamp = re.search(r'(\d{4}-\d\d-\d\d_\d{6})', filename)
stamp_str = stamp.group(1)
stamp_dt = datetime.datetime.strptime(stamp_str, "%Y-%m-%d_%H%M%S")
print("File: {} ({}, Totals: {:.3f}km, {:.1f}mins, {:.2f}€)".format(filename, stamp_str, self.total_distance, self.total_minutes, self.total_money), end="")
jf = open(filename, "r")
doc = json.load(jf)
jf.close()
root = doc["placemarks"]
print(" ({} cars available)".format(len(root)))
self.parse_cars(stamp_dt, root)
def parse_cars(self, stamp, placemarks):
not_in_use = []
for pm in placemarks:
self.update_car(stamp, pm)
not_in_use.append(pm["name"])
for car_id in self.state:
if not car_id in not_in_use:
if not car_id in self.in_use:
print("{} OCCUPIED!".format(car_id))
self.state[car_id]["occupied"] = stamp
self.in_use.append(car_id)
else:
if car_id in self.in_use:
#print("{} RETURNED!".format(car_id))
if "occupied" in self.state[car_id]:
del self.state[car_id]["occupied"]
self.in_use.remove(car_id)
def update_car(self, stamp, new_data):
car_id = new_data["name"]
if car_id in self.state:
old_data = self.state[car_id]
car_type = self.find_car_type(new_data["vin"])
if new_data["address"] != old_data["address"]:
## CAR MOVED
old_spot = (old_data["coordinates"][1], old_data["coordinates"][0])
new_spot = (new_data["coordinates"][1], new_data["coordinates"][0])
distance = geopy.distance.distance(old_spot, new_spot).kilometers
if not "occupied" in old_data:
old_data["occupied"] = "unknown"
minutes = -1
else:
minutes = (stamp - old_data["occupied"]).total_seconds() / 60
self.total_minutes += minutes
self.total_money += math.ceil(minutes) * car_type[1]
self.total_distance += distance
print("{} {} moved for {:.3f}km/{:.1f}mins: {} ({}%, {}, {}) {} -> {} {} ({}%, {}, {})".format(
car_type[0],
car_id,
distance,
minutes,
old_data["address"],
old_data["fuel"],
old_data["interior"],
old_data["exterior"],
old_data["occupied"],
stamp,
new_data["address"],
new_data["fuel"],
new_data["interior"],
new_data["exterior"]
))
elif new_data["fuel"] != old_data["fuel"]:
print("{} FUEL CHANGE: {} -> {}".format(car_id, old_data["fuel"], new_data["fuel"]))
if new_data["vin"] != old_data["vin"]:
print("{} ################# VIN CHANGE!!!".format(car_id))
self.state[car_id] = new_data
if __name__=="__main__":
C2GAnalyse()