88 lines
2.2 KiB
Python
88 lines
2.2 KiB
Python
import logging
|
|
|
|
from db_connector import RemoteDB
|
|
import matplotlib.pyplot as plt
|
|
import pandas as pd
|
|
import plotly.express as px
|
|
import kaleido
|
|
import numpy as np
|
|
|
|
logging.getLogger("matplotlib").setLevel(logging.WARNING)
|
|
|
|
|
|
# TODO
|
|
def plt_acc_by_year(db):
|
|
acc_year_sql = """
|
|
SELECT COUNT(*), accidentyear FROM accidents
|
|
GROUP BY accidentyear
|
|
ORDER BY accidentyear;
|
|
"""
|
|
result = db.execute_query(acc_year_sql)
|
|
result_df = pd.DataFrame(result)
|
|
|
|
fig = px.bar(result_df, y='year', x='count', orientation='h', title='No. of Accidents per Year')
|
|
fig.write_image("fig/acc_by_year.png")
|
|
fig.write_html("html/acc_by_year.png")
|
|
|
|
|
|
def plt_acc_by_weekday(db):
|
|
acc_weekday_sql = f"""
|
|
SELECT COUNT(*) AS count, accidentweekday_en AS weekday
|
|
FROM accidents
|
|
GROUP BY accidentweekday_en
|
|
ORDER BY COUNT(*);
|
|
"""
|
|
|
|
result = db.execute_query(acc_weekday_sql)
|
|
result_df = pd.DataFrame(result)
|
|
|
|
fig = px.bar(result_df, y='weekday', x='Count', orientation='h', title='No. of Accidents per Weekday')
|
|
fig.write_image("fig/acc_by_weekday.png")
|
|
fig.write_html("html/acc_by_weekday.html")
|
|
|
|
|
|
def plt_acc_by_day_year(db):
|
|
acc_year_day_sql = """
|
|
SELECT accidentyear AS year, accidentweekday_en AS weekday, COUNT(*) AS count
|
|
FROM accidents
|
|
GROUP BY weekday, year
|
|
ORDER BY year, COUNT(*);
|
|
"""
|
|
|
|
|
|
def plt_acc_by_daytime(db):
|
|
acc_weekday_sql = f"""
|
|
SELECT COUNT(*) AS count, accidenthour AS hour
|
|
FROM accidents
|
|
GROUP BY accidenthour
|
|
ORDER BY COUNT(*);
|
|
"""
|
|
|
|
result = db.execute_query(acc_weekday_sql)
|
|
result_df = pd.DataFrame(result)
|
|
|
|
fig = px.bar(result_df, y='hour', x='count', orientation='h')
|
|
fig.write_image("fig/acc_by_day.png")
|
|
fig.write_html("html/acc_by_day.html")
|
|
|
|
|
|
# Utilities ===========================================================================================================
|
|
def save_as_barplot(df, xname, yname, orientation, file_name):
|
|
pass
|
|
|
|
|
|
def save_as_html():
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
remote_db = RemoteDB()
|
|
try:
|
|
# plt_acc_by_year(remote_db)
|
|
# plt_acc_by_weekday(remote_db)
|
|
plt_acc_by_daytime(remote_db)
|
|
except Exception as e:
|
|
print(f"Exception {e} in plots.py")
|
|
finally:
|
|
remote_db.close()
|