mirror of
https://github.com/flobz/psa_car_controller.git
synced 2026-08-22 17:36:15 +00:00
33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
import sqlite3
|
|
from datetime import datetime
|
|
import pytz
|
|
|
|
callback_fct = None
|
|
default_db_file = 'info.db'
|
|
def convert_datetime(st):
|
|
return datetime.strptime(st.decode("utf-8"), "%Y-%m-%d %H:%M:%S+00:00").replace(tzinfo=pytz.UTC)
|
|
|
|
|
|
def update_callback():
|
|
if callback_fct is not None:
|
|
callback_fct()
|
|
return
|
|
|
|
def get_db(db_file=default_db_file):
|
|
sqlite3.register_converter("DATETIME", convert_datetime)
|
|
conn = sqlite3.connect(db_file, detect_types=sqlite3.PARSE_DECLTYPES | sqlite3.PARSE_COLNAMES)
|
|
conn.row_factory = sqlite3.Row
|
|
conn.execute("CREATE TABLE IF NOT EXISTS position (Timestamp DATETIME PRIMARY KEY, VIN TEXT, longitude REAL, "
|
|
"latitude REAL, mileage REAL, level INTEGER, level_fuel INTEGER, moving BOOLEAN, temperature INTEGER);")
|
|
try:
|
|
conn.execute("ALTER TABLE position ADD level_fuel INTEGER;")
|
|
except:
|
|
pass
|
|
conn.execute("CREATE TABLE IF NOT EXISTS battery (start_at DATETIME PRIMARY KEY,stop_at DATETIME,VIN TEXT, "
|
|
"start_level INTEGER, end_level INTEGER, co2 INTEGER, kw INTEGER);")
|
|
conn.create_function("update_trips", 0, update_callback)
|
|
conn.execute("CREATE TEMP TRIGGER IF NOT EXISTS update_trigger AFTER INSERT ON position BEGIN "
|
|
"SELECT update_trips(); END;")
|
|
conn.commit()
|
|
return conn
|