124 lines
4.8 KiB
Python
124 lines
4.8 KiB
Python
# project/__init__.py
|
|
|
|
import os
|
|
from flask import Flask, render_template, request, session, send_from_directory, jsonify, url_for, flash, redirect, Response
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from werkzeug.utils import secure_filename
|
|
from sqlalchemy import and_
|
|
import mysql.connector
|
|
|
|
UPLOAD_FOLDER = '/root/tag-server/flask/app/docs'
|
|
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
|
|
|
|
app = Flask('app', static_url_path='')
|
|
app.config.update(
|
|
DEBUG=True,
|
|
SQLALCHEMY_DATABASE_URI='mysql+mysqlconnector://website:henrypump@127.0.0.1/poconsole'
|
|
# SQLALCHEMY_DATABASE_URI='sqlite:///../database.db',
|
|
)
|
|
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
|
|
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
|
|
app.secret_key = 'henry_pump'
|
|
db = SQLAlchemy(app)
|
|
|
|
def allowed_file(filename):
|
|
return '.' in filename and \
|
|
filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
|
|
|
|
|
|
@app.route('/', defaults={'path': ''})
|
|
@app.route('/<path:path>')
|
|
def catch_all(path):
|
|
return app.send_static_file('index.html')
|
|
|
|
from .datalogger import datalogger
|
|
from .datalogger.models import *
|
|
|
|
@app.route('/api/latest')
|
|
def get_latest_tag_vals():
|
|
res = db.engine.execute('SELECT v1.id as id, v1.created_on as dtime, t.id as t_id, t.name as name, t.tag as tag, v1.value as value, t.units as units, t.description as description, t.min_expected as min_expected, t.max_expected as max_expected FROM tag_vals v1 INNER JOIN tags t ON t.id = v1.tag_id WHERE v1.id = (SELECT v2.id FROM tag_vals v2 WHERE v2.tag_id = v1.tag_id ORDER BY v2.id DESC LIMIT 1) ORDER BY t.id')
|
|
lat = res.fetchall()
|
|
latest_tags = list(map(latest_to_obj, lat))
|
|
return jsonify(latest_tags)
|
|
|
|
|
|
@app.route('/api/valuesbetween/<string:ids>/<string:start>/<string:end>')
|
|
def get_tag_vals_between(ids, start, end):
|
|
ids = ids.split(',')
|
|
res = Tag_val.query.filter(and_(Tag_val.tag_id.in_(ids), Tag_val.created_on > start, Tag_val.created_on <= end)).all()
|
|
return jsonify([i.serialize for i in res])
|
|
|
|
|
|
@app.route('/api/multipletags/<string:ids>')
|
|
def get_multiple_tags(ids):
|
|
ids = ids.split(',')
|
|
res = Tag.query.filter(Tag.id.in_(ids)).all()
|
|
return jsonify([i.serialize for i in res])
|
|
|
|
@app.route('/doc/upload', methods=['POST'])
|
|
def upload_file():
|
|
# check if the post request has the file part
|
|
if 'file' not in request.files:
|
|
flash('No file part')
|
|
return redirect("/#/docs")
|
|
file = request.files['file']
|
|
# if user does not select file, browser also
|
|
# submit a empty part without filename
|
|
if file.filename == '':
|
|
flash('No selected file')
|
|
return redirect("/#/docs")
|
|
if file and allowed_file(file.filename):
|
|
filename = secure_filename(file.filename)
|
|
file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
|
|
d = Doc(name=filename)
|
|
db.session.add(d)
|
|
db.session.commit()
|
|
return redirect(url_for('uploaded_file',
|
|
filename=filename))
|
|
return redirect("/#/docs")
|
|
|
|
@app.route('/docs/<filename>')
|
|
def uploaded_file(filename):
|
|
return send_from_directory(app.config['UPLOAD_FOLDER'],
|
|
filename)
|
|
|
|
@app.route('/csv/all')
|
|
def get_csv_all():
|
|
csv_string = "datetime,"
|
|
all_tags = [i.serialize for i in Tag.query.all()]
|
|
all_tag_names = [x['name'] for x in all_tags]
|
|
for x in all_tag_names:
|
|
csv_string += "{},".format(x)
|
|
csv_string += "\n"
|
|
|
|
all_vals = [i.serialize for i in Tag_val.query.all()]
|
|
val_objs = [{'value': x['value'], 'tag_name': x['tag']['name'], 'datetime': x['created_on']} for x in all_vals]
|
|
for v in val_objs:
|
|
tag_ind = all_tag_names.index(v['tag_name'])
|
|
csv_string += "{},".format(v['datetime']) + "," * tag_ind + "{},".format(v['value']) + "," * (len(all_tag_names) - tag_ind) + "\n"
|
|
return Response(
|
|
csv_string,
|
|
mimetype="text/csv",
|
|
headers={"Content-disposition":
|
|
"attachment; filename=datadump.csv"})
|
|
|
|
@app.route('/csv/<string:ids>')
|
|
def get_csv_selected(ids):
|
|
csv_string = "datetime,"
|
|
all_tags = [i.serialize for i in Tag.query.filter(Tag.id.in_(ids)).all()]
|
|
all_tag_names = [x['name'] for x in all_tags]
|
|
for x in all_tag_names:
|
|
csv_string += "{},".format(x)
|
|
csv_string += "\n"
|
|
|
|
all_vals = [i.serialize for i in Tag_val.query.filter(Tag_val.tag_id.in_(ids)).all()]
|
|
val_objs = [{'value': x['value'], 'tag_name': x['tag']['name'], 'datetime': x['created_on']} for x in all_vals]
|
|
for v in val_objs:
|
|
tag_ind = all_tag_names.index(v['tag_name'])
|
|
csv_string += "{},".format(v['datetime']) + "," * tag_ind + "{},".format(v['value']) + "," * (len(all_tag_names) - tag_ind) + "\n"
|
|
return Response(
|
|
csv_string,
|
|
mimetype="text/csv",
|
|
headers={"Content-disposition":
|
|
"attachment; filename=datadump{}.csv".format(ids.replace(",","-"))})
|