hbnet/scripts/dashboard/dashboard.py

###############################################################################
#   HBLink - Copyright (C) 2020 Cortney T. Buffington, N0MJS <n0mjs@me.com>
#   GPS/Data - Copyright (C) 2020 Eric Craw, KF7EEL <kf7eel@qsl.net>
#
#   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, write to the Free Software Foundation,
#   Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA
###############################################################################

'''
This is a web dashboard for the GPS/Data application.
'''

from flask import Flask, render_template, request, Response, Markup
import ast, os
from dashboard_settings import *
import folium
from folium.plugins import MarkerCluster
import re

app = Flask(__name__)


tbl_hdr = '''
<table style="border-color: black; margin-left: auto; margin-right: auto;" border="2" cellspacing="6" cellpadding="2"><tbody>
'''
tbl_ftr = '''
</tbody>
</table>
'''

def get_loc_data():
    try:
        dash_loc = ast.literal_eval(os.popen('cat /tmp/gps_data_user_loc.txt').read())
        tmp_loc = ''
        loc_hdr = '''
    <tr>
    <td style="text-align: center;">
    <h2><strong>&nbsp;Callsign&nbsp;</strong></h2>
    </td>
    <td style="text-align: center;">
    <h2>&nbsp;<strong>Latitude</strong>&nbsp; </h2>
    </td>
    <td style="text-align: center;">
    <h2>&nbsp;<strong>Longitude</strong>&nbsp;</h2>
    </td>
    <td style="text-align: center;">
    <h2>&nbsp;<strong>Local Time</strong>&nbsp;</h2>
    </td>
    </tr>
    '''
        last_known_loc_list = []
        display_number = 15
        for e in dash_loc:
            if display_number == 0:
                break
            else:
                if e['call'] in last_known_loc_list:
                    pass
                if e['call'] not in last_known_loc_list:
                    last_known_loc_list.append(e['call'])
                    display_number = display_number - 1
                    tmp_loc = tmp_loc + '''<tr>
    <td style="text-align: center;"><a href="view_map?track=''' + e['call'] + '''&map_size=full"><strong>''' + e['call'] + '''</strong></a></td>
    <td style="text-align: center;"><strong>&nbsp;''' + str(e['lat']) + '''&nbsp;</strong></td>
    <td style="text-align: center;"><strong>&nbsp;''' + str(e['lon']) + '''&nbsp;</strong></td>
    <td style="text-align: center;">&nbsp;''' + e['time'] + '''&nbsp;</td>
    </tr>'''
        return str(str('<h1 style="text-align: center;">Last Known Location</h1>') + tbl_hdr + loc_hdr + tmp_loc + tbl_ftr)
    except:
        return str('<h1 style="text-align: center;">No data</h1>')


def get_bb_data():
    try:
        dash_bb = ast.literal_eval(os.popen('cat /tmp/gps_data_user_bb.txt').read())
        tmp_bb = ''
        
        bb_hdr = '''
    <tr>
    <td style="text-align: center;">
    <h2><strong>&nbsp;Callsign&nbsp;</strong></h2>
    </td>
    <td style="text-align: center;">
    <h2>&nbsp;<strong>DMR ID</strong>&nbsp; </h2>
    </td>
    <td style="text-align: center;">
    <h2>&nbsp;<strong>Bulletin</strong>&nbsp;</h2>
    </td>
    <td style="text-align: center;">
    <h2>&nbsp;<strong>Local Time</strong>&nbsp;</h2>
    </td>
    </tr>
    '''
        display_number = 10
        
        for e in dash_bb:
            if display_number == 0:
                break
            else:
                display_number = display_number - 1
                tmp_bb = tmp_bb + '''<tr>
        <td style="text-align: center;"><strong>&nbsp;''' + e['call'] + '''&nbsp;</strong></td>
        <td style="text-align: center;">''' + str(e['dmr_id']) + '''</td>
        <td style="text-align: center;"><strong>&nbsp;''' + e['bulletin'] + '''&nbsp;</strong></td>
        <td style="text-align: center;">&nbsp;''' + e['time'] + '''&nbsp;</td>
        </tr>'''

        return str('<h1 style="text-align: center;">Bulletin Board</h1>' + tbl_hdr + bb_hdr + tmp_bb + tbl_ftr)
    except:
        return str('<h1 style="text-align: center;">No data</h1>')

