Reduce services feature and reformatting

This commit is contained in:
2016-04-14 07:32:06 +02:00
parent 8f99b798bb
commit fb521194f7
9 changed files with 51 additions and 197 deletions

View File

@@ -1,51 +1,39 @@
'''
"""
@author: shing19m
'''
@author: jlusiardi
"""
from datetime import datetime
import sqlite3
class Db(object):
'''
"""
classdocs
'''
"""
def __init__(self, db_file):
'''
"""
Constructor
'''
"""
self.db = sqlite3.connect(db_file)
def __del__(self):
self.db.close()
def getAllPitStops(self):
def get_all_pit_stops(self):
return self._perform_query('select * from pitstops order by id asc')
def getAllServices(self):
return self._perform_query('select * from services')
def getLastPitStop(self):
def get_last_pit_stop(self):
pitstops = self._perform_query('select * from pitstops order by id desc limit 1')
if len(pitstops) == 0:
return {'date': datetime.strftime(datetime.now(), '%Y-%m-%d'), 'odometer': 0, 'litres': 0}
return pitstops[0]
def get_service_warning_info(self):
info = self._perform_query('select (odometer_planned - (select odometer from pitstops order by id desc limit 1)) km_left, tasks from services where date is null order by odometer_planned asc limit 1;')
if len(info) == 0:
return None
return info[0]
def get_next_undone_service(self):
services = self._perform_query('select * from services where date is null limit 1')
if len(services) == 0:
return None
return services[0]
def add_pit_stop(self, date, odometer, litres):
self.db.execute('insert into pitstops (date, odometer, litres) values (?, ?, ?)', [date, odometer, litres])
self.db.commit()
def get_salt_for_user(self, user):
salt = self._perform_query_param('select salt from users where name = ?', [user])
@@ -81,10 +69,6 @@ class Db(object):
result.append(row_result)
return result
def addPitStop(self, date, odometer, litres):
self.db.execute('insert into pitstops (date, odometer, litres) values (?, ?, ?)', [date, odometer, litres])
self.db.commit()
def init_db(self, resource):
with resource as f:
sql_commands = f.read()