64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import json
|
||
|
|
import sqlite3
|
||
|
|
|
||
|
|
class C2GImport:
|
||
|
|
def __init__(self):
|
||
|
|
self.dbfile = "car2go.db3"
|
||
|
|
self.conn = sqlite3.connect(self.dbfile)
|
||
|
|
self.conn.row_factory = sqlite3.Row
|
||
|
|
self.c = self.conn.cursor()
|
||
|
|
self.c.execute("PRAGMA foreign_keys=on;")
|
||
|
|
self.run()
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
self.c.execute("SELECT DISTINCT stamp FROM car_state ORDER BY stamp ASC;");
|
||
|
|
stamps = []
|
||
|
|
for row in self.c:
|
||
|
|
stamps.append(row[0])
|
||
|
|
|
||
|
|
print("Found {} time stamps.".format(len(stamps)))
|
||
|
|
|
||
|
|
result = {}
|
||
|
|
used_keys = []
|
||
|
|
step_width = 1200 # 20 minutes between steps
|
||
|
|
cur_step = -1
|
||
|
|
cur_state = [[], []]
|
||
|
|
for stamp in stamps:
|
||
|
|
if stamp > cur_step + step_width:
|
||
|
|
if cur_step > 0:
|
||
|
|
result[cur_step] = cur_state
|
||
|
|
cur_state = [[], []] # first: removed cars, second: newly added cars
|
||
|
|
cur_step = stamp
|
||
|
|
used_keys.append(cur_step)
|
||
|
|
self.c.execute("SELECT occupied, latitude, longitude FROM car_state WHERE stamp=?;", (stamp,));
|
||
|
|
for row in self.c:
|
||
|
|
point = [row[1], row[2], 0.8]
|
||
|
|
if row[0]==0:
|
||
|
|
# Car got free
|
||
|
|
if point in cur_state[0]:
|
||
|
|
# car was marked as occupied, just remove marker
|
||
|
|
cur_state[0].remove(point)
|
||
|
|
else:
|
||
|
|
cur_state[1].append(point)
|
||
|
|
else:
|
||
|
|
# Car got occupied
|
||
|
|
if point in cur_state[1]:
|
||
|
|
# car was marked as free, just remove marker
|
||
|
|
cur_state[1].remove(point)
|
||
|
|
else:
|
||
|
|
cur_state[0].append(point)
|
||
|
|
|
||
|
|
result["keys"] = used_keys
|
||
|
|
|
||
|
|
with open("maps/map_data.js", "wt") as f:
|
||
|
|
f.write("window.data = ")
|
||
|
|
f.write(json.dumps(result))
|
||
|
|
f.write(";")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__=="__main__":
|
||
|
|
C2GImport()
|