def check_emergency():
    # open emergency txt
    try:
        sos_file = ast.literal_eval(os.popen('cat /tmp/gps_data_user_sos.txt').read())
        if '@NOTICE' in sos_file['message'] and '@SOS' not in sos_file['message']:
            notice_header = '<span style="background-color: #ffffff; color: #008000;">NOTICE:</span>'
        else:
            notice_header = '<span style="background-color: #ff0000; color: #ffffff;">EMERGENCY ACTIVATION</span>'
        value = Markup("""
        <h1 style="text-align: center;">""" +  notice_header  + """</h1>
        <p>&nbsp;</p>
        <table style="width: 441px; margin-left: auto; margin-right: auto;" border="3">
        <tbody>
        <tr>
        <td style="width: 78.3667px;"><span style="text-decoration: underline;"><strong>From:</strong></span></td>
        <td style="width: 345.633px; text-align: center;"><strong>""" + sos_file['call'] + """ - """ + str(sos_file['dmr_id']) + """</strong></td>
        </tr>
        <tr>
        <td style="width: 78.3667px;"><span style="text-decoration: underline;"><strong>Message:</strong></span></td>
        <td style="width: 345.633px; text-align: center;"><strong>""" + sos_file['message'] + """</strong></td>
        </tr>
        <tr>
        <td style="width: 78.3667px;"><span style="text-decoration: underline;"><strong>Time:</strong></span></td>
        <td style="width: 345.633px; text-align: center;"><strong>""" + sos_file['time'] + """</strong></td>
        </tr>
        </tbody>
        </table>
        <p>&nbsp;</p>
         <button onclick="window.open('view_map?track=""" + sos_file['call'] + """&reload=30','_blank' );" type="button" class="emergency_button"><h1>View Station on Map</h1></button>
         <p style="text-align: center;"><a href="https://aprs.fi/""" + sos_file['call'] + """"><strong>View on aprs.fi</strong></a></p> 
         <hr />

        """)
        return value
    except:
        return ''

def aprs_to_latlon(x):
    degrees = int(x) // 100
    minutes = x - 100*degrees
    return degrees + minutes/60 

@app.route('/')
def index():
    value = Markup('<strong>The HTML String</strong>')
    #return get_data()
    return render_template('index.html', title = dashboard_title, logo = logo, emergency = check_emergency())
@app.route('/bulletin_board')
def dash_bb():
    return get_bb_data()
    #return render_template('index.html', data = str(get_data()))
@app.route('/positions')
def dash_loc():
    return get_loc_data()

@app.route('/help/')
def help():
    #return get_data()
    return render_template('help.html', title = dashboard_title, logo = logo, description = description, data_call_type = data_call_type, data_call_id = data_call_id, aprs_ssid = aprs_ssid)
@app.route('/about/')
def about():
    #return get_data()
    return render_template('about.html', title = dashboard_title, logo = logo, contact_name = contact_name, contact_call = contact_call, contact_email = contact_email, contact_website = contact_website)
