deploy.py

#!/usr/bin/env python3
# Copyright 2021 Jonas Beck

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

from B12_webscraping_tools import *
import matplotlib.pyplot as plt
from datetime import datetime
import subprocess as cmd


def update_plots():
    data = import_logged_data()
    data_clean = prune_data(data)

    fig, ax = plot_capacity_matrix(
        data_clean, y_axis="weekday", x_axis="min", x_increment=30, figsize=(15, 5)
    )
    ax.set_title("Average traffic at the B12.")
    plt.savefig("capacity_matrix.png", facecolor="white")
    plt.close()

    # plotting fails if B12 is not open
    try:
        ax = plot_avg_capacity(
            data_clean, time_binsize="30T", start_datetime=datetime.today().date(), figsize=(15, 5)
        )
        ax.set_title("Todays traffic at the B12.")
        plt.savefig("capacity_timeline.png", facecolor="white")
        plt.close()
    except TypeError:
        print("[Failure] %s - No data to plot. The B12 might be closed" % datetime.strftime(
            datetime.now(), "% H: % M, % m/%d/%Y"))
    try:
        # Push everything to git
        msg = "Update to plots. @ %s" % datetime.strftime(
            datetime.now(), "%H:%M, %m/%d/%Y")
        cmd.run("git add capacity_matrix.png capacity_timeline.png",
                check=True, shell=True, stdout=cmd.DEVNULL)
        cmd.run(f"git commit -m '{msg}'", check=True,
                shell=True, stdout=cmd.DEVNULL)
        cmd.run("git push", check=True,
                shell=True, stdout=cmd.DEVNULL)

        print("[Success] Updated plots were pushed to GitHub.")
    except:
        print("[Failure] Updated plots could not be pushed to GitHub.")


deploy_data_logger(update_interval=5, update_func=update_plots, run_every=3)