def send_event():
db = DbClient()
geo_location = request.args.get('origin')
lat = geo_location.split(',')[0]
lat = str(float("{0:.3f}".format(float(lat))))
lng = geo_location.split(',')[1]
lng = str(float("{0:.3f}".format(float(lng))))
driver_id = request.args.get('driver_id')
status = request.args.get('trashStatus')
res = db.sensors.get_sensor_by_geo_location(lat, lng)
if res:
update_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M")
if status == '0':
status = 'Broken'
elif status == '1':
status = 'Missing'
else:
status = 'online'
try:
db.sensors.update_sensor_status_by_id(id=res['id'],
status=status,
update_date=update_time)
return 'sensor updated', 200
except Exception as e:
print(e)
return 'failed to update sensor', 400
def update_sensor_by_id(data):
db = DbClient()
sensor_id = data.split("_")[0]
capacity = data.split("_")[1]
db.sensors.update_sensor_capacity_by_id(sensor_id, capacity)
db.statistics.insert_statistics(
sensor_id,
datetime.datetime.now().strftime("%Y-%m-%d %H:%M"), capacity)
return jsonify({'status': 'ok'}), 200
def download(data):
db = DbClient()
if not validate_token(request):
return render_template("error_page.html")
user = get_data_by_token(request.cookies.get('token', None))
if data == 'full':
sensors = db.sensors.get_sensors()
sensors = [tuple(sensor.values()) for sensor in sensors]
utils.write_sensors_to_csv(sensors)
return send_file('export.csv', attachment_filename='export.csv')
def insert_driver():
db = DbClient()
content = request.json
# if db.drivers.get_driver_by_id(content['id']):
# pass
if db.drivers.get_driver_by_id(content['id']):
db.drivers.update_driver_by_id(content['id'], content['name'],
content['lat'], content['lng'],
content['truck_size']) # update
else:
db.drivers.insert_driver(content['id'], content['name'],
content['lat'], content['lng'],
content['truck_size']) # insert new
return "success"
def insert_sensor_date():
db = DbClient()
content = request.json
# fake_date = '2019-05-11'
# date = datetime.datetime.now().strftime("%Y-%m-%d %H:%M")
# db_handler.insert_statistics(content['id'], fake_date, content['capacity'])
date = content['date']
db.statistics.insert_statistics(content['id'], date, content['capacity'])
if db.sensors.get_sensor_by_id(content['id']):
db.sensors.update_sensor_by_id(content['id'], content['address'],
content['capacity'], content['lat'],
content['lng'], content['status'], date)
else:
db.sensors.insert_sensor(content['id'], content['address'],
content['capacity'], content['lat'],
content['lng'], content['status'], date)
return "ok", 200
def new_databins():
db = DbClient()
if not validate_token(request):
return render_template("error_page.html")
user = get_data_by_token(request.cookies.get('token', None))
if request.method == 'POST':
result = request.form
if result['radio-stacked'] == "capacity":
sensors = db.sensors.get_sensor_between_capacity(
result['capacity'], 100)
if not sensors:
return render_template("databins.html")
elif result['radio-stacked'] == 'id':
sensors = db.sensors.get_sensor_by_id(result['Bin_ID'])
if sensors is not None:
pass
else:
return render_template("databins.html")
else:
sensors = db.sensors.get_sensor_by_address(result['address'])
if not sensors:
return render_template("databins.html")
else:
sensors_low = db.sensors.get_sensor_between_capacity(0, 25)
if not sensors_low:
sensors_low = []
sensors_mid = db.sensors.get_sensor_between_capacity(26, 75)
if not sensors_mid:
sensors_mid = []
sensors_full = db.sensors.get_sensor_between_capacity(76, 100)
if not sensors_full:
sensors_full = []
sensors = sensors_low + sensors_mid + sensors_full
if type(sensors) is dict:
tuple_sensors = [
tuple(sensors.values()),
]
sensors = [
sensors,
]
else:
tuple_sensors = [tuple(sensor.values()) for sensor in sensors]
utils.write_sensors_to_csv(tuple_sensors)
return render_template("databins.html", sensors=sensors)
def get_charts_data():
db = DbClient()
res = []
sensors = db.sensors.get_sensors()
days = db.statistics.get_statistcs_by_days(7)
avg_days = []
sum_days = []
for day in days:
temp_dict_avg = {
'day':
day['date'].strftime("%Y-%m-%d"),
'avg':
int(
db.statistics.get_avg_statatics_from_day(
day['date'].strftime("%Y-%m-%d"))['avg(capacity)'])
}
temp_dict_sum = {
'day':
day['date'].strftime("%Y-%m-%d"),
'sum':
int(
db.statistics.get_sum_volume_from_day(
day['date'].strftime("%Y-%m-%d"))['sum(capacity)'])
}
avg_days += [
temp_dict_avg,
]
sum_days += [
temp_dict_sum,
]
online = len(db.sensors.get_sensors_by_status("online"))
offline = len(db.sensors.get_sensors_by_status("offline"))
response = {
'avg_array': avg_days,
'sum_array': sum_days,
'online': online,
'offline': offline,
'threshold': conf.trash_threshold
}
return jsonify(response), 200
def get_trash_bins_to_pickup():
db = DbClient()
res_json = {
'arr': [],
'driver': {
'lat': 32.1668891,
'lng': 34.8266644
},
'capacity': 0,
'count_bins': 0
}
total_capacity = 0
res = db.sensors.get_sensor_over_x_capacity(conf.trash_threshold)
count_bins = 0
if res:
for x in res:
res_json['arr'].append({'lat': x['lat'], 'lng': x['lng']})
total_capacity += x['capacity']
count_bins += 1
res_json['capacity'] = total_capacity
res_json['count_bins'] = count_bins
return jsonify(res_json), 200
def register():
db = DbClient()
if request.method == 'POST':
try:
data = request.values
user = data.get('username')
password = data.get('password')
if not is_password_valid(password):
raise PasswordInvalid()
else:
password = str(
encrypt(conf.PASSWORD_ENCRYPTION_KEY,
str.encode(password)))
if not (user and password):
raise EmptyForm()
user_data = db.users.get_user_by_username(user)
if user_data:
raise UserAlreadyExists(user)
else:
db.users.add_user(user, password)
return 'ok', 200 # DO what you want is good
except PasswordInvalid as e:
logger.warning(e.__str__())
return render_template("error_page.html", error_msg=e.__str__())
except UserAlreadyExists as e:
logger.warning(e.__str__())
return render_template("error_page.html", error_msg=e.__str__())
except (EmptyForm, ValueError) as e:
logger.warning(e.__str__())
return render_template("error_page.html", error_msg=e.__str__())
except Exception as e:
logger.exception(f'Failed register')
return render_template("error_page.html", error_msg=e.__str__())
else:
return render_template('register.html')
def new_stats():
db = DbClient()
if not validate_token(request):
return render_template("error_page.html")
user = get_data_by_token(request.cookies.get('token', None))
return render_template("analytics.html")
def new_calc():
db = DbClient()
if not validate_token(request):
return render_template("error_page.html")
user = get_data_by_token(request.cookies.get('token', None))
username = db.users.get_user_by_id(user['user_id'])['user']
config_trashold = conf.trash_threshold
present_treshold = conf.trash_threshold
if request.method == 'POST':
if request.form.get('range'):
present_percent = request.form.get('range')
capacity = int(present_percent)
present_treshold = capacity
if request.form.get('trashold'):
result = request.form.get('trashold')
if '%' in result:
threshold = int(result[:-1])
else:
threshold = int(result)
conf.trash_threshold = int(threshold)
capacity = int(conf.trash_threshold)
config_trashold = capacity
present_treshold = capacity
else:
capacity = conf.trash_threshold
pickup_sensors = db.sensors.get_sensor_over_x_capacity(capacity)
if not pickup_sensors:
pickup_sensors = []
remain_sensors = db.sensors.get_sensor_under_x_capacity(capacity)
if not remain_sensors:
remain_sensors = []
threshold = conf.trash_threshold
risk_sensors = []
total_trash_to_pickup = 0
for sensor in remain_sensors:
fill_avg = utils.get_avg_fill_per_sensor(
db.statistics.get_sensor_stat_by_id(sensor['id']))
if int(sensor['capacity']) + fill_avg >= threshold:
risk_sensors.append(sensor)
sensor['fill_avg'] = fill_avg
for sensor in pickup_sensors:
fill_avg = utils.get_avg_fill_per_sensor(
db.statistics.get_sensor_stat_by_id(sensor['id']))
total_trash_to_pickup += sensor['capacity']
sensor['fill_avg'] = fill_avg
calculate_capacity = sensor['capacity']
days_until_full = 0
while calculate_capacity <= capacity:
calculate_capacity += fill_avg
days_until_full += 1
sensor['day_until_full'] = days_until_full
truck_needed = round(total_trash_to_pickup / 1000)
if truck_needed == 0:
truck_needed = 1
return render_template("calc.html",
sensors=pickup_sensors,
capacityint=capacity,
total_to_pickup=len(pickup_sensors),
config_trashold=config_trashold,
present_treshold=present_treshold,
risked=len(risk_sensors),
truck_needed=truck_needed,
unrisked=len(pickup_sensors) + len(remain_sensors) -
len(risk_sensors),
username=username)
def main():
db = DbClient()
try:
if not validate_token(request):
return render_template(
"error_page.html",
error_msg=
"You are not login, please enter to login page to access to the system"
)
user = get_data_by_token(request.cookies.get('token', None))
if request.method == 'POST':
sensors_low = []
sensors_mid = []
sensors_full = []
sensors, checked_list = utils.get_sensors_by_main_filters(request)
for sensor in sensors:
sensor = [
sensor,
]
if int(sensor[0][2]) < 25:
sensors_low = sensors_low + sensor
elif int(sensor[0][2]) < 75:
sensors_mid = sensors_mid + sensor
else:
sensors_full = sensors_full + sensor
else:
sensors = []
sensors_low = db.sensors.get_sensor_between_capacity(0, 25)
if sensors_low:
sensors += sensors_low
sensors_mid = db.sensors.get_sensor_between_capacity(26, 75)
if sensors_mid:
sensors += sensors_mid
sensors_full = db.sensors.get_sensor_between_capacity(76, 100)
if sensors_full:
sensors += sensors_full
sensors = [tuple(sensor.values()) for sensor in sensors]
utils.write_sensors_to_csv(sensors)
sensors = [[x[1], x[4], x[5], x[3], x[2], x[0]] for x in sensors]
return render_template("index.html",
sensors=sensors,
persent_count=len(sensors),
total_count=len(sensors))
utils.write_sensors_to_csv(sensors)
sensor_count = db.sensors.get_count_sensors()['count(*)']
sensors = [[x[1], x[4], x[5], x[3], x[2], x[0]] for x in sensors]
return render_template("index.html",
sensors=sensors,
capacity_empty=checked_list[0],
capacity_mid=checked_list[1],
capacity_full=checked_list[2],
over_trashold=checked_list[3],
below_trashold=checked_list[4],
ConnectedBins=checked_list[5],
FailedBins=checked_list[6],
persent_count=len(sensors),
total_count=sensor_count)
except Exception as e:
print(traceback.format_exc())
print(e)
def get_sensor_by_id(sensor_id):
db = DbClient()
res = db.sensors.get_sensor_by_id(sensor_id)
return jsonify(res), 200
def get_count_sensors():
db = DbClient()
res = db.sensors.get_count_sensors() # ['count(*)']
return res
def create_all_tables():
db = DbClient()
# db_handler.create_all_tables()
return 'ok', 200
def login():
'''
POST login to system and generate JWT token
:param request: flask request object
'''
db = DbClient()
if validate_token(request):
sensors = []
sensors_low = db.sensors.get_sensor_between_capacity(0, 25)
if sensors_low:
sensors += sensors_low
sensors_mid = db.sensors.get_sensor_between_capacity(26, 75)
if sensors_mid:
sensors += sensors_mid
sensors_full = db.sensors.get_sensor_between_capacity(76, 100)
if sensors_full:
sensors += sensors_full
sensors = [tuple(sensor.values()) for sensor in sensors]
utils.write_sensors_to_csv(sensors)
sensors = [[x[1], x[4], x[5], x[3], x[2], x[0]] for x in sensors]
return render_template("index.html",
sensors=sensors,
persent_count=len(sensors),
total_count=len(sensors))
if request.method == 'POST':
try:
data = request.values
user = data.get('username')
password = data.get('password')
if not (user and password):
logger.info(
f'Login failed on {request.remote_addr}, missing credentials'
)
raise InvalidCredentials(user, password)
user_data = db.users.get_user_by_username(user)
if not user_data:
raise UserNotExists(user)
elif not (str.encode(password) == decrypt(
conf.PASSWORD_ENCRYPTION_KEY, user_data['password'])):
raise InvalidCredentials(user)
else:
token = generate_token(user_data['id'])
logger.info(f'Token for user {user} created. token: {token}')
sensors = []
sensors_low = db.sensors.get_sensor_between_capacity(0, 25)
if sensors_low:
sensors += sensors_low
sensors_mid = db.sensors.get_sensor_between_capacity(26, 75)
if sensors_mid:
sensors += sensors_mid
sensors_full = db.sensors.get_sensor_between_capacity(76, 100)
if sensors_full:
sensors += sensors_full
sensors = [tuple(sensor.values()) for sensor in sensors]
utils.write_sensors_to_csv(sensors)
sensors = [[x[1], x[4], x[5], x[3], x[2], x[0]]
for x in sensors]
resp = make_response(
render_template("index.html", sensors=sensors))
resp.set_cookie('token', token)
return resp
except UserNotVerified as e:
logger.warning(e.__str__())
return render_template("error_page.html", error_msg=e.__str__())
except UserNotExists as e:
logger.warning(e.__str__())
error_result = e.__str__()
return render_template("error_page.html", error_msg=error_result)
except InvalidCredentials as e:
logger.warning(e.__str__())
return render_template("error_page.html", error_msg=e.__str__())
except Exception as e:
logger.exception(f'Failed login from {request.remote_addr}')
return render_template("error_page.html", error_msg=e.__str__())
else:
return render_template('login.html')
def get_sensors_by_main_filters(request):
db = DbClient()
checked_list = []
sensors = []
if request.form.get('empty_Bins'):
sensors_low = db.sensors.get_sensor_between_capacity(0, 25)
if sensors_low:
sensors += sensors_low
checked_list.append('checked')
else:
sensors_low = []
checked_list.append('')
if request.form.get('mid_Bins'):
sensors_mid = db.sensors.get_sensor_between_capacity(26, 75)
if sensors_mid:
sensors += sensors_mid
checked_list.append('checked')
else:
sensors_mid = []
checked_list.append('')
if request.form.get('full_Bins'):
sensors_full = db.sensors.get_sensor_between_capacity(76, 100)
if sensors_full:
sensors += sensors_full
checked_list.append('checked')
else:
sensors_full = []
checked_list.append('')
if request.form.get('over_trashold'):
over_trashold = db.sensors.get_sensor_between_capacity(
conf.trash_threshold, 100)
if over_trashold:
sensors += over_trashold
checked_list.append('checked')
else:
over_trashold = []
checked_list.append('')
if request.form.get('below_trashold'):
sensors_below = db.sensors.get_sensor_between_capacity(
0, conf.trash_threshold)
if sensors_below:
sensors += sensors_below
checked_list.append('checked')
else:
sensors_below = []
checked_list.append('')
if request.form.get('ConnectedBins'):
ConnectedBins = db.sensors.get_sensors_by_status("online")
if ConnectedBins:
sensors += ConnectedBins
checked_list.append('checked')
else:
ConnectedBins = []
checked_list.append('')
if request.form.get('FailedBins'):
FailedBins = db.sensors.get_sensors_by_status("offline")
if FailedBins:
sensors += FailedBins
checked_list.append('checked')
else:
FailedBins = []
checked_list.append('')
sensors = [tuple(sensor.values()) for sensor in sensors]
sensors = list(dict.fromkeys(sensors))
return sensors, checked_list