bme_submitter.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# Copyright 2019 Daniel Estevez <daniel@destevez.net>
#
# This file is part of gr-satellites
#
# SPDX-License-Identifier: GPL-3.0-or-later

# This contains code by Peter Horvath <hp@hvt.bme.hu>
# (source: private communication)

import json

from gnuradio import gr
import pmt
import requests
from requests.auth import HTTPBasicAuth


class bme_submitter(gr.basic_block):
    """
    Submits telemetry to https://gnd.bme.hu:8080/
    """
    def __init__(self, user, password, satellite):
        gr.basic_block.__init__(
            self,
            name='bme_submitter',
            in_sig=[],
            out_sig=[])
        self.satellite = satellite

        self.authenticate(user, password)

        self.message_port_register_in(pmt.intern('in'))
        self.set_msg_handler(pmt.intern('in'), self.handle_msg)

    def authenticate(self, user, password):
        self.auth_token = None
        try:
            rauth = requests.post(
                'https://gnd.bme.hu:8080/api/tokens',
                auth=HTTPBasicAuth(user, password), timeout=10)
        except Exception as e:
            print(f'Authentication failed: {e}')
            return
        if rauth.status_code == 200:
            # We hit the jackpot.
            # Let's use the token obtained in the authentication header.
            auth_resp = rauth.json()
            # The token is valid for 60 minutes.
            # After 45 a new one can be requested.
            self.auth_token = auth_resp['token']
        elif rauth.status_code == 401:
            # Unauthorized (wrong credentials)
            print('Wrong credentials, have you registered at'
                  'https://gnd.bme.hu:8080/ ?')
        else:
            print(f'Authentication failed, error code = {rauth.status_code}')

    def putPacket(self, frame):
        if self.auth_token is None:
            print('Not uploading packet to BME, as we are not authenticated')
            return
        packets = [{'satellite': self.satellite,
                    'packet': frame.hex().upper()}]
        auth_header = {'Authorization': 'Bearer ' + self.auth_token}
        rpacket = requests.post(
            'https://gnd.bme.hu:8080/api/packets/bulk',
            json={'packets': packets}, headers=auth_header, timeout=10)
        packet_resp = rpacket.json()
        if rpacket.status_code == 200:
            uploaded_packets = packet_resp["results"]
            for p in uploaded_packets:
                if 'error' in p:
                    print('Checksum error')
        else:
            print('Packet upload failed, token might have expired!')

    def handle_msg(self, msg_pmt):
        msg = pmt.cdr(msg_pmt)
        if not pmt.is_u8vector(msg):
            print('[ERROR] Received invalid message type. Expected u8vector')
            return

        frame = bytes(pmt.u8vector_elements(msg))
        self.putPacket(frame)