@app.route('/view_map')
def view_map():
    reload_time = request.args.get('reload')
    track_call = request.args.get('track')
    map_size = request.args.get('map_size')
    user_loc = ast.literal_eval(os.popen('cat /tmp/gps_data_user_loc.txt').read())
    last_known_list = []
    try:
        if track_call:
            #folium_map = folium.Map(location=map_center, zoom_start=int(zoom_level))
            #marker_cluster = MarkerCluster().add_to(folium_map)
            for user_coord in user_loc:
                user_lat = aprs_to_latlon(float(re.sub('[A-Za-z]','', user_coord['lat'])))
                user_lon = aprs_to_latlon(float(re.sub('[A-Za-z]','', user_coord['lon'])))
                if 'S' in user_coord['lat']:
                    user_lat = -user_lat
                if 'W' in user_coord['lon']:
                    user_lon = -user_lon
                if user_coord['call'] not in last_known_list and user_coord['call'] == track_call:
                    folium_map = folium.Map(location=[user_lat, user_lon], tiles=map_theme, zoom_start=15)
                    marker_cluster = MarkerCluster().add_to(folium_map)
                    folium.Marker([user_lat, user_lon], popup="""<i>
                    <table style="width: 150px;">
                    <tbody>
                    <tr>
                    <td style="text-align: center;">Last Location:</td>
                    </tr>
                    <tr>
                    <td style="text-align: center;"><strong>"""+ str(user_coord['call']) +"""</strong></td>
                    </tr>
                    <tr>
                    <td style="text-align: center;"><em>"""+ user_coord['time'] +"""</em></td>
                    </tr>
                    </tbody>
                    </table>
                    </i>
                    """, icon=folium.Icon(color="red", icon="record"), tooltip=str(user_coord['call'])).add_to(folium_map)
                    last_known_list.append(user_coord['call'])
                if user_coord['call'] in last_known_list and user_coord['call'] == track_call:
                    folium.CircleMarker([user_lat, user_lon], popup="""
                    <table style="width: 150px;">
                    <tbody>
                    <tr>
                    <td style="text-align: center;"><strong>""" + user_coord['call'] + """</strong></td>
                    </tr>
                    <tr>
                    <td style="text-align: center;"><em>""" + user_coord['time'] + """</em></td>
                    </tr>
                    </tbody>
                    </table>
                    """, tooltip=str(user_coord['call']), fill=True, fill_color="#3186cc", radius=4).add_to(marker_cluster)
            #return folium_map._repr_html_()
                if not reload_time:
                    reload_time = 120
            if not map_size:
                map_view = '''<table style="width: 1000px; height: 600px; margin-left: auto; margin-right: auto;" border="1">
                        <tbody>
                        <tr>
                        <td>
                        ''' + folium_map._repr_html_() + '''</td>
                        </tr>
                        </tbody>
                        </table>'''
            if map_size == 'full':
                map_view = folium_map._repr_html_()

            return  '{} {}'.format('''
                    <head>
                        <meta charset="UTF-8">
                        <meta http-equiv="refresh" content="''' + str(reload_time) + """" > 
                        <title>""" + dashboard_title + """ - Tracking """+ track_call + """</title>
                    </head>
                    <p style="text-align: center;"><strong>""" + dashboard_title + """ - Tracking """ + track_call + """</strong></p>
                    <p style="text-align: center;"><em>Page automatically reloads every """ + str(reload_time) + """ seconds.</em></p>
                    <p style="text-align: center;">
                        <select name="sample" onchange="location = this.value;">
                         <option value="view_map?track=""" + track_call + """&reload=120">2 Minutes</option>
                         <option value="view_map?track=""" + track_call + """&reload=">Don't Reload</option>
                         <option value="view_map?track=""" + track_call + """&reload=30">30 Seconds</option>
                         <option value="view_map?track=""" + track_call + """&reload=5">5 Minutes</option>
                         <option value="view_map?track=""" + track_call + """&reload=600">10 Minutes</option> 
                        </select> 
                    <p style="text-align: center;"><button onclick="self.close()">Close</button><button onclick="history.back()">Back</button>
                    </p>
                     """, map_view)
    except:
        return """<h1 style="text-align: center;">Station not found.</h1>
                  <p style="text-align: center;"><button onclick="self.close()">Close Window</button>
                </p>"""
    if not track_call:
        folium_map = folium.Map(location=map_center, tiles=map_theme, zoom_start=int(zoom_level))
        marker_cluster = MarkerCluster().add_to(folium_map)
        for user_coord in user_loc:
            user_lat = aprs_to_latlon(float(re.sub('[A-Za-z]','', user_coord['lat'])))
            user_lon = aprs_to_latlon(float(re.sub('[A-Za-z]','', user_coord['lon'])))
            if 'S' in user_coord['lat']:
                user_lat = -user_lat
            if 'W' in user_coord['lon']:
                user_lon = -user_lon
            if user_coord['call'] not in last_known_list:
                folium.Marker([user_lat, user_lon], popup="""<i>
                <table style="width: 150px;">
                <tbody>
                <tr>
                <td style="text-align: center;">Last Location:</td>
                </tr>
                <tr>
                <td style="text-align: center;"><strong>""" + user_coord['call'] + """</strong></td>
                </tr>
                <tr>
                <td style="text-align: center;"><em>""" + user_coord['time'] + """</em></td>
                </tr>
                <tr>
                <td style="text-align: center;"><strong><A href="view_map?track=""" + user_coord['call'] + """" target="_blank">Track Station</A></strong></td>
                </tr>
                </tbody>
                </table>

                </i>""", icon=folium.Icon(color="red", icon="record"), tooltip=str(user_coord['call'])).add_to(folium_map)
                last_known_list.append(user_coord['call'])
            if user_coord['call'] in last_known_list:
                folium.CircleMarker([user_lat, user_lon], popup="""
                <table style="width: 150px;">
                <tbody>
                <tr>
                <td style="text-align: center;"><strong>""" + user_coord['call'] + """</strong></td>
                </tr>
                <tr>
                <td style="text-align: center;"><em>""" + user_coord['time'] + """</em></td>
                </tr>
                </tbody>
                </table>
                """, tooltip=str(user_coord['call']), fill=True, fill_color="#3186cc", radius=4).add_to(marker_cluster)

        return folium_map._repr_html_()
@app.route('/map/')
def map():
    return render_template('map.html', title = dashboard_title, logo = logo)

@app.route('/bulletin_rss.xml')
def bb_rss():
    try:
        dash_bb = ast.literal_eval(os.popen('cat /tmp/gps_data_user_bb.txt').read())
        post_data = ''
        rss_header = """<?xml version="1.0" encoding="UTF-8" ?>
        <rss version="2.0">
        <channel>
          <title>""" + dashboard_title + """ - Bulletin Board Feed</title>
          <link>""" + rss_link + """</link>
          <description>This is the Bulletin Board feed from """ + dashboard_title + """</description>"""
        for entry in dash_bb:
            post_data = post_data + """
             <item>
                <title>""" + entry['call'] + ' - ' + str(entry['dmr_id']) + """</title>
                <link>""" + rss_link + """</link>
                <description>""" + entry['bulletin'] + """ - """ + entry['time'] + """</description>
              </item>
    """
        return Response(rss_header + post_data + "\n</channel>\n</rss>", mimetype='text/xml')
    except:
        return str('<h1 style="text-align: center;">No data</h1>')
if __name__ == '__main__':
    app.run(debug = True, port=dash_port, host=dash_host)