1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
#!/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)
|
