craftbeerpi3/modules/logs/__init__.py

import datetime
import os
from flask import Blueprint, request, send_from_directory, json
from flask_classy import FlaskView, route
from modules.core.core import cbpi


class LogView(FlaskView):

    _log_directory = "./logs"

    @route('/', methods=['GET'])
    def get_all_logfiles(self):
        """
        Get a list of all Log Files
        ---
        tags:
          - logs
        responses:
          200:
            description: List of all log files
        """
        result = []
        for filename in os.listdir(self._log_directory):
            if filename.endswith(".log"):
                result.append(filename)
        return json.dumps(result)

    @route('/actions')
    def actions(self):
        """
        Get a list of all brewing actions
        ---
        tags:
          - logs
        responses:
          200:
            description: List of all log files
        """
        filename = "./logs/action.log"
        if os.path.isfile(filename) == False:
            return
        import csv
        array = []
        with open(filename, 'rb') as f:
            reader = csv.reader(f)
            for row in reader:
                try:
                    array.append([int((datetime.datetime.strptime(row[0], "%Y-%m-%d %H:%M:%S") - datetime.datetime(1970, 1, 1)).total_seconds()) * 1000, row[1]])
                except:
                    pass
        return json.dumps(array)

    @route('/<file>', methods=["DELETE"])
    def clearlog(self, file):
        """
        Delete a log file by name
        ---
        tags:
          - logs
        parameters:
          - in: path
            name: file
            schema:
              type: string
            required: true
            description: File name
        responses:
          204:
            description: Log deleted
        """
        if not self.check_filename(file):
            return ('File Not Found', 404)

        filename = "./logs/%s" % file
        if os.path.isfile(filename) == True:
            os.remove(filename)
            cbpi.notify("log deleted succesfully", "")
            return ('', 204)
        else:
            cbpi.notify("Failed to delete log", "", type="danger")
            return ('', 404)


    def read_log_as_json(self, type, id):
        filename = "./logs/%s_%s.log" % (type, id)
        if os.path.isfile(filename) == False:
            return

        import csv
        array = []
        with open(filename, 'rb') as f:
            reader = csv.reader(f)
            for row in reader:
                try:
                    array.append([int((datetime.datetime.strptime(row[0], "%Y-%m-%d %H:%M:%S") - datetime.datetime(1970, 1, 1)).total_seconds()) * 1000, float(row[1])])
                except:
                    pass
        return array

    def convert_chart_data_to_json(self, chart_data):
        return {"name": chart_data["name"], "data": self.read_log_as_json(chart_data["data_type"], chart_data["data_id"])}

    @route('/<t>/<int:id>', methods=["POST"])
    def get_logs_as_json(self, t, id):

        """
        Get Log as json
        ---
        tags:
          - logs
        parameters:
          - in: path
            name: id
            schema:
              type: string
            required: true
            description: id of the file
        responses:
          200:
            description: Log File Data
        """

        data = request.json
        result = []
        if t == "s":
            name = cbpi.cache.get("sensors").get(id).name
            result.append({"name": name, "data": self.read_log_as_json("sensor", id)})

        if t == "k":
            kettle = cbpi.cache.get("kettle").get(id)
            result = map(self.convert_chart_data_to_json, cbpi.brewing.get_controller(kettle.logic).get("class").chart(kettle))

        if t == "f":
            fermenter = cbpi.cache.get("fermenter").get(id)
            result = map(self.convert_chart_data_to_json, cbpi.fermentation.get_controller(fermenter.logic).get("class").chart(fermenter))

        return json.dumps(result)

    @route('/download/<file>')
    @cbpi.nocache
    def download(self, file):

        """
        Download a log file by name
        ---
        tags:
          - logs
        parameters:
          - in: path
            name: file
            schema:
              type: string
            required: true
            description: filename
        responses:
          200:
            description: Log file downloaded
        """
        if not self.check_filename(file):
            return ('File Not Found', 404)
        return send_from_directory('../../logs', file, as_attachment=True, attachment_filename=file)

    def check_filename(self, name):
        import re
        pattern = re.compile('^([A-Za-z0-9-_])+.log$')

        return True if pattern.match(name) else False

@cbpi.addon.core.initializer()
def init(cbpi):
    """
    Initializer for the message module
    :param app: the flask app
    :return: None
    """
    LogView.register(cbpi._app, route_base='/api/logs')