python-csanal/calc_trips.py

76 lines
2.7 KiB
Python
Raw Normal View History

2018-01-14 02:23:42 +00:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import geopy.distance
import math
import sqlite3
ANSI_UP_DEL = u"\u001b[F\u001b[K"
class C2GCalcTrips:
def __init__(self):
self.dbfile = "car2go.db3"
self.conn = sqlite3.connect(self.dbfile)
self.c = self.conn.cursor()
self.c.execute("PRAGMA foreign_keys=on;")
self.run()
def run(self):
all_cars = self.load_cars()
print("Clearing table.")
self.c.execute("DELETE FROM trips;")
print("Going to analyse {} cars.".format(len(all_cars)))
inserter = self.conn.cursor()
num_trips = 0
cars_done = 0
for carId in all_cars:
is_occupied = False
self.c.execute("SELECT stamp, occupied, fuel, latitude, longitude FROM car_state WHERE carId=? ORDER BY stamp ASC;", (carId,))
for row in self.c:
if row[1] == 1:
is_occupied = True
occu_state = row
elif row[1] == 0 and is_occupied:
is_occupied = False
seconds = row[0] - occu_state[0]
minutes = seconds / 60
if seconds <= 70:
# probably a glitch
continue
price = math.ceil(minutes) * all_cars[carId]
old_spot = (occu_state[3], occu_state[4])
new_spot = (row[3], row[4])
distance = geopy.distance.distance(old_spot, new_spot).kilometers
dataset = (
carId,
occu_state[0],
row[0],
minutes,
distance,
(occu_state[2] - row[2]),
price
)
#print(repr(dataset))
inserter.execute("INSERT INTO trips (carId, stamp_departure, stamp_arrival, duration_minutes, distance_km, fuel_spent, price) VALUES (?,?,?,?,?,?,?);", dataset)
num_trips += 1
cars_done += 1
print("{}/{} cars done. {} trips found.".format(cars_done, len(all_cars), num_trips))
if cars_done < len(all_cars):
print(ANSI_UP_DEL, end="")
print("")
self.conn.commit()
self.c.execute("VACUUM;")
self.conn.close()
def load_cars(self):
print("Loading known cars from database.")
self.c.execute("SELECT c.carId, c.plate, cm.pricePerMinute FROM cars c LEFT JOIN car_models cm ON c.vinPrefix=cm.vinPrefix;")
result = {}
for row in self.c.fetchall():
result[row[0]] = row[2]
return result
if __name__=="__main__":
C2GCalcTrips()